generated file: db/models/__init__.py

Author: coslynx

db/models/__init__.py

@@ -0,0 +1,33 @@
+from sqlalchemy import Column, Integer, String, ForeignKey, Boolean, DateTime
+from sqlalchemy.orm import relationship
+from sqlalchemy.ext.declarative import declarative_base
+from datetime import datetime
+
+from .schemas import UserSchema
+
+Base = declarative_base()
+
+class User(Base):
+    __tablename__ = "users"
+
+    id = Column(Integer, primary_key=True, index=True)
+    username = Column(String, unique=True, index=True)
+    password = Column(String)
+    api_key = Column(String, unique=True, index=True)
+
+    def __repr__(self):
+        return f"<User(id={self.id}, username='{self.username}')>"
+
+class APIKey(Base):
+    __tablename__ = "api_keys"
+
+    id = Column(Integer, primary_key=True, index=True)
+    user_id = Column(Integer, ForeignKey("users.id"))
+    api_key = Column(String, unique=True, index=True)
+    is_active = Column(Boolean, default=True)
+    created_at = Column(DateTime, default=datetime.utcnow)
+
+    user = relationship("User", backref="api_keys")
+
+    def __repr__(self):
+        return f"<APIKey(id={self.id}, user_id={self.user_id}, api_key='{self.api_key}', is_active={self.is_active})>"