46dea3304f
This commit integrates the backend API for fetching and updating report data. It also includes a normalization function to handle data consistency between the API and local storage. Co-authored-by: anthonymuncher <anthonymuncher@gmail.com>
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
import uuid
|
|
from sqlalchemy import Column, String, Float, Enum, DateTime, ForeignKey, Index
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.sql import func
|
|
from app.database import Base
|
|
import enum
|
|
|
|
# ----------------------
|
|
# Enums
|
|
# ----------------------
|
|
class TicketStatus(str, enum.Enum):
|
|
NEW = "New"
|
|
IN_PROGRESS = "In Progress"
|
|
FIXED = "Fixed"
|
|
|
|
class SeverityLevel(str, enum.Enum):
|
|
LOW = "Low"
|
|
MEDIUM = "Medium"
|
|
HIGH = "High"
|
|
NA = "N/A"
|
|
|
|
# ----------------------
|
|
# User Model
|
|
# ----------------------
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()), index=True)
|
|
name = Column(String, nullable=False)
|
|
email = Column(String, unique=True, nullable=False)
|
|
|
|
tickets = relationship("Ticket", back_populates="user", cascade="all, delete-orphan")
|
|
|
|
def __repr__(self):
|
|
return f"<User(id={self.id}, name={self.name}, email={self.email})>"
|
|
|
|
# ----------------------
|
|
# Ticket Model
|
|
# ----------------------
|
|
class Ticket(Base):
|
|
__tablename__ = "tickets"
|
|
|
|
id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()), index=True)
|
|
user_id = Column(String, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
|
image_path = Column(String, nullable=False)
|
|
category = Column(String, nullable=False)
|
|
severity = Column(Enum(SeverityLevel), nullable=False, default=SeverityLevel.NA)
|
|
description = Column(String, default="")
|
|
status = Column(Enum(TicketStatus), nullable=False, default=TicketStatus.NEW)
|
|
latitude = Column(Float, nullable=False)
|
|
longitude = Column(Float, nullable=False)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
updated_at = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now())
|
|
|
|
user = relationship("User", back_populates="tickets")
|
|
|
|
__table_args__ = (
|
|
Index("idx_category_status", "category", "status"),
|
|
)
|
|
|
|
def __repr__(self):
|
|
return f"<Ticket(id={self.id}, category={self.category}, severity={self.severity}, status={self.status}, user_id={self.user_id})>"
|
|
|
|
# ----------------------
|
|
# Ticket Audit Model
|
|
# ----------------------
|
|
class TicketAudit(Base):
|
|
__tablename__ = "ticket_audit"
|
|
|
|
id = Column(String, primary_key=True, default=lambda: str(uuid.uuid4()))
|
|
ticket_id = Column(String, ForeignKey("tickets.id", ondelete="CASCADE"))
|
|
old_status = Column(Enum(TicketStatus))
|
|
new_status = Column(Enum(TicketStatus))
|
|
updated_at = Column(DateTime(timezone=True), server_default=func.now())
|