From 6b6f20a032eb31d747366c8bc270ca81731eaf6a Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Fri, 4 Sep 2026 17:09:24 +0200 Subject: Fixed pre-commit issues --- db/models.py | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) (limited to 'db/models.py') diff --git a/db/models.py b/db/models.py index 8a01c8c..8946bd2 100644 --- a/db/models.py +++ b/db/models.py @@ -1,13 +1,16 @@ import re -from datetime import datetime, UTC -from urllib.parse import urlsplit -from uuid import uuid4, UUID, uuid7 +from datetime import UTC, datetime +from uuid import UUID, uuid7 from sqlalchemy import DateTime -from sqlmodel import SQLModel, Field, Relationship +from sqlmodel import Field, Relationship, SQLModel from starlette.authentication import BaseUser -from authentication.mixins import OAuth2ClientMixin, OAuth2AuthorizationCodeMixin, OAuth2TokenMixin +from authentication.mixins import ( + OAuth2AuthorizationCodeMixin, + OAuth2ClientMixin, + OAuth2TokenMixin, +) from authentication.utils import bcrypt_sha256_hash @@ -21,7 +24,7 @@ class BaseModel(SQLModel): default=None, nullable=True, sa_column_kwargs={ - 'onupdate': lambda: datetime.now(tz=UTC), + "onupdate": lambda: datetime.now(tz=UTC), }, sa_type=DateTime(timezone=True), ) @@ -39,10 +42,12 @@ class User(BaseModel, BaseUser, table=True): def is_authenticated(self) -> bool: return True + class Session(BaseModel, table=True): - user_id: UUID = Field(foreign_key='user.id', exclude=True) + user_id: UUID = Field(foreign_key="user.id", exclude=True) user: User = Relationship() + class ClientApplication(BaseModel, OAuth2ClientMixin, table=True): name: str @@ -52,10 +57,12 @@ class ClientApplication(BaseModel, OAuth2ClientMixin, table=True): for pattern in self.redirect_uris ) + class AuthCode(BaseModel, OAuth2AuthorizationCodeMixin, table=True): user_id: UUID = Field(foreign_key="user.id", ondelete="CASCADE") user: User = Relationship() + class AuthToken(BaseModel, OAuth2TokenMixin, table=True): user_id: UUID = Field(foreign_key="user.id", ondelete="CASCADE") user: User = Relationship() -- cgit v1.3