WebAdminPanel/database.py

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

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])