from peewee import *

database = MySQLDatabase('db_dev_12280', user='money111', password='money_770', host='8.140.162.237', port=3307)


# 数据模型
class BaseModel(Model):
    class Meta:
        database = database


class DatabaseConnection:
    def __enter__(self):
        # 建立数据库连接
        database.connect()
        return database

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 关闭数据库连接
        database.close()


# 数据表结构
class Test(BaseModel):
    id = AutoField()
    name = CharField(unique=True)
    age = IntegerField()


# demo
def demo():
    with DatabaseConnection():
        # 查询所有
        for user in Test.select():
            print(f"ID: {user.id}, Name: {user.name}, Age: {user.age}")
    print("查询完成")