73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
from flask.ext.sqlalchemy import SQLAlchemy
|
|
import utils
|
|
|
|
db = SQLAlchemy()
|
|
|
|
class Users(db.Model):
|
|
uid = db.Column(db.Integer, primary_key=True)
|
|
tid = db.Column(db.Integer)
|
|
name = db.Column(db.String(64))
|
|
username = db.Column(db.String(64), unique=True)
|
|
username_lower = db.Column(db.String(64), unique=True)
|
|
email = db.Column(db.String(64), unique=True)
|
|
password = db.Column(db.String(128))
|
|
admin = db.Column(db.Boolean, default=False)
|
|
|
|
def __init__(self, name, username, email, password):
|
|
self.name = name
|
|
self.username = username
|
|
self.username_lower = username.lower()
|
|
self.email = email.lower()
|
|
self.password = utils.hash_password(password)
|
|
|
|
class Teams(db.Model):
|
|
tid = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(64), unique=True)
|
|
join_code = db.Column(db.String(128), unique=True)
|
|
school = db.Column(db.Text)
|
|
size = db.Column(db.Integer)
|
|
score = db.Column(db.Integer)
|
|
observer = db.Column(db.Boolean)
|
|
owner = db.Column(db.Integer)
|
|
|
|
def __init__(self, name, school):
|
|
self.name = name
|
|
self.school = school
|
|
|
|
class Problems(db.Model):
|
|
pid = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(128))
|
|
category = db.Column(db.String(128))
|
|
description = db.Column(db.Text)
|
|
hint = db.Column(db.Text)
|
|
flag = db.Column(db.Text)
|
|
disabled = db.Column(db.Boolean, default=False)
|
|
value = db.Column(db.Integer)
|
|
solves = db.Column(db.Integer, default=0)
|
|
|
|
def __init__(self, name, category, description, hint, flag, value):
|
|
self.name = name
|
|
self.category = category
|
|
self.description = description
|
|
self.hint = hint
|
|
self.flag = flag
|
|
self.value = value
|
|
|
|
class Files(db.Model):
|
|
fid = db.Column(db.Integer, primary_key=True)
|
|
pid = db.Column(db.Integer)
|
|
location = db.Column(db.Text)
|
|
|
|
def __init__(self, pid, location):
|
|
self.pid = pid
|
|
self.location = location
|
|
|
|
class Solves(db.Model):
|
|
sid = db.Column(db.Integer, primary_key=True)
|
|
pid = db.Column(db.Integer)
|
|
tid = db.Column(db.Integer)
|
|
date = db.Column(db.Integer, default=utils.get_time_since_epoch())
|
|
|
|
def __init__(self, pid, tid):
|
|
self.pid = pid
|
|
self.tid = tid
|