mirror of
https://github.com/fuckpiracyshield/api.git
synced 2024-12-23 02:50:52 +01:00
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
|
import sys
|
||
|
import os
|
||
|
|
||
|
# I hate python imports
|
||
|
current = os.path.dirname(os.path.realpath(__file__))
|
||
|
parent = os.path.dirname(current)
|
||
|
sys.path.append(parent)
|
||
|
|
||
|
import tornado
|
||
|
|
||
|
from ioutils.protected import ProtectedHandler
|
||
|
from ioutils.errors import ErrorCode, ErrorMessage
|
||
|
|
||
|
from piracyshield_service.forensic.create_archive import ForensicCreateArchiveService
|
||
|
|
||
|
from piracyshield_component.exception import ApplicationException
|
||
|
|
||
|
class UploadForensicHandler(ProtectedHandler):
|
||
|
|
||
|
"""
|
||
|
Appends the forensic evidence.
|
||
|
"""
|
||
|
|
||
|
async def post(self, ticket_id):
|
||
|
if self.initialize_account() == False:
|
||
|
return
|
||
|
|
||
|
# verify permissions
|
||
|
self.permission_service.can_upload_ticket()
|
||
|
|
||
|
# TODO: we could add some mime type filters but we might need to interrogate the ticket model to have a dynamic implementation.
|
||
|
|
||
|
# no archive passed?
|
||
|
if 'archive' not in self.request.files:
|
||
|
return self.error(status_code = 400, error_code = ErrorCode.MISSING_FILE, message = ErrorMessage.MISSING_FILE)
|
||
|
|
||
|
# get the file data
|
||
|
archive_handler = self.request.files['archive'][0]
|
||
|
|
||
|
try:
|
||
|
forensic_create_archive_service = ForensicCreateArchiveService()
|
||
|
|
||
|
await tornado.ioloop.IOLoop.current().run_in_executor(
|
||
|
None,
|
||
|
forensic_create_archive_service.execute,
|
||
|
ticket_id,
|
||
|
archive_handler['filename'],
|
||
|
archive_handler['body']
|
||
|
)
|
||
|
|
||
|
self.success()
|
||
|
|
||
|
except ApplicationException as e:
|
||
|
self.error(status_code = 400, error_code = e.code, message = e.message)
|