Reset hesla, doplneni mailu

src/database/models.py

@@ -18,8 +18,8 @@ class BaseModel:
         return {c: getattr(self, c) for c in self._get_column_names() if c not in self.excluded_columns_in_dict}
 
     @classmethod
-    async def get_one(cls, db_session: AsyncSession, id: int):
-        return (await db_session.scalars(select(cls).filter_by(id=id))).one()
+    async def get_one(cls, db_session: AsyncSession, **kwargs):
+        return (await db_session.scalars(select(cls).filter_by(**kwargs))).one()
 
     @classmethod
     async def create(cls, db_session: AsyncSession, data: dict):
@@ -35,7 +35,7 @@ class BaseModel:
             raise ValueError("Provide either obj or id!")
 
         if not obj:
-            obj = await cls.get_one(db_session, id)
+            obj = await cls.get_one(db_session, id=id)
         for key, value in data.items():
             if key in cls._get_column_names() and getattr(obj, key) != value:
                 setattr(obj, key, value)
@@ -371,3 +371,14 @@ class User(BaseModel):
     licences: Mapped[Set['License']] = relationship()
     flights: Mapped[Set['Flight']] = relationship()
     organizations: Mapped[Set['Organization']] = relationship(secondary=user_is_in_organization)
+
+
+class PasswordResetToken(BaseModel):
+    __tablename__ = "password_reset_token"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    user_id: Mapped[int] = mapped_column(Integer, ForeignKey('user.id'))
+    token: Mapped[str] = mapped_column(String(64), nullable=False)
+    valid_until: Mapped[datetime] = mapped_column(DateTime, nullable=True)
+
+    user: Mapped['User'] = relationship()