如何定义属性而不重复__init__中的setter方法?

时间:2016-12-15 14:50:33

标签: python

我试图定义一个类,它通过合成包含faker.Faker()random.Random()的实例,它们各自的seed具有相同的值:

import faker
import random

class DistanceRule(object):
    def __init__(self, seed=None):
        self.fake = faker.Faker()
        self.random = random.Random()
        self._seed = seed
        self.seed_fake_and_random()

    @property
    def seed(self):
        return self._seed

    @seed.setter
    def seed(self, seed):
        self._seed = seed
        self.seed_fake_and_random()

    def seed_fake_and_random(self):
        """Seed the instance's faker.Faker() and random.Random() with a
        single seed value.
        """
        self.fake.seed(self._seed)
        self.random.seed(self._seed)

以便运行以下脚本

if __name__ == "__main__":
    d1 = DistanceRule(seed=5)
    print d1.random.randint(1, 10), d1.fake.name()
    d2 = DistanceRule(seed=12)
    d2.seed = 5
    print d2.random.randint(1, 10), d2.fake.name()

产生

7 Natalie Pope
7 Natalie Pope

这捕获了所需的行为,即设置seed属性也会重新播种实例的随机数据生成器。让我烦恼的唯一事情是__init__@seed.setter方法中存在代码重复。还有更干的方法吗?

1 个答案:

答案 0 :(得分:2)

由于您重新定义了setter,您可以在实例初始化时调用它:

def __init__(self, seed=None):
    self.fake = faker.Faker()
    self.random = random.Random()
    self.seed = seed