From 8f21d183e6270101f46636dafb469800ce9fbd79 Mon Sep 17 00:00:00 2001 From: Benjamyn Love Date: Fri, 10 Mar 2023 08:58:23 +1100 Subject: [PATCH] Added other models --- priceybot2/models.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/priceybot2/models.py b/priceybot2/models.py index 8c20860..f5954a3 100644 --- a/priceybot2/models.py +++ b/priceybot2/models.py @@ -1,4 +1,5 @@ from flask_login import UserMixin +from datetime import datetime from . import db class User(UserMixin, db.Model): @@ -9,3 +10,19 @@ class User(UserMixin, db.Model): google_id = db.Column(db.String(30), unique=True) administrator = db.Column(db.Boolean, default=False) +class Quote(db.Model): + id = db.Column(db.Integer, primary_key=True) + quote = db.Column(db.String(2000)) + date_added = db.Column(db.DateTime(), default=datetime.now()) + date_last_used = db.Column(db.DateTime()) + times_used = db.Column(db.Integer, default=0) + +class Credit(db.Model): + id = db.Column(db.Integer, primary_key=True) + amount = db.Column(db.Integer) + user_id = db.Column(db.Integer, db.ForeignKey('user.id')) + +class Transactions(db.Model): + id = db.Column(db.Integer, primary_key=True) + user_id = db.Column(db.Integer, db.ForeignKey('user.id')) + quote_id = db.Column(db.Integer, db.ForeignKey('quote.id')) \ No newline at end of file