fakeowl1
/
WebAdminPanel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modified: database.py

This commit is contained in:
hok7z 2023-04-16 16:11:08 +03:00
parent 6121d5a3ff
commit 652ec5a242
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
database.py
View File

@ -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 playhouse.db_url import connect
from datetime import date, datetime from datetime import date, datetime
from config import db_url from config import db_url
db = connect(db_url) db = connect(db_url)
class WebUser(Model): class WebUser(Model):
username = CharField() username = CharField()
password_hash = CharField() password_hash = CharField()
@ -15,6 +16,7 @@ class WebUser(Model):
db_table = "webusers" db_table = "webusers"
database = db database = db
class Member(Model): class Member(Model):
user_id = BigIntegerField() user_id = BigIntegerField()
first_name = CharField() first_name = CharField()
@ -26,11 +28,11 @@ class Member(Model):
joined = DateField(default=date.today()) joined = DateField(default=date.today())
class Meta: class Meta:
db_table = "members" db_table = "members"
database = db database = db
class Message(Model): class Message(Model):
user = ForeignKeyField(Member, lazy_load=True) user = ForeignKeyField(Member, lazy_load=True)
timestamp = DateTimeField(default=datetime.now()) timestamp = DateTimeField(default=datetime.now())
@ -39,5 +41,6 @@ class Message(Model):
db_table = "messages" db_table = "messages"
database = db database = db
def build_database(): def build_database():
db.create_tables([WebUser, Message]) db.create_tables([WebUser, Message])