如何优雅处理单个整数、整数列表与整数生成器参数？

方案1：利用迭代器协议，按需处理（适合遍历场景） #

针对render这类仅需遍历的场景，无需强制转换为列表，直接返回可迭代对象即可，节省内存开销：

import types

def normalize_iterable(obj):
    if isinstance(obj, (list, types.GeneratorType)):
        return obj
    # 单个元素包装为单元素可迭代对象
    return (obj,)

def render(ids):
    for id in normalize_iterable(ids):
        ...

优势：避免生成器转列表的内存消耗，仅在必要时包装单个元素，逻辑简洁高效。

方案2：通用可迭代对象处理（支持更多类型） #

扩展支持所有可迭代类型（如元组、集合），同时排除字符串避免误判：

import collections.abc
import types

def normalize_iterable(obj, force_list=False):
    # 排除字符串，防止单个字符串被拆分为字符迭代
    if isinstance(obj, str):
        return [obj] if force_list else (obj,)
    # 判断是否为可迭代对象
    if isinstance(obj, collections.abc.Iterable):
        return list(obj) if force_list else obj
    # 处理单个元素
    return [obj] if force_list else (obj,)

def render(ids):
    for id in normalize_iterable(ids):
        ...

def foo(ids):
    ids = normalize_iterable(ids, force_list=True)
    return ids[0]

优势：兼容性更强，覆盖更多合法输入类型，force_list参数灵活满足索引访问需求。

方案3：用singledispatch实现多态处理（扩展性更佳） #

借助Python的单分派装饰器，按参数类型分发处理逻辑，代码可读性和扩展性更好：

from functools import singledispatch
import types

@singledispatch
def normalize(obj, force_list=False):
    # 默认处理单个元素
    return [obj] if force_list else (obj,)

@normalize.register(list)
def _(obj, force_list=False):
    return obj if not force_list else list(obj)

@normalize.register(types.GeneratorType)
def _(obj, force_list=False):
    return obj if not force_list else list(obj)

# 可快速扩展支持元组等类型
@normalize.register(tuple)
def _(obj, force_list=False):
    return obj if not force_list else list(obj)

def render(ids):
    for id in normalize(ids):
        ...

def foo(ids):
    ids = normalize(ids, force_list=True)
    return ids[0]

优势：新增支持类型时只需添加对应注册函数，逻辑清晰，符合Pythonic的多态设计。

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