diff --git a/database.py b/database.py index 8de9238..2f5742a 100644 --- a/database.py +++ b/database.py @@ -1,12 +1,13 @@ -from peewee import DateTimeField, ForeignKeyField, Model, CharField, BigIntegerField, DateField +from peewee import Model +from peewee import BigIntegerField, CharField, DateTimeField, DateField, ForeignKeyField from playhouse.db_url import connect from datetime import date, datetime -from config import db_url - +from config import db_url db = connect(db_url) + class WebUser(Model): username = CharField() password_hash = CharField() @@ -15,29 +16,31 @@ class WebUser(Model): db_table = "webusers" database = db + class Member(Model): - user_id = BigIntegerField() + user_id = BigIntegerField() first_name = CharField() - username = CharField(null=True) + username = CharField(null=True) role = CharField(default="member") warns = BigIntegerField(default=0) - joined = DateField(default=date.today()) - + joined = DateField(default=date.today()) class Meta: db_table = "members" database = db + class Message(Model): - user = ForeignKeyField(Member,lazy_load=True) - timestamp = DateTimeField(default=datetime.now()) + user = ForeignKeyField(Member, lazy_load=True) + timestamp = DateTimeField(default=datetime.now()) class Meta: db_table = "messages" database = db + def build_database(): - db.create_tables([WebUser,Message]) + db.create_tables([WebUser, Message])