WebAdminPanel/database.py

from peewee import DateTimeField, ForeignKeyField, Model, CharField, BigIntegerField, DateField
from playhouse.db_url import connect

from datetime import date, datetime
from config import db_url 


db = connect(db_url)

class WebUser(Model):
    username = CharField()
    password_hash = CharField()

    class Meta:
        db_table = "webusers"
        database = db

class Member(Model):
    user_id    = BigIntegerField()
    first_name = CharField()
    username  = CharField(null=True)
    
    role = CharField(default="member")

    warns = BigIntegerField(default=0)
    
    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()) 
         
    class Meta:
        db_table = "messages"
        database = db

def build_database():
    db.create_tables([WebUser,Message])