动态定义依赖于先前定义的属性的属性

时间:2018-11-14 22:12:21

标签: python python-attrs

我想要一个对象,该对象表示路径root和用os.path.join(root)构造的任意数量的子目录。我想以self.rootself.path_aself.path_b等形式访问这些路径。除了直接通过self.path_a访问它们之外,我还希望能够遍历他们。不幸的是,下面的方法不允许通过attr.astuple(paths)

对其进行迭代

下面的第一部分代码是我想到的。它可以工作,但对我来说有点不客气。由于这是我第一次使用attrs,因此我想知道是否有更直观/惯用的方式来处理此问题。我花了很长时间才弄清楚如何编写下面的简单类,所以我认为我可能会遗漏一些明显的东西。

我的方法

@attr.s
class Paths(object):
    subdirs = attr.ib()
    root = attr.ib(default=os.getcwd())
    def __attrs_post_init__(self):
        for name in self.subdirs:
            subdir = os.path.join(self.root, name)
            object.__setattr__(self, name, subdir)

    def mkdirs(self):
        """Create `root` and `subdirs` if they don't already exist."""
        if not os.path.isdir(self.root):
            os.mkdir(self.root)
        for subdir in self.subdirs:
            path = self.__getattribute__(subdir)
            if not os.path.isdir(path):
                os.mkdir(path)

输出

>>> p = Paths(subdirs=['a', 'b', 'c'], root='/tmp')
>>> p
Paths(subdirs=['a', 'b', 'c'], root='/tmp')
>>> p.a
'/tmp/a'
>>> p.b
'/tmp/b'
>>> p.c
'/tmp/c'

以下是我的第一次尝试,但是没有用。

尝试失败

@attr.s
class Paths(object):
    root = attr.ib(default=os.getcwd())
    subdir_1= attr.ib(os.path.join(root, 'a'))
    subdir_2= attr.ib(os.path.join(root, 'b'))

输出

------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-31-71f19d55e4c3> in <module>()
    1 @attr.s
----> 2 class Paths(object):
    3     root = attr.ib(default=os.getcwd())
    4     subdir_1= attr.ib(os.path.join(root, 'a'))
    5     subdir_2= attr.ib(os.path.join(root, 'b'))

<ipython-input-31-71f19d55e4c3> in Paths()
    2 class Paths(object):
    3     root = attr.ib(default=os.getcwd())
--> 4     subdir_1= attr.ib(os.path.join(root, 'a'))
    5     subdir_2= attr.ib(os.path.join(root, 'b'))
    6

~/miniconda3/lib/python3.6/posixpath.py in join(a, *p)
    76     will be discarded.  An empty last part will result in a path that
    77     ends with a separator."""
--> 78     a = os.fspath(a)
    79     sep = _get_sep(a)
    80     path = a

TypeError: expected str, bytes or os.PathLike object, not _CountingAttr

2 个答案:

答案 0 :(得分:1)

第一次尝试:您不能只是将随机数据附加到该类上,并希望attrs(在本例中为astuple)能够获取它。 attrs特别尝试避免魔术和猜测,这意味着您必须确实在类上定义属性。

第二次尝试:您不能在类范围内使用属性名称(即在class Paths:内,但不能在方法外使用,因为–正如Python所告诉的–在这一点上,它们仍然是{{1 }}。

我能想到的最优雅的方法是一个通用工厂,该工厂将路径作为参数并构建完整路径:

@attr.s

您可以这样使用:

In [1]: import attr

In [2]: def make_path_factory(path):
   ...:     def path_factory(self):
   ...:         return os.path.join(self.root, path)
   ...:     return attr.Factory(path_factory, takes_self=True)

attrs是attrs,您当然可以进一步定义自己的attr.ib包装器:

In [7]: @attr.s
   ...: class C(object):
   ...:     root = attr.ib()
   ...:     a = attr.ib(make_path_factory("a"))
   ...:     b = attr.ib(make_path_factory("b"))

In [10]: C("/tmp")
Out[10]: C(root='/tmp', a='/tmp/a', b='/tmp/b')

In [11]: attr.astuple(C("/tmp"))
Out[11]: ('/tmp', '/tmp/a', '/tmp/b')

答案 1 :(得分:0)

无法猜测您为什么要以self.paths.path的身份访问。但是,这就是我要做的:

class D(object):
    root = os.getcwd()
    paths = dict()

    def __init__(self, paths=[]):
        self.paths.update({'root': self.root})
        for path in paths:
            self.paths.update({path: os.path.join(self.root, path)})

    def __str__(self):
        return str(self.paths)    

d = D(paths=['static', 'bin', 'source'])
print(d)
print(d.paths['bin'])

输出

{'root': '/home/runner', 'static': '/home/runner/static', 'bin': '/home/runner/bin', 'source': '/home/runner/source'}
/home/runner/bin

您可以使其更加复杂。只是一个例子。希望对您有所帮助。