在Python项目开发中，我们经常会从环境变量中读取配置参数，比如服务端口、重试次数等数值类型的配置。但环境变量默认返回的都是字符串类型，直接转换为整数时如果处理不当，很容易出现值缺失、内容非数字等异常，同时缺少类型注解也会降低代码的可维护性。掌握安全转换的方法并添加合适的类型注解，能有效提升代码的健壮性。
常见的问题场景
直接使用int(os.getenv("PORT"))的方式存在两个明显问题：一是如果环境变量不存在，os.getenv会返回None，调用int(None)会直接抛出TypeError；二是如果环境变量的值是非数字字符串，比如"abc"，转换时会抛出ValueError。另外没有类型注解的话，其他开发者调用函数时无法直观知道返回值的类型。
安全转换的核心实现
我们可以先封装一个通用的转换函数，处理异常场景，同时添加类型注解明确输入输出类型。这里使用Optional表示环境变量可能不存在的情况，返回类型标注为Optional[int]，表示可能返回整数或者None。
import os
from typing import Optional
def get_env_int(env_name: str, default: Optional[int] = None) -> Optional[int]:
"""
安全从环境变量获取整数配置
:param env_name: 环境变量名称
:param default: 环境变量不存在时的默认返回值
:return: 转换后的整数或者默认值
"""
env_value = os.getenv(env_name)
# 环境变量不存在时返回默认值
if env_value is None:
return default
try:
# 尝试转换为整数
return int(env_value)
except ValueError:
# 转换失败时返回默认值
return default
类型注解的详细说明
上述函数中的类型注解含义如下：
env_name: str：参数env_name的类型是字符串，表示环境变量名称default: Optional[int] = None：参数default的类型是可空整数，默认值为None，表示如果环境变量不存在就返回这个值-> Optional[int]：函数返回值的类型是可选整数，即可能是int类型或者None
如果需要强制要求环境变量必须存在，不能返回None，可以调整类型注解，去掉Optional，同时在函数内部处理缺失场景时直接抛出异常：
import os
from typing import NoReturn
def get_env_int_required(env_name: str) -> int:
"""
强制从环境变量获取整数配置，不存在或非数字时抛出异常
:param env_name: 环境变量名称
:return: 转换后的整数
"""
env_value = os.getenv(env_name)
if env_value is None:
raise EnvironmentError(f"环境变量 {env_name} 不存在")
try:
return int(env_value)
except ValueError:
raise ValueError(f"环境变量 {env_name} 的值 {env_value} 无法转换为整数")
使用示例
在实际使用中，可以根据是否需要默认值选择不同的函数：
# 读取端口配置，不存在时使用默认端口8080
port = get_env_int("PORT", default=8080)
print(f"服务端口: {port}")
# 读取必须存在的重试次数配置
try:
retry_count = get_env_int_required("RETRY_COUNT")
print(f"重试次数: {retry_count}")
except (EnvironmentError, ValueError) as e:
print(f"配置读取失败: {e}")
注意事项
需要注意os.getenv返回的是字符串或者None，不要直接对返回值做isdigit()判断再转换，因为负整数比如"-100"的isdigit()返回False，但其实是合法的整数。直接尝试转换并捕获异常的方式能覆盖更多合法场景。另外类型注解只是静态类型提示，不会在运行时强制校验类型，如果需要运行时校验可以结合pydantic等库使用。