73 lines
2.9 KiB
Django/Jinja
73 lines
2.9 KiB
Django/Jinja
from sqlalchemy.orm import Session
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
from fastapi import HTTPException
|
|
from app.modules.{{entity_name}}.models import {{model_name}}
|
|
from app.modules.{{entity_name}}.schema import {{schema_name}}
|
|
|
|
|
|
class {{ class_name }}Service:
|
|
@staticmethod
|
|
def create_{{ entity_name }}(db: Session, data: {{ schema_name }}):
|
|
{% for field in required_fields %}
|
|
if not data.{{ field }}:
|
|
raise ValueError("{{ field }} es requerido")
|
|
{% endfor %}
|
|
|
|
{% for field in unique_fields %}
|
|
if {{ field }}_exists(db, data.{{ field }}):
|
|
raise ValueError("{{ field }} ya existe")
|
|
{% endfor %}
|
|
|
|
new_{{ entity_name }} = {{ model_name }}(**data.dict())
|
|
db.add(new_{{ entity_name }})
|
|
db.commit()
|
|
db.refresh(new_{{ entity_name }})
|
|
return new_{{ entity_name }}
|
|
|
|
@staticmethod
|
|
def get_{{ entity_name }}(db: Session, {{ entity_name }}_id: int, current_user):
|
|
{{ entity_name }} = db.query({{ model_name }}).filter({{ model_name }}.id == {{ entity_name }}_id).first()
|
|
if not {{ entity_name }}:
|
|
raise ValueError("{{ entity_name }} no encontrado")
|
|
return {{ entity_name }}
|
|
|
|
@staticmethod
|
|
def update_{{ entity_name }}(db: Session, {{ entity_name }}_id: int, data: {{ update_schema_name }}, current_user):
|
|
{{ entity_name }} = db.query({{ model_name }}).filter({{ model_name }}.id == {{ entity_name }}_id).first()
|
|
if not {{ entity_name }}:
|
|
raise ValueError("{{ entity_name }} no encontrado")
|
|
|
|
if current_user.role not in ["ROOT", "ADMIN"]:
|
|
raise ValueError("No tienes permisos para actualizar este {{ entity_name }}")
|
|
|
|
update_data = data.dict(exclude_unset=True)
|
|
for key, value in update_data.items():
|
|
setattr({{ entity_name }}, key, value)
|
|
|
|
db.commit()
|
|
db.refresh({{ entity_name }})
|
|
return {{ entity_name }}
|
|
|
|
@staticmethod
|
|
def delete_{{ entity_name }}(db: Session, {{ entity_name }}_id: int, current_user):
|
|
{{ entity_name }} = db.query({{ model_name }}).filter({{ model_name }}.id == {{ entity_name }}_id).first()
|
|
if not {{ entity_name }}:
|
|
raise ValueError("{{ entity_name }} no encontrado")
|
|
|
|
if current_user.role not in ["ROOT", "ADMIN"]:
|
|
raise ValueError("No tienes permisos para eliminar este {{ entity_name }}")
|
|
|
|
try:
|
|
{{ entity_name }}.is_active = False
|
|
{{ entity_name }}.deleted_at = datetime.utcnow()
|
|
{{ entity_name }}.deleted_by = current_user.id
|
|
db.commit()
|
|
db.refresh({{ entity_name }})
|
|
except Exception as e:
|
|
db.rollback()
|
|
raise ValueError(f"Error al eliminar el {{ entity_name }}: {e}")
|
|
|
|
return {"message": "{{ entity_name }} eliminado correctamente"}
|
|
|
|
|