101 lines
4.7 KiB
Python
101 lines
4.7 KiB
Python
|
# storage/json_storage.py
|
||
|
|
import json
|
||
|
|
import os
|
||
|
|
import logging
|
||
|
|
from passlib.context import CryptContext
|
||
|
|
from typing import Optional
|
||
|
|
from models.auth import UserCreate, UserResponse
|
||
|
|
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
class JsonStorage:
|
||
|
|
"""Хранилище пользователей в JSON-файле"""
|
||
|
|
|
||
|
|
def __init__(self, file_path: str = "/etc/gonets/users.json"):
|
||
|
|
self.file_path = file_path
|
||
|
|
self.pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
|
||
|
|
self._ensure_file_exists()
|
||
|
|
|
||
|
|
def _ensure_file_exists(self):
|
||
|
|
"""Создаёт JSON-файл, если он не существует"""
|
||
|
|
try:
|
||
|
|
logger.info(f"Проверка существования файла {self.file_path}")
|
||
|
|
if not os.path.exists(self.file_path):
|
||
|
|
logger.info(f"Создание директории {os.path.dirname(self.file_path)}")
|
||
|
|
os.makedirs(os.path.dirname(self.file_path), exist_ok=True)
|
||
|
|
logger.info(f"Создание файла {self.file_path}")
|
||
|
|
with open(self.file_path, "w") as f:
|
||
|
|
json.dump({"users": [], "next_id": 1}, f)
|
||
|
|
os.chmod(self.file_path, 0o600)
|
||
|
|
logger.info(f"Создан файл {self.file_path}")
|
||
|
|
else:
|
||
|
|
logger.info(f"Файл {self.file_path} уже существует")
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Ошибка при создании файла {self.file_path}: {type(e).__name__} - {str(e)}")
|
||
|
|
raise
|
||
|
|
|
||
|
|
def _read_data(self) -> dict:
|
||
|
|
"""Читает данные из JSON-файла"""
|
||
|
|
try:
|
||
|
|
with open(self.file_path, "r") as f:
|
||
|
|
return json.load(f)
|
||
|
|
except FileNotFoundError:
|
||
|
|
logger.info(f"Файл {self.file_path} не существует, возвращаем пустую структуру")
|
||
|
|
return {"users": [], "next_id": 1}
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Ошибка при чтении файла {self.file_path}: {type(e).__name__} - {str(e)}")
|
||
|
|
raise
|
||
|
|
|
||
|
|
def _write_data(self, data: dict):
|
||
|
|
"""Записывает данные в JSON файл"""
|
||
|
|
try:
|
||
|
|
with open(self.file_path, "w") as f:
|
||
|
|
json.dump(data, f, indent=2)
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Ошибка при записи файла {self.file_path}: {type(e).__name__} - {str(e)}")
|
||
|
|
raise
|
||
|
|
|
||
|
|
def create_user(self, user: UserCreate) -> UserResponse:
|
||
|
|
"""Создаёт нового пользователя"""
|
||
|
|
try:
|
||
|
|
data = self._read_data()
|
||
|
|
users = data["users"]
|
||
|
|
next_id = data["next_id"]
|
||
|
|
if any(u["username"] == user.username for u in users):
|
||
|
|
logger.error(f"Пользователь {user.username} уже существует")
|
||
|
|
raise ValueError("Пользователь уже существует")
|
||
|
|
password_hash = self.pwd_context.hash(user.password)
|
||
|
|
new_user = {
|
||
|
|
"id": next_id,
|
||
|
|
"username": user.username,
|
||
|
|
"password_hash": password_hash,
|
||
|
|
"is_admin": user.is_admin
|
||
|
|
}
|
||
|
|
users.append(new_user)
|
||
|
|
data["next_id"] = next_id + 1
|
||
|
|
self._write_data(data)
|
||
|
|
logger.info(f"Пользователь {user.username} успешно зарегистрирован")
|
||
|
|
return UserResponse(id=new_user["id"], username=new_user["username"], is_admin=new_user["is_admin"])
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Ошибка при создании пользователя {user.username}: {type(e).__name__} - {str(e)}")
|
||
|
|
raise
|
||
|
|
|
||
|
|
def get_user_by_username(self, username: str) -> Optional[dict]:
|
||
|
|
"""Получает пользователя по имени"""
|
||
|
|
try:
|
||
|
|
if not os.path.exists(self.file_path):
|
||
|
|
logger.info(f"Файл {self.file_path} не существует, пользователь {username} не найден")
|
||
|
|
return None
|
||
|
|
data = self._read_data()
|
||
|
|
for user in data["users"]:
|
||
|
|
if user["username"] == username:
|
||
|
|
return user
|
||
|
|
return None
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Ошибка при получении пользователя {username}: {type(e).__name__} - {str(e)}")
|
||
|
|
raise
|
||
|
|
|
||
|
|
def verify_password(self, plain_password: str, hashed_password: str) -> bool:
|
||
|
|
"""Проверяет пароль"""
|
||
|
|
return self.pwd_context.verify(plain_password, hashed_password)
|