重写peewee上的select方法

from peewee import *

# 定义数据库模型
class User(Model):
    name = CharField()
    age = IntegerField()

    class Meta:
        database = MySQLDatabase('mydatabase', user='myuser', password='mypassword')

    @classmethod
    def custom_select(cls, query):
        # 在select方法中添加额外的查询条件
        query = query.where(cls.age >= 18)

        # 扩展查询结果，添加计算字段
        query = query.select(cls.name, cls.age, (cls.age + 1).alias('age_next_year'))

        # 优化查询性能，添加索引
        query = query.index(cls.age)

        # 实现自定义逻辑，添加聚合计算
        query = query.select(fn.COUNT(cls.id).alias('count'))

        return query

# 使用自定义的select方法进行查询
users = User.custom_select(User.select()).execute()

# 打印查询结果
for user in users:
    print(user.name, user.age, user.age_next_year, user.count)