InviteMe/app/models.py

import uuid
from datetime import datetime

from werkzeug.security import generate_password_hash, check_password_hash

from app import db


class User(db.Model):
    __tablename__ = "users"

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(100), unique=True, nullable=False)
    password_hash = db.Column(db.String(255), nullable=False)
    role = db.Column(db.String(20), nullable=False, default="staff")
    is_active = db.Column(db.Boolean, default=True, nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)

    created_events = db.relationship("Event", backref="creator", lazy=True)

    def set_password(self, password):
        self.password_hash = generate_password_hash(password)

    def check_password(self, password):
        return check_password_hash(self.password_hash, password)

    @property
    def is_admin(self):
        return self.role == "admin"

    @property
    def is_organizer(self):
        return self.role == "organizer"

    @property
    def is_staff(self):
        return self.role == "staff"

    def can_manage_event(self, event):
        if self.is_admin:
            return True
        if self.is_organizer and event and event.created_by_id == self.id:
            return True
        return False

    def __repr__(self):
        return f"<User {self.username} ({self.role})>"


class Event(db.Model):
    __tablename__ = "events"

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255), nullable=False)
    description = db.Column(db.Text, nullable=True)
    event_date = db.Column(db.DateTime, nullable=True)
    location = db.Column(db.String(255), nullable=True)
    is_active = db.Column(db.Boolean, default=True, nullable=False)
    created_by_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=True)
    created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)

    # Ticket design fields
    ticket_title = db.Column(db.String(255), nullable=True)
    ticket_subtitle = db.Column(db.String(255), nullable=True)
    ticket_body_text = db.Column(db.Text, nullable=True)
    ticket_background_color = db.Column(db.String(7), nullable=True, default="#ffffff")
    ticket_text_color = db.Column(db.String(7), nullable=True, default="#000000")
    ticket_accent_color = db.Column(db.String(7), nullable=True, default="#0d6efd")
    ticket_show_qr = db.Column(db.Boolean, default=True, nullable=False)
    ticket_show_photo = db.Column(db.Boolean, default=False, nullable=False)
    ticket_layout = db.Column(db.String(20), nullable=True, default="portrait")
    logo_filename = db.Column(db.String(255), nullable=True)

    guests = db.relationship("Guest", backref="event", lazy=True, cascade="all, delete-orphan")
    check_ins = db.relationship("CheckIn", backref="event", lazy=True, cascade="all, delete-orphan")

    def __repr__(self):
        return f"<Event {self.name}>"


class Guest(db.Model):
    __tablename__ = "guests"

    id = db.Column(db.Integer, primary_key=True)
    event_id = db.Column(db.Integer, db.ForeignKey("events.id"), nullable=False)
    department = db.Column(db.String(255), nullable=False)
    last_name = db.Column(db.String(255), nullable=True)
    first_name = db.Column(db.String(255), nullable=True)
    middle_name = db.Column(db.String(255), nullable=True)
    full_name = db.Column(db.String(255), nullable=False)
    position = db.Column(db.String(255), nullable=True)
    email = db.Column(db.String(255), nullable=False)
    phone = db.Column(db.String(50), nullable=True)
    photo_filename = db.Column(db.String(255), nullable=True)
    token = db.Column(db.String(36), unique=True, nullable=False, default=lambda: str(uuid.uuid4()))
    qr_filename = db.Column(db.String(255), nullable=True)
    is_active = db.Column(db.Boolean, default=True, nullable=False)
    invitation_sent = db.Column(db.Boolean, default=False, nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)

    check_ins = db.relationship("CheckIn", backref="guest", lazy=True, cascade="all, delete-orphan")

    @property
    def is_checked_in(self):
        return len(self.check_ins) > 0

    @property
    def last_check_in(self):
        if not self.check_ins:
            return None
        return max(check_in.checked_in_at for check_in in self.check_ins)

    def build_full_name(self):
        parts = [p for p in [self.last_name, self.first_name, self.middle_name] if p]
        return " ".join(parts) if parts else self.full_name

    def __repr__(self):
        return f"<Guest {self.full_name} ({self.email})>"


class CheckIn(db.Model):
    __tablename__ = "check_ins"

    id = db.Column(db.Integer, primary_key=True)
    guest_id = db.Column(db.Integer, db.ForeignKey("guests.id"), nullable=False)
    event_id = db.Column(db.Integer, db.ForeignKey("events.id"), nullable=True)
    checked_in_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
    device_info = db.Column(db.String(500), nullable=True)

    def __repr__(self):
        return f"<CheckIn guest_id={self.guest_id} at={self.checked_in_at}>"