36 lines
945 B
Python
36 lines
945 B
Python
from sqlalchemy import Column, String
|
|
from flask import current_app as app
|
|
|
|
from atst.models import Base, types, mixins
|
|
from atst.database import db
|
|
from atst.uploader import UploadError
|
|
|
|
|
|
class AttachmentError(Exception):
|
|
pass
|
|
|
|
|
|
class Attachment(Base, mixins.TimestampsMixin):
|
|
__tablename__ = "attachments"
|
|
|
|
id = types.Id()
|
|
filename = Column(String, nullable=False)
|
|
object_name = Column(String, unique=True, nullable=False)
|
|
|
|
@classmethod
|
|
def attach(cls, fyle):
|
|
try:
|
|
filename, object_name = app.uploader.upload(fyle)
|
|
except UploadError as e:
|
|
raise AttachmentError("Could not add attachment. " + str(e))
|
|
|
|
attachment = Attachment(filename=filename, object_name=object_name)
|
|
|
|
db.session.add(attachment)
|
|
db.session.commit()
|
|
|
|
return attachment
|
|
|
|
def __repr__(self):
|
|
return "<Attachment(name='{}', id='{}')>".format(self.filename, self.id)
|