python - 覆盖 __new__ 以返回初始化对象时如何使用上下文

Question

我已经实现了一个工厂类，它在 __new__ 方法中返回一个初始化的类

class MyFactory(object):
    def __new__(self, my_obj, *args, **kwargs):
        if my_obj == ...:
            return A(*args, **kwargs)
        else:
            return B(*args, **kwargs)

A 和B 类都扩展了一个支持上下文的基类MyBase()

class MyBase(object):
    def __enter__(self):
        ...

    def __exit__(self, exc_type, exc_val, exc_tb):
        ...

我正在尝试像这样使用这个设置

with MyFactory(my_obj, ...) as something:
   something.do_something()

但是显然会抛出错误，因为 python 期望 __enter__、__exit__ 对出现在 MyFactory 类上

我应该做哪些更改才能使上述用法成为可能？

编辑:

我承认，这实际上是正确的做法。我的错误是由于未在 __enter__ 方法中返回 self :facepalm:

Heres the working code for those of you interested

Answer 1

目前 MyFactory 仅扩展 oject

class MyFactory(object):

我认为您的 MyFactory 对象需要使用以下语法扩展 MyBase:

class MyFactory(MyBase):
    ...

这应该启用 __exit__ 和 __enter__ 方法的父类(super class)实例化，而无需在子类中再次明确描述它们。

它有点混搭，但由于您要从 __new__ 调用返回对象，所以它可能是一个公平的组合。