瀏覽代碼

DB skeleton

Kirk Trombley 5 年之前
父節點
當前提交
9bfdfa281e
共有 2 個文件被更改,包括 27 次插入4 次删除
  1. 20 4
      server/app.py
  2. 7 0
      server/db.py

+ 20 - 4
server/app.py

@@ -1,12 +1,28 @@
 from flask import Flask
+import toml
 
+from db import db
 from forms import forms
 from game_api import game
 
-app = Flask(__name__)
-app.register_blueprint(forms)
-app.register_blueprint(game, url_prefix="/game")
+
+def init_app():
+    app = Flask(__name__)
+
+    with open("secrets.toml") as infile:
+        secrets = toml.load(infile)
+    app.secret_key = secrets["secret_key"]
+    app.config["SQLALCHEMY_DATABASE_URI"] = secrets["db_uri"]
+    app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
+
+    app.register_blueprint(forms)
+    app.register_blueprint(game, url_prefix="/game")
+
+    db.init_app(app)
+    db.create_all(app=app)
+
+    return app
 
 
 if __name__ == "__main__":
-    app.run("0.0.0.0", 5000)
+    init_app().run("0.0.0.0", 5000)

+ 7 - 0
server/db.py

@@ -0,0 +1,7 @@
+from flask_sqlalchemy import SQLAlchemy
+
+db = SQLAlchemy()
+
+class Game(db.Model):
+    game_id = db.Column(db.String, primary_key=True)
+    timer = db.Column(db.Integer)