From 0848116ff84d75ebc4c14333f5af6a2371028f5c Mon Sep 17 00:00:00 2001 From: CakesTwix Date: Tue, 19 Nov 2024 09:23:44 +0200 Subject: [PATCH] Fix db creations --- app/db.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/app/db.py b/app/db.py index 0e993f1..ae48155 100644 --- a/app/db.py +++ b/app/db.py @@ -4,13 +4,12 @@ from sqlalchemy.orm import DeclarativeBase, MappedAsDataclass db = SQLAlchemy() - -class Base(DeclarativeBase, MappedAsDataclass): - pass - - -class ImagesCache(Base): +class ImagesCache(db.Model): __tablename__ = "image_cache" codename: Mapped[str] = mapped_column(primary_key=True) image: Mapped[str] = mapped_column() + + def __init__(self, codename, image): + self.codename = codename + self.image = image