wowlikon/LibraryAPI
1
0
Fork 0
mirror of https://github.com/wowlikon/LiB.git synced 2026-02-04 12:31:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Исправление автомиграции и создания администратора

Browse Source
This commit is contained in:
wowlikon
2025-12-19 21:28:52 +03:00
parent f6ac03a869
commit 719631158d
4 changed files with 84 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
library_service/auth.py
+12 -8
View File
@@ -11,7 +11,7 @@ from sqlmodel import Session, select
from library_service.models.db import Role, User
from library_service.models.dto import TokenData
from library_service.settings import get_session
from library_service.settings import get_session, get_logger
# Конфигурация из переменных окружения
@@ -20,12 +20,16 @@ SECRET_KEY = os.getenv("SECRET_KEY", "your-secret-key-change-in-production")
ACCESS_TOKEN_EXPIRE_MINUTES = int(os.getenv("ACCESS_TOKEN_EXPIRE_MINUTES", "30"))
REFRESH_TOKEN_EXPIRE_DAYS = int(os.getenv("REFRESH_TOKEN_EXPIRE_DAYS", "7"))
# Хэширование паролей
pwd_context = CryptContext(schemes=["argon2"], deprecated="auto")
# Получение логгера
logger = get_logger("uvicorn")
# OAuth2 схема
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/token")
# Хэширование паролей
pwd_context = CryptContext(schemes=["argon2"], deprecated="auto")
def verify_password(plain_password: str, hashed_password: str) -> bool:
"""Проверка пароль по его хешу."""
@@ -161,7 +165,7 @@ def seed_roles(session: Session) -> dict[str, Role]:
session.commit()
session.refresh(role)
roles[role_data["name"]] = role
print(f"[+] Created role: {role_data['name']}")
logger.info(f"[+] Created role: {role_data['name']}")
return roles
@@ -173,7 +177,7 @@ def seed_admin(session: Session, admin_role: Role) -> User | None:
).all()
if existing_admins:
print(f"[*] Admin already exists: {existing_admins[0].username}")
logger.info(f"[*] Admin already exists: {existing_admins[0].username}")
return None
admin_username = os.getenv("DEFAULT_ADMIN_USERNAME", "admin")
@@ -183,8 +187,8 @@ def seed_admin(session: Session, admin_role: Role) -> User | None:
if not admin_password:
import secrets
admin_password = secrets.token_urlsafe(16)
print(f"[!] Generated admin password: {admin_password}")
print("[!] Please save this password and set DEFAULT_ADMIN_PASSWORD env var")
logger.warning(f"[!] Generated admin password: {admin_password}")
logger.warning("[!] Please save this password and set DEFAULT_ADMIN_PASSWORD env var")
admin_user = User(
username=admin_username,
@@ -200,7 +204,7 @@ def seed_admin(session: Session, admin_role: Role) -> User | None:
session.commit()
session.refresh(admin_user)
print(f"[+] Created admin user: {admin_username}")
logger.info(f"[+] Created admin user: {admin_username}")
return admin_user