Pyright为何不允许非字面量集合直接参与str.join？赋值新变量为何可行？

问题场景 #

以下代码会触发Pyright错误：

1. Operator "|" not supported for types "set[str]" and "set[str]"
   ...... Operator "|" not supported for types "set[str]" and "set[str]" when expected type is "Iterable[LiteralString]"

错误代码示例：

def update_no_proxy():
    def _normalize_url(url: str) -> str:
        return urlparse(url).netloc

    old_no_proxy_urls = set(os.getenv('NO_PROXY', '').split(','))
    no_proxy_new_urls = {_normalize_url(url) for url in NO_PROXY_URLS}
    no_proxy_new_urls = {*no_proxy_new_urls, get_local_ip_address()}
    no_proxy_urls: str = ','.join(old_no_proxy_urls | no_proxy_new_urls)
    for env_name in NO_PROXY_ENV_NAMES:
        os.environ[env_name] = no_proxy_urls.strip(', ')

而将集合合并结果先赋值给新变量all_no_proxy_urls后，代码不再报错：

def update_no_proxy():
    def _normalize_url(url: str) -> str:
        return urlparse(url).netloc

    old_no_proxy_urls = set(os.getenv('NO_PROXY', '').split(','))
    no_proxy_new_urls = {_normalize_url(url) for url in NO_PROXY_URLS}
    no_proxy_new_urls = {*no_proxy_new_urls, get_local_ip_address()}
    all_no_proxy_urls = old_no_proxy_urls | no_proxy_new_urls
    no_proxy_urls = ','.join(all_no_proxy_urls)
    for env_name in NO_PROXY_ENV_NAMES:
        os.environ[env_name] = no_proxy_urls.strip(', ')

原因解析 #

这是因为Pyright的类型推断在两种场景下的上下文约束不一样：

• 直接把集合合并表达式old_no_proxy_urls | no_proxy_new_urls传给','.join()时，Pyright会根据join期望的Iterable[LiteralString]类型反向约束这个表达式的类型。虽然集合的|操作返回的是set[str]，但Pyright这里没正确处理set[str]和Iterable[LiteralString]的兼容性，反而误判成|操作符不支持当前的类型组合。
• 先把合并结果赋值给all_no_proxy_urls时，Pyright会先单独推断这个变量的类型是set[str]，而set[str]本身就是合法的字符串可迭代类型，完全符合join的参数要求，这时再把变量传给join，类型检查自然就通过了。

简单说，中间变量的引入打破了join方法对表达式的反向类型约束，让Pyright能正确识别集合合并结果的类型，从而通过类型检查。

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