diff --git a/src/game_table_model.py b/src/game_table_model.py index 73d69b2..8ac33b0 100644 --- a/src/game_table_model.py +++ b/src/game_table_model.py @@ -11,6 +11,7 @@ class GameTable(db.Model): active = db.Column(db.Boolean) version = db.Column(db.Integer) docker_id = db.Column(db.Integer) + user_id = db.Column(db.Integer) def __repr__(self): return f"" @@ -22,4 +23,5 @@ class GameTable(db.Model): "table_link": self.game_table_link, "active": self.active, "version": self.version, + "user_id": self.user_id } diff --git a/src/tables.py b/src/tables.py index e3a88d1..d01093e 100644 --- a/src/tables.py +++ b/src/tables.py @@ -5,11 +5,13 @@ from database import get_db, SmartCursor from game_table_model import GameTable from key_model import TableKey from main import db as potato +from flask_login import login_user, logout_user, login_required, current_user tables = Blueprint("tables_api", __name__) @tables.route("/tables", methods=["POST"]) +@login_required def create_table(): allTables = GameTable.query.all() @@ -47,6 +49,7 @@ def update_table(table_id): @tables.route("tables/", methods=["DELETE"]) +@ def delete_table(table_id): table = GameTable.query.get_or_404(table_id) container_managment.delete_file_package(table) diff --git a/src/user_model.py b/src/user_model.py index 0dd92dd..5e5ede5 100644 --- a/src/user_model.py +++ b/src/user_model.py @@ -14,6 +14,7 @@ class User(UserMixin, db.Model): id = db.Column(db.Integer, primary_key=True) password_hash = db.Column(db.Text) user_name = db.Column(db.Text, unique=True) + is_admin = db.Column(db.Boolean) sessions = relationship("Session", back_populates="user") def set_password(self, newPassword: str):