python - 基于构造函数参数的类的条件访问

Question

from folder.test.test1 import Test1
from folder.test.test2 import Test2
from other_folder import Test3

class Sample(Test1, Test2, Test3):
    def __init__(self, version):
        Test3.__init__(self)

        if version == 'gen1':
            self.__class__ = Test1
            Test1.__init__()
        elif version == 'gen2':
            self.__class__ = Test2
            Test2.__init__()

    def login(self, ip):
        pass

if __name__ == '__main__':
    ob = Sample(version='gen2')
    ob.login('192.168.1.100')

我收到此错误:

AttributeError: 'Test2' object has no attribute 'login'

尽管我有一个 Sample 类对象，但无法访问类 Sample 的 login() 方法。

我希望对象能够访问 Sample、Test2 和 Test3 方法(不是 Test1 的方法，因为版本是'gen2')。如果版本为 'gen1'，则 Sample 对象只能访问 Sample、Test1 和Test3 方法，而不是 Test2 的方法。

Answer 1

一种解决方案是将其分为 2 个类(如果可以的话):

class BaseSample(Test3):
    def login(self, ip):
        pass

class SampleGen1(Test1, BaseSample):
    pass

class SampleGen2(Test2, BaseSample):
    pass

然后在代码中使用适当的类:

if version == 'gen1':
    ob = SampleGen1()
elif version == 'gen2':
    ob = SampleGen2()
ob.login('192.168.1.100')

这对你有用吗？