68 lines
2 KiB
Python
68 lines
2 KiB
Python
import os
|
|
from datetime import datetime
|
|
from flask import current_app as app, request, g, Response
|
|
from marshmallow import ValidationError
|
|
from teal.resource import View
|
|
|
|
from ereuse_devicehub.db import db
|
|
from ereuse_devicehub.resources.tradedocument.models import TradeDocument
|
|
from ereuse_devicehub.resources.hash_reports import insert_hash
|
|
|
|
|
|
def save_doc(data, user):
|
|
"""
|
|
This function allow save a snapshot in json format un a TMP_SNAPSHOTS directory
|
|
The file need to be saved with one name format with the stamptime and uuid joins
|
|
"""
|
|
filename = data['file_name']
|
|
lot = data['lot']
|
|
now = datetime.now()
|
|
year = now.year
|
|
month = now.month
|
|
day = now.day
|
|
hour = now.hour
|
|
minutes = now.minute
|
|
|
|
name_file = f"{year}-{month}-{day}-{hour}-{minutes}_{user}_{filename}"
|
|
path_dir_base = os.path.join(app.config['PATH_DOCUMENTS_STORAGE'] , user)
|
|
path = os.path.join(path_dir_base, str(lot.id))
|
|
path_name = os.path.join(path, name_file)
|
|
|
|
os.system(f'mkdir -p {path}')
|
|
|
|
with open(path_name, 'wb') as doc_file:
|
|
doc_file.write(data['file'])
|
|
|
|
return path_name
|
|
|
|
|
|
class TradeDocumentView(View):
|
|
|
|
def one(self, id: str):
|
|
doc = TradeDocument.query.filter_by(id=id, owner=g.user).one()
|
|
return self.schema.jsonify(doc)
|
|
|
|
def post(self):
|
|
"""Add one document."""
|
|
# import pdb; pdb.set_trace()
|
|
|
|
data = request.get_json(validate=True)
|
|
data['path_name'] = save_doc(data, g.user.email)
|
|
bfile = data.pop('file')
|
|
insert_hash(bfile)
|
|
|
|
# import pdb; pdb.set_trace()
|
|
doc = TradeDocument(**data)
|
|
db.session.add(doc)
|
|
db.session().final_flush()
|
|
ret = self.schema.jsonify(doc)
|
|
ret.status_code = 201
|
|
db.session.commit()
|
|
return ret
|
|
|
|
def delete(self, id):
|
|
doc = TradeDocument.query.filter_by(id=id, owner=g.user).one()
|
|
db.session.delete(doc)
|
|
db.session.commit()
|
|
return Response(status=204)
|