Fixes + Use SQLAlchemy

1 files changed, 31 insertions, 0 deletions

diff --git a/code/utils/models.py b/code/utils/models.py
new file mode 100644
index 0000000..3dbd6be
--- /dev/null
+++ b/code/utils/models.py
@@ -0,0 +1,31 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    BigInteger,
+)
+
+
+from utils.database import Base
+
+
+class Requests(Base):
+    __tablename__ = "requests"
+
+    id = Column(Integer, primary_key=True, index=True)
+    title = Column(String)
+    release_year = Column(Integer)
+    local_id = Column(Integer)
+    tmdbid = Column(Integer)
+    tvdbid = Column(Integer)
+    user_id = Column(BigInteger)
+
+
+class JellyfinAccounts(Base):
+    __tablename__ = "jellyfin_accounts"
+
+    id = Column(Integer, primary_key=True, index=True)
+    user_id = Column(BigInteger)
+    jellyfin_user_id = Column(String)
+    deletion_time = Column(DateTime)