91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
# 第 2-3 课完整示例：异常处理
|
|
#
|
|
# 本程序通过固定测试数据演示异常处理，不会要求用户输入。
|
|
# 文件示例只会读取当前课程目录中的 example_data 文件夹。
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
LESSON_DIR = Path(__file__).parent
|
|
DATA_DIR = LESSON_DIR / "example_data"
|
|
CONFIG_FILE = DATA_DIR / "agent_count.txt"
|
|
|
|
|
|
def parse_positive_integer(number_text):
|
|
"""把文本转换为正整数，失败时返回 None。"""
|
|
try:
|
|
number = int(number_text)
|
|
except ValueError:
|
|
# ValueError 表示值的内容不符合转换要求。
|
|
print(f"“{number_text}”不是有效整数。")
|
|
return None
|
|
|
|
if number <= 0:
|
|
print("数字必须大于 0。")
|
|
return None
|
|
|
|
return number
|
|
|
|
|
|
def divide_tool_count(tool_count, agent_count):
|
|
"""计算平均工具数量，除数为零时返回 None。"""
|
|
try:
|
|
average = tool_count / agent_count
|
|
except ZeroDivisionError:
|
|
# ZeroDivisionError 表示程序尝试用数字除以零。
|
|
print("Agent 数量不能为 0。")
|
|
return None
|
|
else:
|
|
# 只有 try 中没有发生异常时，才会执行 else。
|
|
return average
|
|
|
|
|
|
def read_agent_count(file_path):
|
|
"""读取文件中的 Agent 数量，并保证输出结束提示。"""
|
|
try:
|
|
content = file_path.read_text(encoding="utf-8")
|
|
return int(content.strip())
|
|
except FileNotFoundError:
|
|
print(f"没有找到文件：{file_path.name}")
|
|
except ValueError:
|
|
print(f"{file_path.name} 中保存的不是有效整数。")
|
|
finally:
|
|
# 无论是否发生异常，finally 都会执行。
|
|
print(f"已完成对 {file_path.name} 的读取尝试。")
|
|
|
|
return None
|
|
|
|
|
|
def prepare_example_file():
|
|
"""在课程专用目录中准备安全的示例文件。"""
|
|
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
|
CONFIG_FILE.write_text("3\n", encoding="utf-8")
|
|
|
|
|
|
def main():
|
|
"""依次运行整数转换、除法和文件读取示例。"""
|
|
print("一、处理整数转换异常")
|
|
valid_number = parse_positive_integer("5")
|
|
invalid_number = parse_positive_integer("五")
|
|
print(f"有效结果：{valid_number}")
|
|
print(f"无效结果：{invalid_number}")
|
|
|
|
print("=" * 30)
|
|
print("二、处理除以零异常")
|
|
average = divide_tool_count(6, 3)
|
|
failed_average = divide_tool_count(6, 0)
|
|
print(f"正常结果：{average}")
|
|
print(f"异常结果：{failed_average}")
|
|
|
|
print("=" * 30)
|
|
print("三、处理文件读取异常")
|
|
prepare_example_file()
|
|
agent_count = read_agent_count(CONFIG_FILE)
|
|
missing_count = read_agent_count(DATA_DIR / "missing.txt")
|
|
print(f"文件中的 Agent 数量：{agent_count}")
|
|
print(f"不存在文件的读取结果：{missing_count}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|