models.py

import enum

from flask_sqlalchemy import BaseQuery, SQLAlchemy
from sqlalchemy import func
from sqlalchemy.event import listens_for
from sqlalchemy.types import TypeDecorator
from sqlalchemy_searchable import SearchQueryMixin
from sqlalchemy_utils import TSVectorType

from flask_login import LoginManager, UserMixin
from werkzeug.security import generate_password_hash, check_password_hash
import sqlalchemy
import json

db = SQLAlchemy()
login = LoginManager()


@login.user_loader
def load_user(id):
    return User.query.get(int(id))


class DictType(TypeDecorator):

    impl = sqlalchemy.Text()

    def process_bind_param(self, value, dialect):
        if value is not None:
            value = json.dumps(value)

        return value

    def process_result_value(self, value, dialect):
        if value is not None:
            value = json.loads(value)
        return value


class RoomMenu(db.Model):
    id = db.Column(db.Integer, primary_key=True, unique=True)
    room_id = db.Column(db.Integer)
    data = db.Column(DictType)  # This is a dict with keys in it for that type.
    # TODO: Figure out a suitable UI, maybe even using Javascript?


class User(db.Model, UserMixin):
    # Used to login to the Admin Panel
    id = db.Column(db.Integer, primary_key=True, default=1)
    username = db.Column(db.String(100))
    password_hash = db.Column(db.String)

    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)


@listens_for(User.__table__, "after_create")
def create_default_user(target, connection, **kw):
    """Adds a default user to The Underground.
    By default, we assume admin:admin."""
    table = User.__table__
    connection.execute(
        table.insert().values(
            username="admin",
            password_hash=generate_password_hash("admin"),
        )
    )


class Posters(db.Model):
    poster_id = db.Column(db.Integer, primary_key=True, unique=True)
    msg = db.Column(db.String(15), nullable=False)
    movie_id = db.Column(db.Integer, nullable=False)
    # While a maximum of 47 characters seems arbitrary, it is what it is.
    # It appears they mandate the last byte in their char array is null (and sprintf),
    # so perhaps the internal type is char[48]?
    title = db.Column(db.String(47), nullable=False)


class News(db.Model):
    id = db.Column(db.Integer, primary_key=True, unique=True)
    msg = db.Column(db.String, nullable=False)


class PayPosters(db.Model):
    poster_id = db.Column(db.Integer, primary_key=True, unique=True)
    msg = db.Column(db.String(15), nullable=False)
    movie_id = db.Column(db.Integer, nullable=False)
    title = db.Column(db.String(47), nullable=False)
    type = db.Column(db.Integer, nullable=False)
    aspect = db.Column(db.Boolean, nullable=False)


class ConciergeMiis(db.Model):
    mii_id = db.Column(
        db.Integer, db.ForeignKey("mii_data.mii_id"), primary_key=True, unique=True
    )
    clothes = db.Column(db.Integer, nullable=False)
    action = db.Column(db.Integer, nullable=False)
    prof = db.Column(db.String(129), nullable=False)
    movie_id = db.Column(db.Integer, nullable=False)
    voice = db.Column(db.Boolean, default=False, nullable=False)


# MiiData provides the genuine Mii alongside other common information.
class MiiData(db.Model):
    mii_id = db.Column(db.Integer, primary_key=True)
    data = db.Column(db.LargeBinary(74), nullable=False)
    name = db.Column(db.String(10), nullable=False)
    color1 = db.Column(db.String(6), nullable=False)
    color2 = db.Column(db.String(6), nullable=False)


class MiiMsgInfo(db.Model):
    # We make both the Mii's ID and message type unique, as to allow row
    # identification for SQLAlchemy.
    mii_id = db.Column(
        db.Integer, db.ForeignKey("mii_data.mii_id"), primary_key=True, nullable=False
    )
    type = db.Column(db.Integer, primary_key=True, nullable=False)
    seq = db.Column(db.Integer, primary_key=True, nullable=False)
    msg = db.Column(db.String, nullable=False)
    face = db.Column(db.Integer, nullable=False)


class FullTextSearchable(BaseQuery, SearchQueryMixin):
    pass


class Movies(db.Model):
    query_class = FullTextSearchable

    movie_id = db.Column(db.Integer, autoincrement=True, primary_key=True, unique=True)
    category_id = db.Column(
        db.Integer,
        db.ForeignKey("categories.category_id"),
        primary_key=True,
        nullable=False,
    )
    title = db.Column(db.String(48), nullable=False)
    length = db.Column(db.String(8), nullable=False)
    aspect = db.Column(db.Boolean, nullable=False)
    genre = db.Column(db.Integer, nullable=False)
    sp_page_id = db.Column(db.Integer, nullable=False)
    ds_mov_id = db.Column(db.Integer, nullable=True)
    staff = db.Column(db.Boolean, nullable=False)
    date_added = db.Column(db.DateTime, nullable=False, server_default=func.now())

    search_vector = db.Column(TSVectorType("title"))


class PayMovies(db.Model):
    query_class = FullTextSearchable

    movie_id = db.Column(db.Integer, autoincrement=True, primary_key=True, unique=True)
    title = db.Column(db.String, nullable=False)
    length = db.Column(db.String(), nullable=False)
    note = db.Column(db.String(76), nullable=False)
    price = db.Column(db.Integer, nullable=False)
    released = db.Column(db.String(10), nullable=False)
    category_id = db.Column(
        db.Integer,
        db.ForeignKey("pay_categories.category_id"),
        primary_key=True,
        nullable=False,
    )
    date_added = db.Column(db.DateTime, nullable=False, server_default=func.now())
    search_vector = db.Column(TSVectorType("title", "note"))


class PayCategories(db.Model):
    category_id = db.Column(db.Integer, primary_key=True, unique=True)
    name = db.Column(db.String)
    # Starts at 10, goes up by 1 each time
    genre_id = db.Column(db.Integer)


class PayCategoryHeaders(db.Model):
    title = db.Column(db.String, primary_key=True, unique=True)


class Categories(db.Model):
    category_id = db.Column(db.Integer, primary_key=True, unique=True)
    name = db.Column(db.String)


class RoomBGMTypes(enum.Enum):
    SOFT_GUITAR = 1
    NORMAL = 2
    FOLK = 3
    JAZZY = 4
    TRUMPET = 5
    CHIMES = 6
    WESTERN = 7
    HARP = 8

    @classmethod
    def choices(cls):
        return [(choice, choice.name) for choice in cls]

    @classmethod
    def coerce(cls, item):
        return cls(int(item)) if not isinstance(item, cls) else item

    def __str__(self):
        return str(self.value)


class RoomContentBGMTypes(enum.Enum):
    Sadness = 1
    Upbeat = 2
    Relaxing = 3
    Classical = 4
    Playful = 5
    Chill = 6
    Adventurous = 7
    Jolly = 8
    Happiness = 9
    Casino = 10
    Parade = 11
    Excitement = 12
    idk = 13

    @classmethod
    def choices(cls):
        return [(choice, choice.name) for choice in cls]

    @classmethod
    def coerce(cls, item):
        return cls(int(item)) if not isinstance(item, cls) else item

    def __str__(self):
        return str(self.value)


class Rooms(db.Model):
    room_id = db.Column(db.Integer, autoincrement=True, primary_key=True, unique=True)
    bgm = db.Column(db.Enum(RoomBGMTypes))
    mascot = db.Column(db.Boolean)
    contact = db.Column(db.Boolean)
    intro_msg = db.Column(db.String)
    contact_data = db.Column(db.String)
    news = db.Column(db.String)
    level = db.Column(db.Integer, default=1)


class RoomMiis(db.Model):
    room_id = db.Column(db.Integer, primary_key=True, unique=True, autoincrement=True)
    mii_id = db.Column(db.Integer, db.ForeignKey("mii_data.mii_id"), nullable=False)
    mii_msg = db.Column(db.String)


class EvaluateData(db.Model):
    id = db.Column(db.Integer, autoincrement=True, primary_key=True, unique=True)
    movie_id = db.Column(db.Integer, index=True)
    gender = db.Column(db.Integer)
    blood = db.Column(db.Integer)
    age = db.Column(db.Integer)
    vote = db.Column(db.Integer)


class PollData(db.Model):
    id = db.Column(db.Integer, autoincrement=True, primary_key=True, unique=True)
    poll_id = db.Column(db.Integer, index=True)
    wii_num = db.Column(db.BigInteger)
    choice = db.Column(db.Integer)
    age = db.Column(db.Integer)
    gender = db.Column(db.Integer)


class ContentTypes(enum.Enum):
    Video = 0
    Image = 1

    @classmethod
    def choices(cls):
        return [(choice, choice.name) for choice in cls]

    @classmethod
    def coerce(cls, item):
        return cls(int(item)) if not isinstance(item, cls) else item

    def __str__(self):
        return str(self.value)


class LinkTypes(enum.Enum):
    Regular = 0
    TheatreMovie = 1
    TheatreCategory = 2
    Shop = 3
    Room = 4
    Category = 5

    @classmethod
    def choices(cls):
        return [(choice, choice.name) for choice in cls]

    @classmethod
    def coerce(cls, item):
        return cls(int(item)) if not isinstance(item, cls) else item

    def __str__(self):
        return str(self.value)


class IntroInfo(db.Model):
    cnt_id = db.Column(db.Integer, autoincrement=True, primary_key=True, unique=True)
    cnt_type = db.Column(db.Enum(ContentTypes))
    cat_name = db.Column(db.String)
    link_type = db.Column(db.Enum(LinkTypes))
    link_id = db.Column(db.Integer)