使用单例作为计数器

Question

我有一个自动化测试，它使用为文件夹创建屏幕截图的功能。该函数被多个截图实例调用。在每次测试运行时，都会创建一个新文件夹，所以我不关心计数器重置。为了反映这些截图的拍摄顺序，我必须想出可以按顺序排序的名称。这是我的解决方案：

def make_screenshot_file(file_name):
    order = Counter().count
    test_suites_path = _make_job_directory()
    return make_writable_file(os.path.join(test_suites_path,'screenshot',file_name % order))


class Counter():
    __counter_instance = None

    def __init__(self):
        if Counter.__counter_instance is None:
            self.count = 1
            Counter.__counter_instance = self
        else: 
            Counter.__counter_instance.count += 1
            self.count =  Counter.__counter_instance.count

它对我来说很好用。但我一直认为应该有一种更简单的方法来解决这个问题。在那儿？如果单例是唯一的方法，我的代码可以以任何方式优化吗？

Answer 1

您在这里尝试做的是模拟一个全局变量。

没有充分的理由这样做。如果您真的想要一个全局变量，请将其显式设为全局变量。

您可以创建一个简单的Counter 类，每次访问它时将count 递增1，然后创建它的全局实例。但是正如 DSM 在评论中解释的那样，标准库已经在itertools.count 中免费为您提供了类似的东西。

所以：

import itertools

_counter = itertools.count()
def make_screenshot_file(file_name):
    order = next(_counter)
    test_suites_path = _make_job_directory()
    return make_writable_file(os.path.join(test_suites_path,'screenshot',file_name % order))

我不确定您为什么如此担心这会占用多少存储空间或时间，因为我无法想象任何程序可能对单个对象使用 8 个字节还是 800 个字节可能很重要你永远不会有超过一个 or，或者当你只访问它几次时，它需要 3ns 或 3us 来访问它。

但是如果您担心，正如您从the source 中看到的那样，count 是用 C 实现的，它非常节省内存，如果您不做任何花哨的事情, 归结为一个PyNumber_Add 来生成每个数字，这比解释几行代码要少很多。

---

既然您问了，下面是您可以通过使用_count 类属性而不是__counter_instance 类属性从根本上简化现有代码的方法：

class Counter():
    _count = 0
    def count(self):
        Counter._count += 1
        return Counter.count

当然，现在您必须使用 Counter().count() 而不仅仅是 Counter().count，但如果重要的话，您可以使用 @property 轻松解决这个问题。

值得指出的是，使用经典类而不是新式类（通过在括号内不传递任何内容）是一个非常糟糕的主意，如果你确实想要一个经典类，你应该去掉括号，大多数 Python 程序员会将名称 Counter 与类 collections.Counter 相关联，并且没有理由 count 不能是 @classmethod 或 @staticmethod... 这正是安德鲁 T. 的回答。正如他所指出的那样，这比您正在做的事情要简单得多，而且或多或少是 Pythonic。

但实际上，所有这一切并没有比将 _count 设为模块级全局并添加一个模块级 count() 函数来递增和返回它更好。

Answer 2

为什么不这样做

order = time.time()

或者做类似的事情

import glob #glob is used for unix like path expansion
order = len(glob.glob(os.path.join(test_suites_path,"screenshot","%s*"%filename))

Answer 3

使用静态方法和变量。不是很pythonic，但更简单。

def make_screenshot_file(file_name):
    order = Counter.count() #Note the move of the parens
    test_suites_path = _make_job_directory()
    return make_writable_file(os.path.join(test_suites_path,'screenshot',file_name % order))

class Counter():
  count_n = 0

  @staticmethod
  def count():
    Counter.count_n += 1
    return Counter.count_n


print Counter.count()
print Counter.count()
print Counter.count()
print Counter.count()
print Counter.count()


atarzwell@freeman:~/src$ python so.py
1
2
3
4
5

Answer 4

好吧，你可以使用这个解决方案，只要确保你从不初始化 order kwarg！

函数中的可变 Kwargs 类似于类全局变量。并且该值不会在调用之间重置为默认值，正如您一开始可能会想到的那样！

def make_screenshot_file(file_name , order=[0]):
    order[0] = order[0] + 1
    test_suites_path = _make_job_directory()
    return make_writable_file(os.path.join(test_suites_path,'screenshot',file_name % order[0]))