1

我有一个创建连接池的数据库连接类。现在随着应用程序的增长并且我正在添加不同类型的数据库编写器，我想将数据库连接移动到一个单独的类并从它继承。到目前为止，我有这个：

class ServiceDB:
    @classmethod
    async def init(cls, settings):
        self = ServiceDB()
        self.pool = await asyncpg.create_pool(
            database=settings["POSTGRES_DB"],
            user=settings["POSTGRES_USER"],
            password=settings["POSTGRES_PASSWORD"],
            host=settings["DB_HOST"],
            port=settings["DB_PORT"],
        )
        return self



class ChildWriter(ServiceDB):
    async def write_db(self, query):
        # Write to specific table
        pass



if __name__ == "__main__":
    settings = {'info': "some connection settings"}
    query = "SELECT * FROM 'table'"
    connection = await ChildWriter().init(settings)
    await connection.write_db(msg, query)

当我运行它时，我得到AttributeError: 'ServiceDB' object has no attribute 'write_db'. 如何正确扩展ServiceDB该write_db方法？

4

1 回答 1

1

类方法接收“当前类”作为第一个参数。实例化 this cls，而不是固定的基类。

class ServiceDB:
    @classmethod
    async def init(cls, settings):
        self = cls()  # cls is the *current* class, not just ServiceDB
        self.pool = await asyncpg.create_pool(
            database=settings["POSTGRES_DB"],
            user=settings["POSTGRES_USER"],
            password=settings["POSTGRES_PASSWORD"],
            host=settings["DB_HOST"],
            port=settings["DB_PORT"],
        )
        return self

请注意，理想情况下，所有属性都是通过__init__而不是单独的classmethod构造函数设置的。单独的构造函数应该只传递外部构造的任何属性。

class ServiceDB:
    def __init__(self, pool):
        self.pool = pool

    @classmethod
    async def init(cls, settings, **kwargs):
        pool = await asyncpg.create_pool(
            database=settings["POSTGRES_DB"],
            user=settings["POSTGRES_USER"],
            password=settings["POSTGRES_PASSWORD"],
            host=settings["DB_HOST"],
            port=settings["DB_PORT"],
        )
        return cls(pool=pool, **kwargs)

class ChildWriter(ServiceDB):
    async def write_db(self, query): ...

if __name__ == "__main__":
    settings = {'info': "some connection settings"}
    query = "SELECT * FROM 'table'"
    # call classmethod on class   V
    connection = await ChildWriter.init(settings)
    await connection.write_db(msg, query)
于 2020-07-23T12:16:11.270 回答