调用类方法但只返回生成器

Question

我正在尝试调试类中的方法。 所以在迭代控制台中，我使用下面的行导入了我的类。

从 email_reader 导入 MyClass

然后我想我只需要像下面这样设置我的类的一个实例。

ma_cls = MyClass('MyFolder')

然后我可以像下面一样调用函数 get_email，

ma_cls.get_email（文件名，date_from，date_to）

当我运行它时，断点永远不会被击中，返回的所有内容都在下面。

出[353]：

我究竟做错了什么？

class MyClass:
  def __init__(self,
             outlook_folder: str):
    self.__outlook_folder = outlook_folder

  def get_email(self,
                           csv_filename: str,
                           emails_since: dt.datetime,
                           emails_to: dt.datetime = None,
                           unread_only: bool = False
                           ):

    breakpoint()

    # some logic



 if __name__ == "__main__":

 breakpoint()

 sse = MyClass("MyFolder")

for wt in sse.get_email(
        'some_file_name.CSV',
        emails_since=dt.datetime(2020, 2, 4),
        emails_to=dt.datetime(2020, 2, 5),
        unread_only=False):
    pass

Answer 1

代码块

if __name__ == "__main__":

仅在直接运行脚本时执行（当您直接执行包含MyClass的脚本时）。 如果您正在导入上述文件，则上述if-statement所有内容都将被忽略。