介绍
本文以ATM项目为背景，介绍一个比较实用的编程技巧，使用装饰器将项目中的指定函数添加到字典中。
利用字典通过key访问value的特点，实现用户输入编号，通过字典直接获取并调用编号对应的功能函数。
# 实现的目标：让用户输入使用功能的编号，程序调用相应的函数。
基础版-if条件判断
def login():
print('this is login function')
def register():
print('this is register function')
def transfer():
print('this is transfer function')
# 主程序
def atm():
desc = '1: 登录\n2：注册\n3：转账'
while 1:
print(desc)
cmd = input('请选择您要使用的功能编号：').strip()
if cmd == '1':
login()
elif cmd == '2':
register()
elif cmd == '3':
register()
else:
print('编号不存在，请重新选择')
if __name__ == '__main__':
atm()
# 基础版本，每个数字字符对应不同的函数，通过if-elif-else判断用户的选择，然后调用。
# 优点：简单明亮，思路清晰
# 缺点：代码重复，程序臃肿
提高版-函数字典
# 功能函数略.......
# 主程序
def atm():
# 函数字典
cmd_func = {
'1': ('登录', login),
'2': ('注册', register),
'3': ('转账', transfer),
}
while 1:
for k, v in cmd_func.items():
print(f'({k}){v[0]}', end='\t')
cmd = input('\n请选择您要使用的功能编号：').strip()
if cmd not in cmd_func:
print('编号不存在，请重新选择')
continue
# 编号存在字典中的情况
func = cmd_func.get(cmd)[1] # func是cmd_func字典中value的第二个元素
func() # 调用函数
if __name__ == '__main__':
atm()
# 提高版本，将每个功能函数放在一字典中，key是编号，值中包含有函数名，输入key就可以获取到值中的功能函数
# 优点：思路清晰，代码简洁，程序优美
# 缺点：需要手动创建一个函数字典
进阶版-自动生成函数字典
# 装饰器
def auto(desc):
from functools import wraps
def wrapper(func):
@wraps(func)
def inner(*args, **kwargs):
index = len(cmd_func) + 1 # 调用全局函数字典(初始为空),自动计算编号
cmd_func[str(index)] = (desc, func) # 将编号，描述信息、函数加到字典中
return inner
return wrapper
@auto('登录')
def login():
print('this is login function')
@auto('注册')
def register():
print('this is register function')
@auto('转账')
def transfer():
print('this is transfer function')
# 自动调用被装饰函数的函数，需要手动排除全局名称空间中不需要的名字
def auto_append():
my_func = [v for k, v in globals().items() if callable(v) if k not in ['atm', 'auto', 'auto_append']]
for func in my_func: func()
# 函数字典
cmd_func = {}
# 主程序
def atm():
# 调用函数自动添加函数字典
auto_append()
while 1:
for k, v in cmd_func.items():
print(f'({k}){v[0]}', end='\t')
cmd = input('\n请选择您要使用的功能编号：').strip()
if cmd not in cmd_func:
print('编号不存在，请重新选择')
continue
# 编号存在字典中的情况
func = cmd_func.get(cmd)[1] # func是cmd_func字典中value的第二个元素
func() # 调用函数
if __name__ == '__main__':
atm()
# 进阶版-编写装饰器，每个被装饰的函数调用，就会被添加到一个字典中，然后再写一个自动发现这些他们的函数，自动调用
# 优点：使用字典函数时不需要手动添加
# 缺点：需要字典是全局变量，cmd_func = {}，手动排除不需要的函数，功能函数少时提现不出它的优势
# 上面的情况是全部函数都在一个文件中；
# 如果功能函数在其他文件中，笔触采用导入的方式运行，这种情况使用自动添加函数字典就优势明显。
高级版-跨文件自动添加
这里使用简化版的软件开发目录规范，将不同功能的函数放在不同的文件中，并且所有的文件在一个文件夹下。
ATM/
|-- run.py # 程序启动文件
|-- atm.py # 主函数文件，即atm()函数
|--funcs.py # 存放 login()、register()、transfer()函数的文件
|--tools.py # 存放装饰器函数auto()、auto_append()函数的文件
run.py
from atm import atm # 导入atm.py下的atm()函数
if __name__ == '__main__':
atm() # 运行atm()函数
atm.py
from tools import auto_append # 从tools.py导入auto_append函数
# 函数字典，将login、register、transfer自动加到这个字典中
cmd_func = {}
# 主程序
def atm():
auto_append() # 调用auto_append函数自动添加函数字典
while 1:
for k, v in cmd_func.items():
print(f'({k}){v[0]}', end='\t')
cmd = input('\n请选择您要使用的功能编号：').strip()
if cmd not in cmd_func:
print('编号不存在，请重新选择')
continue
# 编号存在字典中的情况
func = cmd_func.get(cmd)[1] # func是cmd_func字典中value的第二个元素
func() # 调用函数
funcs.py
from tools import auto # 从tools.py导入auto装饰器
@auto('登录') # 装饰login
def login():
print('this is login function')
@auto('注册') # 装饰register
def register():
print('this is register function')
@auto('转账') # 装饰transfer
def transfer():
print('this is transfer function')
tools.py
from functools import wraps
def auto(desc): # 装饰器
from atm import cmd_func # 函数内导入，避免循环导入问题
def wrapper(func):
@wraps(func)
def inner(*args, **kwargs):
index = len(cmd_func) + 1 # 访问atm.py文件的cmd_func字典
cmd_func[str(index)] = (desc, func) # 将函数自定添加到该字典中
return inner
return wrapper
def auto_append():
# 在当前局部名称空间中导入需要添加到字典里面的函数
from funcs import login, register, transfer
my_funcs = locals()
for func in my_funcs.values():
func() # 自动执行，即执行装饰器内的inner函数，自动添加到字典
运行run.py，程序结果界面如下，很完美对吧。通过上述操作，我们实现了自动将功能函数添加到字典中。
不过值得注意的有如下几点：
-
我们是将字典
cmd_func放在了atm.py文件的全局名称空间中，函数装饰器auto在使用它的时候需要从atm.py文件中导入cmd_func。此处需要注意的是循环导入问题。这种情况下解决循环导入问题有两中方案。方案1：在auto函数内导入from atm import cmd_func；方案2：tools.py文件顶部导入import atm，auto函数内使用atm.cmd_func -
如果将字典
cmd_func放在一个独立的py文件中，而不是放在atm.py文件中，那就很方便了，只要在需要它的位置导入并引用就好了，不会造成循环导入问题。 -
函数auto_append内导入login、register、transfer函数，局部名称空间的名字。
补充
说到这里就可以结束了，因为上面的操作很流畅的实现了我们目的。这种操作在功能数量多，跨文件导入时，非常实用。
在刚要结束的时候，我突然想试试，如果不通过run.py文件调用atm函数，而是直接运行atm.py文件，调用atm函数，是不是也可以？于是，在atm.py文件末尾增加if __name__ == '__main__': atm()
from tools import auto_append # 从tools.py导入auto_append函数
# 函数字典，将login、register、transfer自动加到这个字典中
cmd_func = {}
# 主程序
def atm():
auto_append() # 调用auto_append函数自动添加函数字典
while 1:
for k, v in cmd_func.items():
print(f'({k}){v[0]}', end='\t')
cmd = input('\n请选择您要使用的功能编号：').strip()
if cmd not in cmd_func:
print('编号不存在，请重新选择')
continue
# 编号存在字典中的情况
func = cmd_func.get(cmd)[1] # func是cmd_func字典中value的第二个元素
func() # 调用函数
if __name__ == '__main__': # 增加了这个判断，不影响该文件被导入执行
atm()
执行atm.py文件，调用atm函数，结果。。。。。字典居然是空的
也就是说，我的这些操作，就是啥也没做。但我明明做了呀。
这个问题，我研究了半天愣是没想明白。后来跟着debug调试，后来发现其实原因很简单。
使用run.py文件导入atm函数这种方式执行atm函数时
被导入的模块atm已经在导入模块的时候会执行atm.py文件，执行完毕后atm.py模块名称空间里面就有了一个名字cmd_func这个字典。
后面执行atm函数内部的auto_append函数，会进入装饰器auto内部，装饰器内部遇到 from atm import cmd_func时会再次导入atm模块，但因为atm模块已经被加载到内存了，所以此时不会再执行atm.py文件，而是直接在内存中找到了cmd_func这个名字。后面装饰器内部使用的都是这个相同的字典，即atm.py文件内的字典。
直接执行atm.py文件，调用atm函数的方式情况却不同。
执行atm.py文件，程序走到cmd_func = {}处，程序会为这个字典开辟一个内存空间。
后面执行atm函数，进入atm函数内部，遇到auto_append()，就会跳转到auto_append函数内部。
auto_append内部因为导入login，会进入funcs.py文件内，由于装饰器的存在，程序又会进入装饰器auto函数内。
auto内就比较关键了，因为要使用cmd_func字典，程序需要再次从atm.py文件导入这个字典。
现在，内存中是没有atm这个模块的(因为还在执行这个文件)。根据导入模块时的搜索路径(内存、内置、sys.path)，此时就会再次执行atm.py文件，遇到cmd_func={}，就会再次在内存中开辟一块空间存这个字典。
所以我们在装饰器中使用的字典是第二次生成的字典，而atm函数内部使用的字典却是第一次生成的那个字典。他们在不同的作用域里面。
所以这个问题的关键在于：两种方式导入atm.py这个模块的先后顺序不同，造成第一种方式整个程序使用的是一个字典；而第二方式开辟了两个字典，我们操作的是后来的第二个字典，而程序使用的却是第一个。
核心知识点是模块导入时的搜索顺序：内存、内置、sys.path路径