from datetime import datetime

from peewee import *
from peewee import Model

db = MySQLDatabase(
    'message', host='127.0.0.1', port=32768, user='root', password='root')


class BaseModel(Model):
    add_time = DateTimeField(default=datetime.now, verbose_name='建立時間')

    class Meta:
        pass


class Supplier(BaseModel):
    name = CharField(max_length=100, verbose_name='名稱', index=True)
    address = CharField(max_length=100, verbose_name='聯繫地址')
    phone = CharField(max_length=11, verbose_name='聯繫方式')

    class Meta:
        database = db
        table_name = 'supplier'


class Goods(BaseModel):
    supplier = ForeignKeyField(Supplier, verbose_name='商家', backref='goods')
    name = CharField(max_length=100, verbose_name='商品名稱', index=True)
    click_num = IntegerField(default=0, verbose_name='點擊數')
    goods_num = IntegerField(default=0, verbose_name='庫存量')
    price = FloatField(default=0.0, verbose_name='價格')
    brief = TextField(verbose_name='商品簡介')

    class Meta:
        database = db
        table_name = 'goods'


def init_table():
    db.create_tables([Goods, Supplier])


if __name__ == "__main__":
    init_table()