1
0
mirror of https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git synced 2025-08-14 00:25:46 +02:00

Создание чата4

This commit is contained in:
MoonTestUse1
2025-01-05 05:56:00 +06:00
parent 679a2290de
commit 0dd0690990
6 changed files with 196 additions and 206 deletions

View File

@@ -2,116 +2,51 @@
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from typing import List
import logging
from ..database import get_db
from ..crud import employees
from ..schemas.employee import Employee, EmployeeCreate, EmployeeUpdate
from ..utils.auth import get_current_admin, get_password_hash
# Настройка логирования
logger = logging.getLogger(__name__)
from app.core.auth import get_current_user, get_password_hash
from app.database import get_db
from app.models.user import User
from app.schemas.user import UserCreate, User as UserSchema
router = APIRouter(tags=["employees"])
router = APIRouter()
@router.post("", response_model=Employee, status_code=status.HTTP_201_CREATED)
async def create_employee(
employee: EmployeeCreate,
db: Session = Depends(get_db),
_: dict = Depends(get_current_admin)
@router.post("/", response_model=UserSchema)
def create_employee(
user: UserCreate,
db: Session = Depends(get_db)
):
"""Create new employee"""
try:
logger.info(f"Creating employee: {employee}")
hashed_password = get_password_hash(employee.password)
return employees.create_employee(db, employee, hashed_password)
except Exception as e:
logger.error(f"Error creating employee: {e}")
# Проверяем, не существует ли уже пользователь с таким email
db_user = db.query(User).filter(User.email == user.email).first()
if db_user:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail="Error creating employee"
status_code=status.HTTP_400_BAD_REQUEST,
detail="Email already registered"
)
# Создаем нового пользователя
hashed_password = get_password_hash(user.password)
db_user = User(
email=user.email,
full_name=user.full_name,
hashed_password=hashed_password,
is_admin=False
)
db.add(db_user)
db.commit()
db.refresh(db_user)
return db_user
@router.get("", response_model=List[Employee])
async def get_employees(
@router.get("/", response_model=List[UserSchema])
def read_employees(
skip: int = 0,
limit: int = 100,
db: Session = Depends(get_db),
_: dict = Depends(get_current_admin)
current_user: User = Depends(get_current_user),
db: Session = Depends(get_db)
):
"""Get all employees"""
try:
logger.info("Getting all employees")
return employees.get_employees(db, skip=skip, limit=limit)
except Exception as e:
logger.error(f"Error getting employees: {e}")
if not current_user.is_admin:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail="Error getting employees"
status_code=status.HTTP_403_FORBIDDEN,
detail="Not enough permissions"
)
@router.get("/{employee_id}", response_model=Employee)
async def get_employee(
employee_id: int,
db: Session = Depends(get_db),
_: dict = Depends(get_current_admin)
):
"""Get employee by ID"""
try:
logger.info(f"Getting employee by ID: {employee_id}")
db_employee = employees.get_employee(db, employee_id)
if db_employee is None:
raise HTTPException(status_code=404, detail="Employee not found")
return db_employee
except HTTPException:
raise
except Exception as e:
logger.error(f"Error getting employee: {e}")
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail="Error getting employee"
)
@router.put("/{employee_id}", response_model=Employee)
async def update_employee(
employee_id: int,
employee: EmployeeUpdate,
db: Session = Depends(get_db),
_: dict = Depends(get_current_admin)
):
"""Update employee data"""
try:
logger.info(f"Updating employee {employee_id}: {employee}")
db_employee = employees.update_employee(db, employee_id, employee)
if db_employee is None:
raise HTTPException(status_code=404, detail="Employee not found")
return db_employee
except HTTPException:
raise
except Exception as e:
logger.error(f"Error updating employee: {e}")
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail="Error updating employee"
)
@router.delete("/{employee_id}", response_model=Employee)
async def delete_employee(
employee_id: int,
db: Session = Depends(get_db),
_: dict = Depends(get_current_admin)
):
"""Delete employee"""
try:
logger.info(f"Deleting employee: {employee_id}")
db_employee = employees.delete_employee(db, employee_id)
if db_employee is None:
raise HTTPException(status_code=404, detail="Employee not found")
return db_employee
except HTTPException:
raise
except Exception as e:
logger.error(f"Error deleting employee: {e}")
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail="Error deleting employee"
)
users = db.query(User).filter(User.is_admin == False).offset(skip).limit(limit).all()
return users