60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
|
from sqlalchemy.orm import relationship, Mapped, mapped_column, joinedload
|
||
|
from sqlalchemy import ForeignKey, Integer, Date
|
||
|
from sqlalchemy import select
|
||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
from datetime import date
|
||
|
from typing import Optional
|
||
|
|
||
|
from .base import Base, CommonMixin
|
||
|
from .meal import Meal
|
||
|
|
||
|
|
||
|
class RefreshToken(Base, CommonMixin):
|
||
|
"""Diary represents user diary for given day"""
|
||
|
|
||
|
user_id: Mapped[int] = mapped_column(Integer, ForeignKey("user.id"), unique=True)
|
||
|
token: Mapped[str]
|
||
|
|
||
|
@classmethod
|
||
|
async def get_token(
|
||
|
cls,
|
||
|
session: AsyncSession,
|
||
|
user_id: int,
|
||
|
) -> "Optional[RefreshToken]":
|
||
|
"""get_token."""
|
||
|
query = select(cls).where(cls.user_id == user_id)
|
||
|
return await session.scalar(query)
|
||
|
|
||
|
@classmethod
|
||
|
async def create(
|
||
|
cls, session: AsyncSession, user_id: int, token: str
|
||
|
) -> "RefreshToken":
|
||
|
"""create.
|
||
|
|
||
|
:param session:
|
||
|
:type session: AsyncSession
|
||
|
:param user_id:
|
||
|
:type user_id: int
|
||
|
:param token:
|
||
|
:type token: str
|
||
|
:rtype: "RefreshToken"
|
||
|
"""
|
||
|
existing = await cls.get_token(session, user_id)
|
||
|
|
||
|
if existing:
|
||
|
existing.token = token
|
||
|
return existing
|
||
|
|
||
|
token = cls(
|
||
|
user_id=user_id,
|
||
|
token=token,
|
||
|
)
|
||
|
session.add(token)
|
||
|
|
||
|
try:
|
||
|
await session.flush()
|
||
|
except Exception:
|
||
|
raise AssertionError("invalid token")
|
||
|
|
||
|
return token
|