时间戳

首页 > 解决方案 > 在 Python 3+ 中以编程方式向类添加方法

python - 在 Python 3+ 中以编程方式向类添加方法

问题描述

我有一个为 API 提供简单接口的类。我想为每个路由在该类上定义一个方法。

由于大多数路由是相同的，因此可以将许多功能分解为更通用的函数，许多路由只是该函数的部分应用版本

class MyAPIWrapper:

    def _get_results(self, route, **params):
        # Do the important stuff here
        results = ...
        return results

    def get_customers(self):
        return self._get_results(self, 'customers')

    def get_transactions(self):
        return self._get_results(self, 'transactions')

    # etc, etc, etc

但是，很明显，这仍然会导致类定义中出现大量样板。

一种替代方法是添加一个新方法，以编程方式添加每个路由的方法：

import functools

class MyAPIWrapper:

    def __init__(self):
        self._init_route_methods()

    def _get_results(self, route, **params):
        # Do the important stuff here
        results = ...
        return results

    def _init_route_methods(self):
        for route in ['customers', 'transactions', ...]:
            route_fn = functools.partial(self. _get_results, route)
            setattr(self, f'get_{route}', route_fn)

这样做的好处是减少了样板代码的数量，并且可以轻松添加/删除路由。但是，在初始化时添加方法对我来说有点不雅。

有没有更好和/或更惯用的方法来做到这一点？

标签： pythonpython-3.xoop

解决方案

您可能会惊讶于这会成功：

class MyAPIWrapper:
    def _get_results(self, route, **params):
        # Do the important stuff here
        return route

    for route in ['customers', 'transactions']:
        exec("""
    def get_{0}(self):
        return self._get_results('{0}')
    """.strip().format(route))

MyAPIWrapper().get_customers()    # return customers
MyAPIWrapper().get_transactions() # return transactions

优点

  • 良好的可读性
  • 最少的代码更改

缺点

请注意，这exec比 有一点开销setattr(MyAPIWrapper,'get_%s'%route, ...)，这仅在您要在循环中创建数百万个方法时才重要。

如果您想对许多不同的 APIWrapper 类执行相同的操作，请考虑改用类装饰器。

推荐阅读