"""
.. Hint:: 公用模块
异常处理等等
.. code-block:: python
:linenos:
@异常处理返回类型逻辑型
def 如果函数发生错误即可提示兵忽略:
print("没问题")
a = 0/0 # 这个一定报错
"""
import re, sys, traceback, datetime
异常显示信息 = 2
报错_列表索引 = "取列表内容时，索引超出范围"
报错_层级 = "调用层级如下"
报错_除零 = "请勿除以零"
报错_递归 = "递归过深。请确认: 1、的确需要递归 2、递归的收敛正确"
报错_按索引取项 = "不支持按索引取项"
报错_无属性 = "没有属性"
参考_enter = "\n参考：https://stackoverflow.com/questions/1984325/explaining-pythons-enter-and-exit"
[文档]def 设置_异常处理_显示信息(显示信息=2):
"""
:param 显示信息: 0 不显示 1显示简单的 2显示详细的
:return:
"""
global 异常显示信息
异常显示信息 = 显示信息
[文档]class 层信息:
def __init__(self, 行号, 内容, 文件名):
self.行号, self.内容, self.文件名 = 行号, 内容, 文件名
[文档]def 提取(各层):
各行 = []
for 层号 in range(len(各层) - 1, -1, -1):
层 = 各层[层号]
文件名 = 层.filename
各行.append(层信息(层.lineno, 层.line, 文件名))
return 各行
[文档]def 报错信息(例外):
exc_type, exc_value, 回溯信息 = sys.exc_info()
各层 = traceback.extract_tb(回溯信息)
# print(repr(各层))
各行 = []
行信息 = 提取(各层)
类型 = 例外.__class__.__name__
原信息 = str(例外)
if len(行信息) > 0:
关键 = 提示(类型, 原信息)
各行.append(关键)
for 行号, 行 in enumerate(行信息, start=1):
# 在第二层前显示
if 行号 == 2:
各行.append(报错_层级)
各行.append((f"见 File \"{行.文件名}\", ") + f"line {行.行号}：{行.内容}")
return 各行
[文档]def 提示(类型, 原信息):
if 类型 == 'NameError':
return re.sub(r"name '(.*)' is not defined", r"请先定义‘\1’再使用", 原信息)
elif 类型 == 'ZeroDivisionError':
return 报错_除零
elif 类型 == 'RecursionError':
return 报错_递归
elif 类型 == 'UnboundLocalError':
return re.sub(
r"local variable '(.*)' referenced before assignment",
r"请先对本地变量‘\1’赋值再引用",
原信息)
elif 类型 == 'KeyError':
return "字典中不存在此键：" + 原信息
elif 类型 == 'TypeError':
模式 = 'can only concatenate str \(not "(.*)"\) to str'
无法取项 = "'(.*)' object is not subscriptable"
参数缺失 = "(.*) missing 1 required positional argument: '(.*)'"
if re.match(模式, 原信息):
return re.sub(模式, r'字符串只能拼接字符串，请将“\1”先用 str() 转换', 原信息)
匹配 = re.search(无法取项, 原信息)
if 匹配:
return f'{类型中文化(匹配.group(1))}{报错_按索引取项}'
if re.match(参数缺失, 原信息):
return re.sub(参数缺失, r'函数“\1”需要“\2”参数', 原信息)
elif 类型 == 'IndexError' and 原信息 == "list index out of range":
return 报错_列表索引
elif 类型 == 'AttributeError':
信息 = "需要添加此属性：" + 原信息
if 原信息 == "__enter__":
信息 += 参考_enter
return 信息
无属性 = "'(.*)' object has no attribute '(.*)'"
匹配 = re.search(无属性, 原信息)
if 匹配:
return f'{类型中文化(匹配.group(1))}{报错_无属性}‘{匹配.group(2)}’'
elif 类型 == 'FileNotFoundError':
return re.sub(r"\[Errno 2\] No such file or directory: '(.*)'",
r"没找到文件或路径：‘\1’",
原信息)
elif 类型 == 'ModuleNotFoundError':
return re.sub(r"No module named '(.*)'", r"没找到模块：‘\1’", 原信息)
return 类型 + "：" + 原信息
[文档]def 类型中文化(类型):
中英表 = {
"NoneType": "空变量",
"int": "整数变量",
"bool": "真假变量",
"function": "函数",
}
return 中英表[类型] if 类型 in 中英表 else 类型
[文档]def 异常处理返回类型逻辑型(function):
"""
使用方法 在def上面加上 @异常处理返回类型逻辑型
例如
.. code-block:: python
:linenos:
@异常处理返回类型逻辑型
def 如果函数发生错误即可提示兵忽略:
print("没问题")
a = 0/0 # 这个一定报错
"""
def box(*args, **kwargs):
try:
return function(*args, **kwargs)
except Exception as 例外:
if 异常显示信息 == 2:
print("函数异常 %s 时间 %s \r\n%s" % (
function.__name__,
str(datetime.datetime.now()),
"\n".join(报错信息(例外))
))
# 调试用：对比原英文信息
# print("\n\n" + traceback.format_exc())
if 异常显示信息 == 1:
print("函数异常 %s 时间 %s" % (
function.__name__,
str(datetime.datetime.now()),
))
return False
return box