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)
    session_version = db.Column(db.Integer, default=0, 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)
        self.session_version = (self.session_version or 0) + 1

    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 _photo_paths(self):
        """Возвращает возможные пути к фото: сначала папка мероприятия, потом корень."""
        if not self.photo_filename:
            return []
        from pathlib import Path
        from flask import current_app
        base = Path(current_app.config["PHOTO_FOLDER"])
        paths = [base / self.photo_filename]
        if self.event_id:
            paths.insert(0, base / f"event_{self.event_id}" / self.photo_filename)
        return paths

    @property
    def has_photo_file(self):
        try:
            return any(p.exists() for p in self._photo_paths())
        except Exception:
            return False

    @property
    def photo_url(self):
        if not self.has_photo_file:
            return ""
        from pathlib import Path
        from flask import current_app
        for p in self._photo_paths():
            if p.exists():
                rel = p.relative_to(Path(current_app.config["UPLOAD_FOLDER"]).parent)
                return f"/static/{str(rel).replace('\\', '/')}"
        return ""

    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}>"


class AuditLog(db.Model):
    __tablename__ = "audit_logs"

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey("users.id"), nullable=True)
    username = db.Column(db.String(100), nullable=True)
    action = db.Column(db.String(50), nullable=False)
    target_type = db.Column(db.String(50), nullable=True)
    target_id = db.Column(db.String(50), nullable=True)
    details = db.Column(db.Text, nullable=True)
    ip_address = db.Column(db.String(45), nullable=True)
    created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)

    user = db.relationship("User", backref="audit_logs", lazy=True)

    def __repr__(self):
        return f"<AuditLog {self.action} by {self.username} at {self.created_at}>"


class SmtpSettings(db.Model):
    """Глобальные настройки SMTP для рассылки приглашений."""
    __tablename__ = "smtp_settings"

    id = db.Column(db.Integer, primary_key=True)
    mail_server = db.Column(db.String(255), nullable=False, default="smtp.yandex.ru")
    mail_port = db.Column(db.Integer, nullable=False, default=465)
    mail_use_ssl = db.Column(db.Boolean, nullable=False, default=True)
    mail_username = db.Column(db.String(255), nullable=False, default="")
    mail_password = db.Column(db.String(255), nullable=False, default="")
    mail_default_sender = db.Column(db.String(255), nullable=True)
    use_local_outlook = db.Column(db.Boolean, nullable=False, default=False)
    use_ews = db.Column(db.Boolean, nullable=False, default=False)
    ews_url = db.Column(db.String(500), nullable=True)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)

    @property
    def mail_mode(self):
        if self.use_local_outlook:
            return "outlook"
        if self.use_ews:
            return "ews"
        return "smtp"

    @classmethod
    def get_settings(cls):
        settings = cls.query.first()
        if not settings:
            settings = cls()
            db.session.add(settings)
            db.session.commit()
        return settings

    @classmethod
    def apply_to_app(cls, app):
        """Применяет сохранённые настройки к конфигурации Flask-приложения."""
        try:
            settings = cls.query.first()
            if not settings:
                return
            app.config["MAIL_SERVER"] = settings.mail_server or app.config.get("MAIL_SERVER", "")
            app.config["MAIL_PORT"] = settings.mail_port or app.config.get("MAIL_PORT", 465)
            app.config["MAIL_USE_SSL"] = settings.mail_use_ssl
            app.config["MAIL_USERNAME"] = settings.mail_username or ""
            app.config["MAIL_PASSWORD"] = settings.mail_password or ""
            app.config["MAIL_DEFAULT_SENDER"] = settings.mail_default_sender or settings.mail_username or app.config.get("MAIL_DEFAULT_SENDER", "")
            app.config["MAIL_USE_LOCAL_OUTLOOK"] = settings.use_local_outlook
            app.config["MAIL_USE_EWS"] = settings.use_ews
            app.config["EWS_URL"] = settings.ews_url or ""
        except Exception:
            pass

    def __repr__(self):
        return f"<SmtpSettings {self.mail_server}:{self.mail_port} {self.mail_username}>"