Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
ProFastCode committed Jul 13, 2024
1 parent 5d6405a commit b8bd221
Show file tree
Hide file tree
Showing 6 changed files with 20 additions and 17 deletions.
11 changes: 5 additions & 6 deletions app/api/anotations.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,15 +2,14 @@
Dependencies
"""

from fastapi import Depends
from typing_extensions import Annotated
from typing import Annotated

from app.core import deps

Database = Annotated[deps.Database, Depends(deps.get_db)]
Database = Annotated[deps.Database, deps.Depends(deps.get_db)]

Security = Annotated[deps.Security, Depends(deps.get_security)]
Security = Annotated[deps.Security, deps.Depends(deps.get_security)]

Logic = Annotated[deps.Logic, Depends(deps.get_logic)]
Logic = Annotated[deps.Logic, deps.Depends(deps.get_logic)]

CurrentUser = Annotated[deps.User, Depends(deps.get_current_user)]
CurrentUser = Annotated[deps.User, deps.Depends(deps.get_current_user)]
3 changes: 2 additions & 1 deletion app/core/deps.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,10 @@
Dependencies
"""

from typing import Annotated, AsyncGenerator

from fastapi import Depends
from fastapi.security import APIKeyHeader
from typing_extensions import Annotated, AsyncGenerator

from app.core.db import Database, SessionLocal
from app.core.logic import Logic
Expand Down
4 changes: 2 additions & 2 deletions app/core/logic.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
from app import logic
from app.core.db import Database
from app.core.security import Security
from app.logic.users import Users


class Logic:
def __init__(self, db: Database, security: Security):
self.users = Users(db, security)
self.users = logic.Users(db, security)


__all__ = ['Logic']
12 changes: 6 additions & 6 deletions app/core/security/jwt.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,18 +11,18 @@ def __init__(self, secret_key: str):

def decode_token(self, token: str) -> dict | None:
try:
payload = jwt.decode(token, self.secret_key, algorithms=["HS256"])
payload = jwt.decode(token, self.secret_key, algorithms=['HS256'])
except Exception:
raise exps.TOKEN_INVALID

exp = payload.get("exp")
exp = payload.get('exp')
if exp and dt.datetime.now(dt.UTC).timestamp() > exp:
raise exps.TOKEN_EXPIRED
return payload.get("payload")
return payload.get('payload')

def encode_token(self, payload: dict, minutes: int) -> str:
claims = {
"payload": payload,
"exp": dt.datetime.now(dt.UTC) + dt.timedelta(minutes=minutes),
'payload': payload,
'exp': dt.datetime.now(dt.UTC) + dt.timedelta(minutes=minutes),
}
return jwt.encode(claims, self.secret_key, algorithm="HS256")
return jwt.encode(claims, self.secret_key, algorithm='HS256')
3 changes: 3 additions & 0 deletions app/logic/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
from .users import Users

__all__ = ['Users']
4 changes: 2 additions & 2 deletions app/repositories/abstract.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
import abc
from typing import (Any, Generic, List, NoReturn, Optional, Sequence, Type,
TypeAlias, TypeVar)

import sqlmodel as sm
from sqlmodel.ext.asyncio.session import AsyncSession
from typing_extensions import (Any, Generic, List, NoReturn, Optional,
Sequence, Type, TypeAlias, TypeVar)

AbstractModel = TypeVar('AbstractModel', bound=sm.SQLModel)
WhereClauses: TypeAlias = Optional[List[sm.DefaultClause | bool]]
Expand Down

0 comments on commit b8bd221

Please sign in to comment.