diff --git a/inventory/models/photo.py b/inventory/models/photo.py index 30c9fb0..2f6a17b 100644 --- a/inventory/models/photo.py +++ b/inventory/models/photo.py @@ -24,10 +24,9 @@ class Photo(db.Model): user: Mapped[Optional['User']] = relationship('User', back_populates='photo') worklogs: Mapped[List['WorkLog']] = relationship('WorkLog', secondary=worklog_photos, back_populates='photos') - def __init__(self, filename: str, timestamp: Optional[datetime.datetime] = None, caption: Optional[str] = None): + def __init__(self, filename: str, caption: Optional[str] = None): self.filename = filename self.caption = caption or "" - self.timestamp = timestamp def __repr__(self): return f"" diff --git a/inventory/routes/helpers.py b/inventory/routes/helpers.py index b4cbf7c..9183d23 100644 --- a/inventory/routes/helpers.py +++ b/inventory/routes/helpers.py @@ -82,6 +82,6 @@ def generate_hashed_filename(file_storage, model_name: str) -> str: def get_photo_attachable_class_by_name(name: str): for cls in PhotoAttachable.__subclasses__(): - if cls.__tablename__ == name: + if getattr(cls, '__tablename__', None) == name: return cls return None diff --git a/inventory/routes/photos.py b/inventory/routes/photos.py index 93c06d1..915d27a 100644 --- a/inventory/routes/photos.py +++ b/inventory/routes/photos.py @@ -14,6 +14,8 @@ def save_photo(file, model: str) -> str: rel_path = generate_hashed_filename(file, model) # Absolute path to save + assert current_app.static_folder + abs_path = os.path.join(current_app.static_folder, rel_path) os.makedirs(os.path.dirname(abs_path), exist_ok=True)