luxick/estus-shots-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Further refactoring.

Browse Source
This commit is contained in:
luxick
2019-02-12 17:49:07 +01:00
parent c73b374659
commit e3bb076af6
8 changed files with 238 additions and 225 deletions
Download Patch File Download Diff File Expand all files Collapse all files

250
app.py
View File

@@ -9,12 +9,13 @@ import db
import forms import forms
import models import models
import const import const
from config import Config, roles
logging.basicConfig(filename=const.LOG_PATH, level=logging.DEBUG) logging.basicConfig(filename=const.LOG_PATH, level=logging.DEBUG)
logging.info(f'Starting in working dir: {os.getcwd()}') logging.info(f"Starting in working dir: {os.getcwd()}")
logging.info(f'App base path: {const.BASE_PATH}') logging.info(f"App base path: {const.BASE_PATH}")
def create_app(): def create_app():
@@ -25,15 +26,10 @@ def create_app():
app = create_app() app = create_app()
app.secret_key = 'THIS IS A TEST KEY' app.config.from_object(Config)
ROLES = {
'123': 'readonly',
'1234': 'editor'
}
@app.cli.command('initdb') @app.cli.command("initdb")
def init_db_command(): def init_db_command():
"""Initializes the database.""" """Initializes the database."""
db.init_db() db.init_db()
@@ -41,171 +37,167 @@ def init_db_command():
@app.teardown_appcontext @app.teardown_appcontext
def close_connection(exception): def close_connection(exception):
db = getattr(g, '_database', None) db = getattr(g, "_database", None)
if db is not None: if db is not None:
db.close() db.close()
def set_user_role(data): def set_user_role(data):
"""Set the users role in the flask g object for later usage""" """Set the users role in the flask g object for later usage"""
g.is_editor = data == 'editor' g.is_editor = data == "editor"
def authorize(func): def authorize(func):
@functools.wraps(func) @functools.wraps(func)
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
try: try:
set_user_role(session['role']) set_user_role(session["role"])
except KeyError: except KeyError:
return redirect('/login') return redirect("/login")
return func(*args, **kwargs) return func(*args, **kwargs)
return wrapper return wrapper
@app.route('/login', methods=['GET', 'POST']) @app.route("/login", methods=["GET", "POST"])
def login(): def login():
if request.method == 'GET': if request.method == "GET":
return render_template('login.html') return render_template("login.html")
else: else:
try: try:
password = request.form['password'] password = request.form["password"]
session['role'] = ROLES[password] session["role"] = roles()[password]
return redirect('/') return redirect("/")
except KeyError: except KeyError:
return redirect('login') return redirect("login")
@app.route('/logout') @app.route("/logout")
def logout(): def logout():
session.pop('role', None) session.pop("role", None)
return redirect('login') return redirect("login")
@app.route('/') @app.route("/")
@authorize @authorize
def landing(): def landing():
return redirect('/seasons') return redirect("/seasons")
@app.route('/seasons') @app.route("/seasons")
@authorize @authorize
def season_list(): def season_list():
sql, args = db.load_season() sql, args = db.load_season()
results = db. query_db(sql, args, cls=models.Season) results = db.query_db(sql, args, cls=models.Season)
model = { model = {
'seasons': results, "seasons": results,
'columns': [ "columns": [
('code', '#'), ("code", "#"),
('game', 'Game'), ("game", "Game"),
('description', 'Season Description'), ("description", "Season Description"),
('start', 'Started At'), ("start", "Started At"),
('end', 'Ended At') ("end", "Ended At"),
] ],
} }
return render_template('seasons.html', model=model) return render_template("seasons.html", model=model)
@app.route('/seasons/new', methods=['GET']) @app.route("/seasons/new", methods=["GET"])
@authorize @authorize
def season_new(): def season_new():
return render_template('editseason.html', model={}) return render_template("editseason.html", model={})
@app.route('/seasons/<id>/edit') @app.route("/seasons/<id>/edit")
@authorize @authorize
def season_edit(id: int): def season_edit(id: int):
sql, args = db.load_season(id) sql, args = db.load_season(id)
loaded = db.query_db(sql, args, one=True, cls=models.Season) loaded = db.query_db(sql, args, one=True, cls=models.Season)
return render_template('editseason.html', model=loaded) return render_template("editseason.html", model=loaded)
@app.route('/seasons/save', methods=['POST']) @app.route("/seasons/save", methods=["POST"])
@authorize @authorize
def season_save(): def season_save():
try: try:
season = models.Season.from_form(request.form) season = models.Season.from_form(request.form)
except AttributeError as err: except AttributeError as err:
print(err) print(err)
return render_template('editseason.html', model={}) return render_template("editseason.html", model={})
sql, args = db.save_season_query(season) sql, args = db.save_season_query(season)
res = db.update_db(sql, args) res = db.update_db(sql, args)
return redirect('/seasons') return redirect("/seasons")
@app.route('/seasons/<season_id>', methods=['GET']) @app.route("/seasons/<season_id>", methods=["GET"])
@authorize @authorize
def season_overview(season_id: int): def season_overview(season_id: int):
sql, args = db.load_season(season_id) sql, args = db.load_season(season_id)
db_season = db.query_db(sql, args, one=True, cls=models.Season) db_season = db.query_db(sql, args, one=True, cls=models.Season)
infos = { infos = {
'Number': db_season.code, "Number": db_season.code,
'Game': db_season.game, "Game": db_season.game,
'Start Date': db_season.start, "Start Date": db_season.start,
'End Date': db_season.end if db_season.end else 'Ongoing' "End Date": db_season.end if db_season.end else "Ongoing",
} }
model = { model = {"title": f"{db_season.code} {db_season.game}", "season_info": infos}
'title': f'{db_season.code} {db_season.game}', return render_template("seasonoverview.html", model=model)
'season_info': infos
}
return render_template('seasonoverview.html', model=model)
@app.route('/seasons/<season_id>/episodes', methods=['GET']) @app.route("/seasons/<season_id>/episodes", methods=["GET"])
@authorize @authorize
def episode_list(season_id: int): def episode_list(season_id: int):
sql, args = db.load_season(season_id) sql, args = db.load_season(season_id)
db_season = db.query_db(sql, args, one=True, cls=models.Season) db_season = db.query_db(sql, args, one=True, cls=models.Season)
model = { model = {"season_id": season_id, "season_code": db_season.code}
'season_id': season_id, return render_template("episodelist.html", model=model)
'season_code': db_season.code
}
return render_template('episodelist.html', model=model)
@app.route('/seasons/<season_id>/new', methods=['GET']) @app.route("/seasons/<season_id>/new", methods=["GET"])
@authorize @authorize
def episode_new(season_id: int): def episode_new(season_id: int):
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='New Episode', page_title="New Episode",
form_title='Create New Episode', form_title="Create New Episode",
post_url='/episodes/save') post_url="/episodes/save",
)
form = forms.EpisodeForm(request.form) form = forms.EpisodeForm(request.form)
form.season_id.data = season_id form.season_id.data = season_id
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
@app.route('/episodes/save', methods=['POST']) @app.route("/episodes/save", methods=["POST"])
@authorize @authorize
def episode_save(): def episode_save():
form = forms.EpisodeForm(request.form) form = forms.EpisodeForm(request.form)
val = form.validate() val = form.validate()
return render_template('editepisode.html', form=form) return render_template("editepisode.html", form=form)
@app.route('/players/new') @app.route("/players/new")
@authorize @authorize
def player_new(): def player_new():
form = forms.PlayerForm() form = forms.PlayerForm()
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='Players', page_title="Players",
form_title='Create a new Player', form_title="Create a new Player",
post_url='/players/edit/null' post_url="/players/edit/null",
) )
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
@app.route('/players/edit/<player_id>', methods=['GET', 'POST']) @app.route("/players/edit/<player_id>", methods=["GET", "POST"])
@authorize @authorize
def player_edit(player_id: int): def player_edit(player_id: int):
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='Players', page_title="Players",
form_title=f'Edit Player', form_title=f"Edit Player",
post_url=f'/players/edit/{player_id}' post_url=f"/players/edit/{player_id}",
) )
# Edit Existing Player # Edit Existing Player
if request.method == 'GET': if request.method == "GET":
sql, args = db.load_players(player_id) sql, args = db.load_players(player_id)
player = db.query_db(sql, args, one=True, cls=models.Player) player = db.query_db(sql, args, one=True, cls=models.Player)
@@ -217,7 +209,7 @@ def player_edit(player_id: int):
form.hex_id.data = player.hex_id form.hex_id.data = player.hex_id
model.form_title = f'Edit Player "{player.name}"' model.form_title = f'Edit Player "{player.name}"'
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
# Save POSTed data # Save POSTed data
else: else:
@@ -225,45 +217,43 @@ def player_edit(player_id: int):
if form.validate_on_submit(): if form.validate_on_submit():
player = models.Player.from_form(form) player = models.Player.from_form(form)
res = db.save_player(player) res = db.save_player(player)
return redirect('/players') return redirect("/players")
model.form_title = 'Incorrect Data' model.form_title = "Incorrect Data"
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
@app.route('/players') @app.route("/players")
@authorize @authorize
def player_list(): def player_list():
sql, args = db.load_players() sql, args = db.load_players()
players = db.query_db(sql, args, cls=models.Player) players = db.query_db(sql, args, cls=models.Player)
model = { model = {
'player_list': players, "player_list": players,
'columns': [('id', 'ID'), "columns": [
('name', 'Player Name'), ("id", "ID"),
('alias', 'Alias'), ("name", "Player Name"),
('hex_id', 'Hex ID')] ("alias", "Alias"),
("hex_id", "Hex ID"),
],
} }
return render_template('players.html', model=model) return render_template("players.html", model=model)
@app.route('/drinks') @app.route("/drinks")
@authorize @authorize
def drink_list(): def drink_list():
sql, args = db.load_drinks() sql, args = db.load_drinks()
drinks = db.query_db(sql, args, cls=models.Drink) drinks = db.query_db(sql, args, cls=models.Drink)
model = { model = {
'drinks': drinks, "drinks": drinks,
'columns': [ "columns": [("id", "ID"), ("name", "Drink Name"), ("vol", "Alcohol %")],
('id', 'ID'), "controls": [("edit", "Edit")],
('name', 'Drink Name'),
('vol', 'Alcohol %')
],
'controls': [('edit', 'Edit')]
} }
return render_template('drinks.html', model=model) return render_template("drinks.html", model=model)
@app.route('/drinks/<drink_id>/edit', methods=['GET']) @app.route("/drinks/<drink_id>/edit", methods=["GET"])
@authorize @authorize
def drink_edit(drink_id: int): def drink_edit(drink_id: int):
sql, args = db.load_drinks(drink_id) sql, args = db.load_drinks(drink_id)
@@ -275,83 +265,77 @@ def drink_edit(drink_id: int):
form.vol.data = drink.vol form.vol.data = drink.vol
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='Edit Drink', page_title="Edit Drink",
form_title=f'Edit Drink "{drink.name}"', form_title=f'Edit Drink "{drink.name}"',
post_url='/drinks/save' post_url="/drinks/save",
) )
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
@app.route('/drinks/new', methods=['GET']) @app.route("/drinks/new", methods=["GET"])
@authorize @authorize
def new_drink(): def new_drink():
form = forms.DrinkForm() form = forms.DrinkForm()
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='New Drink', page_title="New Drink",
form_title=f'Create a new Drink', form_title=f"Create a new Drink",
post_url='/drinks/save' post_url="/drinks/save",
) )
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
@app.route('/drinks/save', methods=['POST']) @app.route("/drinks/save", methods=["POST"])
@authorize @authorize
def drink_save(): def drink_save():
form = forms.DrinkForm() form = forms.DrinkForm()
if form.validate_on_submit(): if form.validate_on_submit():
drink = models.Drink.from_form(form) drink = models.Drink.from_form(form)
res = db.save_drink(drink) res = db.save_drink(drink)
return redirect('/drinks') return redirect("/drinks")
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='Drinks', page_title="Drinks", form_title="Edit Drink", post_url="/drinks/save"
form_title='Edit Drink',
post_url='/drinks/save'
) )
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
@app.route('/enemies') @app.route("/enemies")
@authorize @authorize
def enemy_list(): def enemy_list():
sql, args = db.load_enemies() sql, args = db.load_enemies()
enemies = db.query_db(sql, args, cls=models.Enemy) enemies = db.query_db(sql, args, cls=models.Enemy)
model = {'enemies': enemies} model = {"enemies": enemies}
return render_template('enemies.html', model=model) return render_template("enemies.html", model=model)
@app.route('/enemies/new', methods=['GET']) @app.route("/enemies/new", methods=["GET"])
@authorize @authorize
def enemy_new(preselect_season=None): def enemy_new(preselect_season=None):
sql, args = db.load_season()
seasons = db.query_db(sql, args, cls=models.Season)
seasons = sorted(seasons, key=lambda s: s.code)
form = forms.EnemyForm() form = forms.EnemyForm()
if preselect_season: if preselect_season:
form.season_id.default = preselect_season form.season_id.default = preselect_season
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='Enemies', page_title="Enemies",
form_title='Create a new Enemy', form_title="Create a new Enemy",
post_url=f'/enemies/edit/null' post_url=f"/enemies/edit/null",
) )
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
@app.route('/enemies/edit/<enemy_id>', methods=['GET', 'POST']) @app.route("/enemies/edit/<enemy_id>", methods=["GET", "POST"])
@authorize @authorize
def enemy_edit(enemy_id: int): def enemy_edit(enemy_id: int):
model = models.GenericFormModel( model = models.GenericFormModel(
page_title='Enemies', page_title="Enemies",
form_title='Edit Enemy', form_title="Edit Enemy",
post_url=f'/enemies/edit/{enemy_id}' post_url=f"/enemies/edit/{enemy_id}",
) )
if request.method == 'GET': if request.method == "GET":
sql, args = db.load_season() sql, args = db.load_season()
seasons = db.query_db(sql, args, cls=models.Season) seasons = db.query_db(sql, args, cls=models.Season)
@@ -365,7 +349,7 @@ def enemy_edit(enemy_id: int):
form.enemy_id.data = enemy_id form.enemy_id.data = enemy_id
model.form_title = f'Edit Enemy "{enemy.name}"' model.form_title = f'Edit Enemy "{enemy.name}"'
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
else: else:
form = forms.EnemyForm() form = forms.EnemyForm()
if form.validate_on_submit(): if form.validate_on_submit():
@@ -376,11 +360,11 @@ def enemy_edit(enemy_id: int):
if form.submit_continue_button.data: if form.submit_continue_button.data:
form.name.data = None form.name.data = None
return enemy_new(preselect_season=enemy.season_id) return enemy_new(preselect_season=enemy.season_id)
return redirect('/enemies') return redirect("/enemies")
model.form_title = 'Incorrect Data' model.form_title = "Incorrect Data"
return render_template('generic_form.html', model=model, form=form) return render_template("generic_form.html", model=model, form=form)
if __name__ == '__main__': if __name__ == "__main__":
app.run() app.run()

5
choices.py
View File

@@ -8,8 +8,8 @@ def season_choices():
""" """
sql, args = db.load_season() sql, args = db.load_season()
seasons = db.query_db(sql, args, cls=models.Season) seasons = db.query_db(sql, args, cls=models.Season)
choices = [(s.id, f'{s.code}: {s.game}') for s in seasons] choices = [(s.id, f"{s.code}: {s.game}") for s in seasons]
choices.insert(0, (-1, 'No Season')) choices.insert(0, (-1, "No Season"))
return choices return choices
@@ -18,6 +18,7 @@ class SeasonChoicesIterable:
This is used to declare choices for WTForms SelectFields at class definition time. This is used to declare choices for WTForms SelectFields at class definition time.
Be aware that this uses an undocumented WTForms feature and is not guaranteed to work. Be aware that this uses an undocumented WTForms feature and is not guaranteed to work.
""" """
def __iter__(self): def __iter__(self):
for choice in season_choices(): for choice in season_choices():
yield choice yield choice

11
config.py Normal file
View File

@@ -0,0 +1,11 @@
import json
import os
def roles():
with open("roles.json", "r") as f:
return json.load(f)
class Config:
SECRET_KEY = os.environ.get("ES_SECRET_KEY")

6
const.py
View File

@@ -2,8 +2,8 @@ import os
BASE_PATH = os.path.dirname(os.path.realpath(__file__)) BASE_PATH = os.path.dirname(os.path.realpath(__file__))
DATABASE_NAME = 'es_debug.db' DATABASE_NAME = "es_debug.db"
DATABASE_PATH = os.path.join(BASE_PATH, DATABASE_NAME) DATABASE_PATH = os.path.join(BASE_PATH, DATABASE_NAME)
LOG_NAME = 'estus-shots.log' LOG_NAME = "estus-shots.log"
LOG_PATH = os.path.join(BASE_PATH, LOG_NAME) LOG_PATH = os.path.join(BASE_PATH, LOG_NAME)

102
db.py
View File

@@ -8,9 +8,9 @@ import const
def connect_db(): def connect_db():
"""Create a new sqlite3 connection and register it in 'g._database'""" """Create a new sqlite3 connection and register it in 'g._database'"""
db = getattr(g, '_database', None) db = getattr(g, "_database", None)
if db is None: if db is None:
log.info(f'Connecting {const.DATABASE_NAME}') log.info(f"Connecting {const.DATABASE_NAME}")
db = g._database = sqlite3.connect(const.DATABASE_PATH) db = g._database = sqlite3.connect(const.DATABASE_PATH)
db.row_factory = sqlite3.Row db.row_factory = sqlite3.Row
@@ -19,7 +19,7 @@ def connect_db():
def query_db(query, args=(), one=False, cls=None): def query_db(query, args=(), one=False, cls=None):
"""Runs an SQL query on an new database connection, returning the fetched rv""" """Runs an SQL query on an new database connection, returning the fetched rv"""
log.info(f'Running query ({query}) with arguments ({args})') log.info(f"Running query ({query}) with arguments ({args})")
cur = connect_db().execute(query, args) cur = connect_db().execute(query, args)
rv = cur.fetchall() rv = cur.fetchall()
cur.close() cur.close()
@@ -45,25 +45,25 @@ def update_db(query, args=()):
def init_db(): def init_db():
"""Initialize the database from the 'schema.sql' script file""" """Initialize the database from the 'schema.sql' script file"""
file_name = 'schema.sql' file_name = "schema.sql"
print(f'Creating database from file: "{file_name}"') print(f'Creating database from file: "{file_name}"')
with connect_db() as conn: with connect_db() as conn:
with open(file_name, 'r') as f: with open(file_name, "r") as f:
try: try:
conn.cursor().executescript(f.read()) conn.cursor().executescript(f.read())
except sqlite3.OperationalError as err: except sqlite3.OperationalError as err:
log.error(f'Cannot create database: {err}') log.error(f"Cannot create database: {err}")
conn.commit() conn.commit()
def save_player_query(player): def save_player_query(player):
if not player.id: if not player.id:
sql = 'insert into player values (?, ?, ?, ?, ?)' sql = "insert into player values (?, ?, ?, ?, ?)"
args = (None, player.real_name, player.alias, player.hex_id, player.anon) args = (None, player.real_name, player.alias, player.hex_id, player.anon)
else: else:
sql = 'update player ' \ sql = (
'set real_name=?, alias=?, hex_id=?, anon=? ' \ "update player " "set real_name=?, alias=?, hex_id=?, anon=? " "where id==?"
'where id==?' )
args = (player.real_name, player.alias, player.hex_id, player.anon, player.id) args = (player.real_name, player.alias, player.hex_id, player.anon, player.id)
return sql, args return sql, args
@@ -74,33 +74,31 @@ def save_player(player):
def load_players(id=None): def load_players(id=None):
sql = 'select * from player' sql = "select * from player"
args = () args = ()
if id: if id:
sql += ' where player.id = ?' sql += " where player.id = ?"
args = (id, ) args = (id,)
sql += ' order by player.id' sql += " order by player.id"
return sql, args return sql, args
def load_drinks(id=None): def load_drinks(id=None):
sql = 'select * from drink' sql = "select * from drink"
args = () args = ()
if id: if id:
sql += ' where drink.id = ?' sql += " where drink.id = ?"
args = (id, ) args = (id,)
sql += ' order by drink.id' sql += " order by drink.id"
return sql, args return sql, args
def save_drink_query(drink): def save_drink_query(drink):
if not drink.id: if not drink.id:
sql = 'insert into drink values (?, ?, ?)' sql = "insert into drink values (?, ?, ?)"
args = (None, drink.name, drink.vol) args = (None, drink.name, drink.vol)
else: else:
sql = 'update drink ' \ sql = "update drink " "set name=?, vol=? " "where id==?"
'set name=?, vol=? ' \
'where id==?'
args = (drink.name, drink.vol, drink.id) args = (drink.name, drink.vol, drink.id)
return sql, args return sql, args
@@ -111,54 +109,58 @@ def save_drink(drink):
def load_enemies(id=None): def load_enemies(id=None):
sql = 'select * from enemy' sql = "select * from enemy"
args = () args = ()
if id: if id:
sql += ' where enemy.id = ?' sql += " where enemy.id = ?"
args = (id, ) args = (id,)
sql += ' order by enemy.id' sql += " order by enemy.id"
return sql, args return sql, args
def save_enemy(enemy: models.Enemy): def save_enemy(enemy: models.Enemy):
if not enemy.id: if not enemy.id:
sql = 'insert into enemy values (?, ?, ?, ?)' sql = "insert into enemy values (?, ?, ?, ?)"
args = (None, enemy.name, enemy.boss, enemy.season_id) args = (None, enemy.name, enemy.boss, enemy.season_id)
else: else:
sql = 'update enemy ' \ sql = "update enemy " "set name=?, boss=?, season_id=? " "where id==?"
'set name=?, boss=?, season_id=? ' \
'where id==?'
args = (enemy.name, enemy.boss, enemy.season_id, enemy.id) args = (enemy.name, enemy.boss, enemy.season_id, enemy.id)
return sql, args return sql, args
def save_season_query(season: models.Season): def save_season_query(season: models.Season):
if not season.id: if not season.id:
sql = 'insert into season values (?, ?, ?, ?, ?, ?)' sql = "insert into season values (?, ?, ?, ?, ?, ?)"
args = (None, args = (
season.game, None,
season.description, season.game,
season.start, season.description,
season.end, season.start,
season.code) season.end,
season.code,
)
else: else:
sql = 'update season ' \ sql = (
'set game=?, description=?, start=?, end=?, code=? ' \ "update season "
'where id==?' "set game=?, description=?, start=?, end=?, code=? "
args = (season.game, "where id==?"
season.description, )
season.start, args = (
season.end, season.game,
season.code, season.description,
season.id) season.start,
season.end,
season.code,
season.id,
)
return sql, args return sql, args
def load_season(id=None): def load_season(id=None):
sql = 'select * from season' sql = "select * from season"
args = () args = ()
if id: if id:
sql += ' where season.id = ?' sql += " where season.id = ?"
args = (id, ) args = (id,)
sql += ' order by season.start' sql += " order by season.start"
return sql, args return sql, args

65
forms.py
View File

@@ -1,42 +1,53 @@
from flask_wtf import FlaskForm from flask_wtf import FlaskForm
from wtforms import DateField, TimeField, StringField, SubmitField, BooleanField, \ from wtforms import (
DecimalField, SelectField DateField,
TimeField,
StringField,
SubmitField,
BooleanField,
DecimalField,
SelectField,
)
from wtforms.validators import DataRequired from wtforms.validators import DataRequired
import choices import choices
class EpisodeForm(FlaskForm): class EpisodeForm(FlaskForm):
season_id = StringField('Season ID', render_kw={'readonly': True}) season_id = StringField("Season ID", render_kw={"readonly": True})
episode_id = StringField('Episode ID', render_kw={'readonly': True}) episode_id = StringField("Episode ID", render_kw={"readonly": True})
code = StringField('Episode Code', validators=[DataRequired()]) code = StringField("Episode Code", validators=[DataRequired()])
title = StringField('Title', validators=[DataRequired()]) title = StringField("Title", validators=[DataRequired()])
description = StringField('Description', validators=[]) description = StringField("Description", validators=[])
date = DateField('Episode Date', format='%Y-%m-%d', validators=[DataRequired()]) date = DateField("Episode Date", format="%Y-%m-%d", validators=[DataRequired()])
start = TimeField('Start Time', format='%H:%M', validators=[DataRequired()]) start = TimeField("Start Time", format="%H:%M", validators=[DataRequired()])
end = TimeField('End Time', format='%H:%M', validators=[DataRequired()]) end = TimeField("End Time", format="%H:%M", validators=[DataRequired()])
submit_button = SubmitField('Submit') submit_button = SubmitField("Submit")
class PlayerForm(FlaskForm): class PlayerForm(FlaskForm):
player_id = StringField('Player ID', render_kw={'readonly': True}) player_id = StringField("Player ID", render_kw={"readonly": True})
real_name = StringField('Real Name') real_name = StringField("Real Name")
alias = StringField('Player Alias', validators=[DataRequired()]) alias = StringField("Player Alias", validators=[DataRequired()])
hex_id = StringField('Hex ID') hex_id = StringField("Hex ID")
anonymize = BooleanField('Anonymize (Show only player alias)') anonymize = BooleanField("Anonymize (Show only player alias)")
submit_button = SubmitField('Submit') submit_button = SubmitField("Submit")
class DrinkForm(FlaskForm): class DrinkForm(FlaskForm):
drink_id = StringField('Drink ID', render_kw={'readonly': True}) drink_id = StringField("Drink ID", render_kw={"readonly": True})
name = StringField('Name', validators=[DataRequired()]) name = StringField("Name", validators=[DataRequired()])
vol = DecimalField('Alcohol %', validators=[DataRequired()]) vol = DecimalField("Alcohol %", validators=[DataRequired()])
submit_button = SubmitField('Submit') submit_button = SubmitField("Submit")
class EnemyForm(FlaskForm): class EnemyForm(FlaskForm):
enemy_id = StringField('Enemy ID', render_kw={'readonly': True})
season_id = SelectField('Season', choices=choices.SeasonChoicesIterable(), coerce=int) enemy_id = StringField("Enemy ID", render_kw={"readonly": True})
name = StringField('Name', validators=[DataRequired()]) season_id = SelectField(
is_boss = BooleanField('Is Boss') "Season", choices=choices.SeasonChoicesIterable(), coerce=int
submit_button = SubmitField('Submit') )
submit_continue_button = SubmitField('Submit and Continue') name = StringField("Name", validators=[DataRequired()])
is_boss = BooleanField("Is Boss")
submit_button = SubmitField("Submit")
submit_continue_button = SubmitField("Submit and Continue")

20
models.py
View File

@@ -80,32 +80,32 @@ class Season:
@classmethod @classmethod
def from_form(cls, form): def from_form(cls, form):
id = form.get('id', None) id = form.get("id", None)
id = int(id) if id else None id = int(id) if id else None
game = form.get('game', None) game = form.get("game", None)
if not game: if not game:
raise AttributeError(INVALID_STR.format('game')) raise AttributeError(INVALID_STR.format("game"))
game = str(game) game = str(game)
description = form.get('description', None) description = form.get("description", None)
start = form.get('start', None) start = form.get("start", None)
try: try:
start = datetime.date.fromisoformat(start) start = datetime.date.fromisoformat(start)
except Exception: except Exception:
raise AttributeError(INVALID_STR.format('start')) raise AttributeError(INVALID_STR.format("start"))
end = form.get('end', None) end = form.get("end", None)
if end: if end:
try: try:
end = datetime.date.fromisoformat(end) end = datetime.date.fromisoformat(end)
except Exception: except Exception:
raise AttributeError(INVALID_STR.format('end')) raise AttributeError(INVALID_STR.format("end"))
code = form.get('code', None) code = form.get("code", None)
if not code: if not code:
raise AttributeError(INVALID_STR.format('code')) raise AttributeError(INVALID_STR.format("code"))
self = cls(id, game, description, start, end, code) self = cls(id, game, description, start, end, code)
return self return self

4
roles.json Normal file
View File

@@ -0,0 +1,4 @@
{
"<editor-password>": "editor",
"<readonly-password": "readonly"
}