diff --git a/project_amber/app.py b/project_amber/app.py index 5bad765..a4b6c55 100644 --- a/project_amber/app.py +++ b/project_amber/app.py @@ -6,6 +6,7 @@ app = Flask(__name__) app.config["SQLALCHEMY_DATABASE_URI"] = config["database"] db = SQLAlchemy(app) + db.create_all() # create all tables on first run @app.route("/") diff --git a/project_amber/models/auth.py b/project_amber/models/auth.py new file mode 100644 index 0000000..31e48eb --- /dev/null +++ b/project_amber/models/auth.py @@ -0,0 +1,19 @@ +from project_amber.app import db + +class User(db.Model): + id = db.Column(db.Integer, primary_key=True) + name = db.Column(db.String(128), unique=True, nullable=False) + password = db.Column(db.String(256)) + def __repr__(self): + return "" % self.id, self.name + +class Session(db.Model): + """ + Holds auth session details (auth token, the time of login, etc). + """ + token = db.Column(db.String(256), primary_key=True) + user = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False) + loginTime = db.Column(db.Integer, nullable=False) + def __repr__(self): + return "" % \ + self.token, self.user, self.loginTime