28 lines
1 KiB
Python
28 lines
1 KiB
Python
from database.db_session import get_session
|
|
from database.models import Auth
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
|
|
def create_user(login, password):
|
|
"""Создание нового пользователя в базе данных."""
|
|
session = get_session()
|
|
try:
|
|
new_user = Auth(login=login, password=password)
|
|
session.add(new_user)
|
|
session.commit()
|
|
except SQLAlchemyError as e:
|
|
print(f"Ошибка при создании пользователя: {e}")
|
|
session.rollback()
|
|
finally:
|
|
session.close()
|
|
|
|
def check_user(login, password):
|
|
"""Проверка данных пользователя для авторизации."""
|
|
session = get_session()
|
|
try:
|
|
user = session.query(Auth).filter_by(login=login, password=password).first()
|
|
return user is not None
|
|
except SQLAlchemyError as e:
|
|
print(f"Ошибка при проверке пользователя: {e}")
|
|
return False
|
|
finally:
|
|
session.close()
|