Python：正确使用Click with __main__和__init__

Question

我一直在使用Click package，但我无法获得以下代码片段以便正常使用。

import numpy as np
import click

@click.command()
@click.option('--get_print', default = True)

class CustomClass():

     def __init__(s, get_print):

        s.data = np.random.normal(0, 1, 10)
        s.get_print = get_print

        if s.get_print:
            print("get_print =", s.get_print)

     def print_mean(s):

        print("Printing mean:")

        mean_ = np.mean(s.data)

        if s.get_print:

            print(mean_)

if __name__ == '__main__':
    CustomClass().print_mean()

似乎永远不会调用print_mean函数。如果我注释掉@Click装饰器，并手动设置get_print=True，则print_mean方法被调用就好了。

我怀疑是@Click以某种方式设置__main__所以当我if __name__ == '__main__'时，它只是再次调用Click - 但也许我错了？

在任何一种情况下，当您希望__init__方法继承所有选项时，正确使用Click的方式是什么？if __name__ == '__main__'实际上能够在类中调用函数，其中定义了__init__？

Answer 1

这只是为了说明我在评论中的意思。使用click函数装饰一个函数，并让该函数实例化您的类并执行您需要的任何成员函数。这就是点击文档建议的内容：

import numpy as np
import click

class CustomClass:
    def __init__(self, get_print):
        self.data = np.random.normal(0, 1, 10)
        self.get_print = get_print

        if self.get_print:
            print("get_print =", self.get_print)

    def print_mean(self):
        print("Printing mean:")
        mean_ = np.mean(self.data)
        if self.get_print:
            print(mean_)

@click.command()
@click.option('--get_print', default=True)
def cli(get_print):
    cc = CustomClass(get_print)
    cc.print_mean()

if __name__ == '__main__':
    cli()

...您可能希望get_print成为默认False的标志：

@click.command()
@click.option('--get_print', default=False, is_flag=True, required=False)
def cli(get_print):
    cc = CustomClass(get_print)
    cc.print_mean()