NexusDashboard/app/models.py
2022-03-12 20:09:35 -06:00

1070 lines
22 KiB
Python

from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
from flask_user import UserMixin
from wtforms import ValidationError
import logging
from flask_sqlalchemy import BaseQuery
from sqlalchemy.dialects import mysql
from sqlalchemy.exc import OperationalError, StatementError
from sqlalchemy.types import JSON
from time import sleep
import random
import string
# retrying query to work around python trash collector
# killing connections of other gunicorn workers
class RetryingQuery(BaseQuery):
__retry_count__ = 3
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
def __iter__(self):
attempts = 0
while True:
attempts += 1
try:
return super().__iter__()
except OperationalError as ex:
if "server closed the connection unexpectedly" not in str(ex):
raise
if attempts < self.__retry_count__:
sleep_for = 2 ** (attempts - 1)
logging.error(
"Database connection error: {} - sleeping for {}s"
" and will retry (attempt #{} of {})".format(
ex, sleep_for, attempts, self.__retry_count__
)
)
sleep(sleep_for)
continue
else:
raise
except StatementError as ex:
if "reconnect until invalid transaction is rolled back" not in str(ex):
raise
self.session.rollback()
db = SQLAlchemy(query_class=RetryingQuery)
migrate = Migrate()
class PlayKey(db.Model):
__tablename__ = 'play_keys'
id = db.Column(db.Integer, primary_key=True)
key_string = db.Column(
mysql.CHAR(19),
nullable=False,
unique=True
)
key_uses = db.Column(
mysql.INTEGER,
nullable=False,
server_default='1'
)
created_at = db.Column(
mysql.TIMESTAMP,
nullable=False,
server_default=db.func.now()
)
active = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='1'
)
notes = db.Column(
mysql.TEXT,
nullable=True,
)
times_used = db.Column(
mysql.INTEGER,
nullable=False,
server_default='0'
)
@staticmethod
def key_is_valid(*, key_string=None):
key = PlayKey.query.filter(PlayKey.key_string == key_string).first()
if not (key and key.active and key.key_uses > 0):
raise ValidationError(
'Not a valid Play Key'
)
else:
return key.id
@staticmethod
def create(*, count=1, uses=1):
for i in range(count):
key = ""
for j in range(4):
key += ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(4)) + '-'
# Remove last dash
key = key[:-1]
new_key = PlayKey(
key_string=key,
key_uses=uses
)
db.session.add(new_key)
db.session.commit()
def delete(self):
db.session.delete(self)
db.session.commit()
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
class Account(db.Model, UserMixin):
__tablename__ = 'accounts'
id = db.Column(
db.Integer(),
primary_key=True
)
username = db.Column(
'name',
db.VARCHAR(35),
nullable=False,
unique=True
)
email = db.Column(
db.Unicode(255),
nullable=True,
server_default='',
unique=False
)
email_confirmed_at = db.Column(db.DateTime())
password = db.Column(
db.Text(),
nullable=False,
server_default=''
)
gm_level = db.Column(
mysql.INTEGER(unsigned=True),
nullable=False,
server_default='0'
)
locked = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
active = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='1'
)
banned = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
play_key_id = db.Column(
mysql.INTEGER,
db.ForeignKey(PlayKey.id, ondelete='CASCADE'),
nullable=True
)
play_key = db.relationship(
'PlayKey',
backref="accounts",
passive_deletes=True
)
created_at = db.Column(
mysql.TIMESTAMP,
nullable=False,
server_default=db.func.now()
)
mute_expire = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
server_default='0'
)
@staticmethod
def get_user_by_id(*, user_id=None):
return Account.query.filter(user_id == Account.id).first()
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class AccountInvitation(db.Model):
__tablename__ = 'account_invites'
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(255), nullable=False)
# save the user of the invitee
invited_by_user_id = db.Column(
db.Integer,
db.ForeignKey(Account.id, ondelete='CASCADE')
)
invited_by_account = db.relationship(
'Account',
backref="account_invites",
passive_deletes=True
)
# token used for registration page to
# identify user registering
token = db.Column(
db.String(100),
nullable=False,
server_default=''
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
@staticmethod
def get_user_by_id(*, user_id=None):
return Account.query.filter(user_id == Account.id).first()
# This table is cursed, see prop_clone_id
class CharacterInfo(db.Model):
__tablename__ = 'charinfo'
id = db.Column(
mysql.BIGINT,
primary_key=True,
autoincrement=False
)
account_id = db.Column(
db.Integer(),
db.ForeignKey(Account.id, ondelete='CASCADE'),
nullable=False
)
account = db.relationship(
'Account',
backref="charinfo",
passive_deletes=True
)
name = db.Column(
mysql.VARCHAR(35),
nullable=False,
)
pending_name = db.Column(
mysql.VARCHAR(35),
nullable=False,
)
needs_rename = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
# Cursed column
# So what this has to be in an autoincrementing entry for a foreign key
# and so to achieve that with sqlalchemy, we have to make it a primary key
# if you look at the initil migration, it the drops this as a primary key,
# cause it's not supposed to be a primary key
# but why does it have to be a primary key?
# sqlalchemy ignores the autoincrement variable for non-primary keys
# thanks for reading this
prop_clone_id = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
primary_key=True,
autoincrement=True,
unique=True,
)
last_login = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
server_default='0'
)
permission_map = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
server_default='0'
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class CharacterXML(db.Model):
__tablename__ = 'charxml'
id = db.Column(
mysql.BIGINT,
primary_key=True,
)
xml_data = db.Column(
db.Text(4294000000),
nullable=False
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class CommandLog(db.Model):
__tablename__ = 'command_log'
id = db.Column(db.Integer, primary_key=True)
character_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
nullable=False
)
character = db.relationship(
'CharacterInfo',
backref="command_log",
passive_deletes=True
)
command = db.Column(
mysql.VARCHAR(256),
nullable=False
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class Friends(db.Model):
__tablename__ = 'friends'
player_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
primary_key=True,
nullable=False
)
player = db.relationship(
'CharacterInfo',
foreign_keys=[player_id],
backref="player",
passive_deletes=True
)
friend_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
primary_key=True,
nullable=False
)
friend = db.relationship(
'CharacterInfo',
foreign_keys=[friend_id],
backref="friend",
passive_deletes=True
)
best_friend = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class Leaderboard(db.Model):
__tablename__ = 'leaderboard'
id = db.Column(db.Integer, primary_key=True)
game_id = db.Column(
mysql.INTEGER(unsigned=True),
nullable=False,
server_default='0'
)
last_played = db.Column(
mysql.TIMESTAMP,
nullable=False,
server_default=db.func.now()
)
character_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
nullable=False
)
character = db.relationship(
'CharacterInfo',
backref="leaderboards",
passive_deletes=True
)
time = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
server_default='0'
)
score = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
server_default='0'
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class Mail(db.Model):
__tablename__ = 'mail'
id = db.Column(
mysql.INTEGER,
primary_key=True
)
sender_id = db.Column(
mysql.INTEGER,
nullable=False
)
sender_name = db.Column(
mysql.VARCHAR(35),
nullable=False
)
receiver_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
nullable=False
)
receiver = db.relationship(
'CharacterInfo',
backref="mail",
passive_deletes=True
)
receiver_name = db.Column(
mysql.VARCHAR(35),
nullable=False
)
time_sent = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False
)
subject = db.Column(
mysql.TEXT,
nullable=False
)
body = db.Column(
mysql.TEXT,
nullable=False
)
attachment_id = db.Column(
mysql.BIGINT,
nullable=False,
server_default='0'
)
attachment_lot = db.Column(
mysql.INTEGER,
nullable=False,
server_default='0'
)
attachment_subkey = db.Column(
mysql.BIGINT,
nullable=False,
server_default='0'
)
attachment_count = db.Column(
mysql.INTEGER(),
nullable=False,
server_default='0'
)
was_read = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class ObjectIDTracker(db.Model):
__tablename__ = 'object_id_tracker'
last_object_id = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
primary_key=True,
server_default='0'
)
class PetNames(db.Model):
__tablename__ = 'pet_names'
id = db.Column(mysql.BIGINT, primary_key=True)
pet_name = db.Column(
mysql.TEXT,
nullable=False
)
approved = db.Column(
mysql.INTEGER(unsigned=True),
nullable=False,
server_default='0'
)
owner_id = db.Column(
mysql.BIGINT,
nullable=True
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class Property(db.Model):
__tablename__ = 'properties'
id = db.Column(
mysql.BIGINT,
primary_key=True,
autoincrement=False
)
owner_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
nullable=False
)
owner = db.relationship(
'CharacterInfo',
foreign_keys=[owner_id],
backref="properties_owner",
passive_deletes=True
)
template_id = db.Column(
mysql.INTEGER(unsigned=True),
nullable=False,
)
clone_id = db.Column(
mysql.BIGINT(unsigned=True),
db.ForeignKey(CharacterInfo.prop_clone_id, ondelete='CASCADE'),
)
clone = db.relationship(
'CharacterInfo',
foreign_keys=[clone_id],
backref="properties_clone",
passive_deletes=True
)
name = db.Column(
mysql.TEXT,
nullable=False
)
description = db.Column(
mysql.TEXT,
nullable=False
)
rent_amount = db.Column(
mysql.INTEGER,
nullable=False,
)
rent_due = db.Column(
mysql.BIGINT,
nullable=False,
)
privacy_option = db.Column(
mysql.INTEGER,
nullable=False,
)
mod_approved = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
last_updated = db.Column(
mysql.BIGINT,
nullable=False,
)
time_claimed = db.Column(
mysql.BIGINT,
nullable=False,
)
rejection_reason = db.Column(
mysql.TEXT,
nullable=False
)
reputation = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
)
zone_id = db.Column(
mysql.INTEGER,
nullable=False,
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class UGC(db.Model):
__tablename__ = 'ugc'
id = db.Column(
mysql.INTEGER,
primary_key=True
)
account_id = db.Column(
db.Integer(),
db.ForeignKey(Account.id, ondelete='CASCADE'),
nullable=False
)
account = db.relationship(
'Account',
backref="ugc",
passive_deletes=True
)
character_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
nullable=False
)
character = db.relationship(
'CharacterInfo',
backref="ugc",
passive_deletes=True
)
is_optimized = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
lxfml = db.Column(
mysql.MEDIUMBLOB(),
nullable=False
)
bake_ao = db.Column(
mysql.BOOLEAN,
nullable=False,
server_default='0'
)
filename = db.Column(
mysql.TEXT,
nullable=False,
server_default=''
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class PropertyContent(db.Model):
__tablename__ = 'properties_contents'
id = db.Column(
mysql.BIGINT,
primary_key=True,
autoincrement=False
)
property_id = db.Column(
db.BIGINT,
db.ForeignKey(Property.id, ondelete='CASCADE'),
nullable=False
)
property_data = db.relationship(
'Property',
backref="properties_contents",
passive_deletes=True
)
ugc_id = db.Column(
db.INT,
db.ForeignKey(UGC.id, ondelete='CASCADE'),
nullable=True
)
ugc = db.relationship(
'UGC',
backref="properties_contents",
passive_deletes=True
)
lot = db.Column(
mysql.INTEGER,
nullable=False,
)
x = db.Column(
mysql.FLOAT(),
nullable=False,
)
y = db.Column(
mysql.FLOAT(),
nullable=False,
)
z = db.Column(
mysql.FLOAT(),
nullable=False,
)
rx = db.Column(
mysql.FLOAT(),
nullable=False,
)
ry = db.Column(
mysql.FLOAT(),
nullable=False,
)
rz = db.Column(
mysql.FLOAT(),
nullable=False,
)
rw = db.Column(
mysql.FLOAT(),
nullable=False,
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class ActivityLog(db.Model):
__tablename__ = 'activity_log'
id = db.Column(mysql.INTEGER, primary_key=True)
character_id = db.Column(
mysql.BIGINT,
db.ForeignKey(CharacterInfo.id, ondelete='CASCADE'),
nullable=False
)
character = db.relationship(
'CharacterInfo',
backref="avtivity_log",
passive_deletes=True
)
activity = db.Column(
mysql.INTEGER,
nullable=False,
)
time = db.Column(
mysql.BIGINT(unsigned=True),
nullable=False,
)
map_id = db.Column(
mysql.INTEGER,
nullable=False,
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class BugReport(db.Model):
__tablename__ = 'bug_reports'
id = db.Column(mysql.INTEGER, primary_key=True)
body = db.Column(
mysql.TEXT,
nullable=False
)
client_version = db.Column(
mysql.TEXT,
nullable=False
)
other_player_id = db.Column(
mysql.TEXT,
nullable=False
)
selection = db.Column(
mysql.TEXT,
nullable=False
)
submitted = db.Column(
mysql.TIMESTAMP,
nullable=False,
server_default=db.func.now()
)
resolved_time = db.Column(
mysql.TIMESTAMP,
nullable=True,
)
resoleved_by_id = db.Column(
db.Integer(),
db.ForeignKey(Account.id, ondelete='CASCADE'),
nullable=True
)
resoleved_by = db.relationship(
'Account',
backref="bugreports",
passive_deletes=True
)
resolution = db.Column(
mysql.TEXT,
nullable=True
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class Server(db.Model):
__tablename__ = 'servers'
id = db.Column(
mysql.INTEGER,
primary_key=True
)
name = db.Column(
mysql.TEXT,
nullable=False
)
ip = db.Column(
mysql.TEXT,
nullable=False
)
port = db.Column(
mysql.INTEGER,
nullable=False
)
state = db.Column(
mysql.INTEGER,
nullable=False
)
version = db.Column(
mysql.INTEGER,
nullable=False,
server_default='0'
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class Reports(db.Model):
__tablename__ = 'reports'
data = db.Column(
JSON(),
nullable=False
)
report_type = db.Column(
db.VARCHAR(35),
nullable=False,
primary_key=True,
autoincrement=False
)
date = db.Column(
db.Date(),
primary_key=True,
autoincrement=False
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()
class AuditLog(db.Model):
__tablename__ = 'audit_logs'
id = db.Column(
mysql.INTEGER,
primary_key=True
)
account_id = db.Column(
db.Integer(),
db.ForeignKey(Account.id, ondelete='CASCADE'),
nullable=False,
)
account = db.relationship(
'Account',
backref="audit_logs",
passive_deletes=True
)
action = db.Column(
mysql.TEXT,
nullable=False
)
date = db.Column(
mysql.TIMESTAMP,
nullable=False,
server_default=db.func.now()
)
def save(self):
db.session.add(self)
db.session.commit()
db.session.refresh(self)
def delete(self):
db.session.delete(self)
db.session.commit()