Python函数能否同时兼具生成器与非生成器两种行为？

核心原因 #

Python里只要函数定义中包含yield关键字，它就会被认定为生成器函数。生成器函数的特性是：调用时不会立即执行函数体内的代码，而是直接返回一个生成器对象。只有当你对这个生成器进行迭代操作（比如for循环、next()、list()等）时，函数体才会开始执行。

你的代码里，虽然save=True的分支完全没有执行yield，但函数定义里存在yield，所以调用时直接返回了生成器对象，函数体里的print('hello')和文件写入逻辑都没触发。

修复方案 #

最合理的方式是拆分逻辑，把生成器部分抽成内部函数，让主函数不再包含yield，这样调用主函数时会立即执行代码：

def encode_file(source, save=False, destination=None):
    # encode the contents of an input file 3 bytes at a time
    print('hello')
    with open(source, 'rb') as infile:
        def _encode_generator():
            while (bytes_to_encode := infile.read(3)):
                l = len(bytes_to_encode)
                if l < 3:
                    bytes_to_encode += (b'\x00' * (3 - l))  # pad bits if short
                yield encode(bytes_to_encode)
        
        # 写入文件分支
        if save:
            if not destination:
                raise ValueError("destination must be provided when save=True")
            print(f'saving to file {destination}')
            with open(destination, 'wb') as outfile:
                for encoded_bytes in _encode_generator():
                    outfile.write(encoded_bytes)
            return
        # 返回生成器分支
        else:
            return _encode_generator()

额外注意 #

你原代码里存在一个逻辑不一致：save=True分支直接写入了原始的bytes_to_encode，而生成器分支yield的是encode(bytes_to_encode)的结果。上面的修复方案统一了逻辑，文件写入时也会写入编码后的字节，确保两种行为的一致性。

另外，如果坚持不重构函数，也可以在调用时手动迭代生成器来触发执行，但这种方式不直观，容易出错：

gen = encode_file('file.bin', save=True, destination='output.base64')
list(gen)  # 迭代生成器，触发函数体执行

内容的提问来源于stack exchange，提问作者First User