python - 我是否能够将变量分离到另一个类中但保持用法相同？

Question

我正在尝试重写一个脚本，但我一直致力于使其易于使用。基本上它是一个汇编脚本(就像破坏的相反)，您可以在其中输入大量变量(例如位置、位置是绝对位置还是相对位置、缩放、旋转、可见性、随机偏移等)来创建动画。第一个版本非常不用户友好，所以这次我试图让它从一开始就很好地工作。

我已经想过我希望它如何工作，并且我已经设法保持它的干净，但有一个缺陷。正如您在下面看到的，可以使用像 SetGroup.frame[i].save() 这样的东西，我不想要它(并且我不想检查如果整个类(class)中 name 为 None)。

这是我的代码:

class SetGroup(object):

    def __init__(self, name=None, _frame_only=False):

        if name is None and not _frame_only:
            raise TypeError('name of group must be provided')

        self.selection = None
        self.origin = None

        self.start = None
        self.end = None
        self.offset = 0
        self.distance = None
        self.random = 0

        self.location = None
        self.rotation = None
        self.scale = None
        self.visibility = None

        if not _frame_only:
            self.frame = defaultdict(lambda: SetGroup(_frame_only=True))

    def save(self):
        self.load()
        #do a bit of error checking here
        self.data[self.name] = {'ObjectSelection': self.selection,
                                'ObjectOrigin': self.origin,
                                'FrameStart': self.start,
                                'FrameEnd': self.end,
                                'FrameOffset': self.offset,
                                'FrameDistance': self.distance,
                                'FrameRandom': self.random,
                                'StartLocation': self.location,
                                'StartRotation': self.rotation,
                                'StartScale': self.scale,
                                'StartVisibility': self.visibility,
                                'ExtraFrames': self.frame}
        pm.fileInfo['AssemblyScript'] = StoreData().save(self.data)

    def load(self):
        try:
            self.data = StoreData().load(pm.fileInfo['AssemblyScript'])
        except KeyError:
            pm.fileInfo['AssemblyScript'] = StoreData().save({})

我希望它的工作方式是这样的:

a = SetGroup('test')
a.location = ((0, 0, 0), True)
a.start = 0
a.end = 10
a.frame[5].location = ((10, 10, 10), False)
a.frame[5].scale = ((2, 1, 1), True)
a.save()

除非有人能想出一种方法使其更易于使用，否则我将如何分离位置，旋转，缩放，和 visibility 到另一个类中并再次将它们链接起来，以便它们仍然在类的核心级别上工作，而且也适用于框架字典？

编辑 - 使其达到基本水平:

class _MovementInfo(object):
    def __init__(self, location=None, rotation=None, scale=None, visibility=None):
        self.location = location
        self.rotation = rotation
        self.scale = scale
        self.visibility = visibility

    def __repr__(self):
        return '_MovementInfo(location={x.location}, rotation={x.rotation}, scale={x.scale}, visibility={x.visibility}'.format(x=self)

然后我在主类中使用它来合并字典:

self.__dict__.update({k: v for k, v in _MovementInfo().__dict__.iteritems() if '__' not in k})
self.frame = defaultdict(_MovementInfo)

Answer 1

我会像这样更改代码:

class SetGroup(_Movement):
    def __init__(self, name=None):
        if name is None:
            # ...
        super().__init__()
        # ...
        self.random = 0  # __init__ should end here

    # ...

但是您应该检查所有帧中的所有 _MovementInfo 是否都是 _MovementInfo 或从它们继承(要检查这一点:isinstance(x, _MovementInfo))，但不是 SetGroup 的(检查一下:不是 isinstance(x, SetGroup))。

super() 是 super(SetGroup, self) 的缩写(对于 python2，您必须使用最后一个选项)，并且基本上是一个包含所有内容的对象基类具有该属性，并允许您调用修改调用它的类的方法。

或者在代码中:

class A(object):
    def __init__(self, y):
        self.x = 2
        self.y = y

class B(A):
    def __init__(self, y, z):
        super().__init__(y)  # equivalent to: A.__init__(self, y)
        self.z = z

b = B(3, 4)
# b's x is 2, b's y is 3 (both set by A.__init__, the last one was passed by B), and b's z is 4 (set by B.__init__)

我希望这有帮助，

代号Lambda