简明python教程之总结：ATM项目中函数字典自动添加的bug

介绍#

本文以ATM项目为背景，介绍一个比较实用的编程技巧，使用装饰器将项目中的指定函数添加到字典中。

利用字典通过key访问value的特点，实现用户输入编号，通过字典直接获取并调用编号对应的功能函数。

Copy

# 实现的目标：让用户输入使用功能的编号，程序调用相应的函数。

---

基础版-if条件判断#

Copy

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判断用户的选择，然后调用。 # 优点：简单明亮，思路清晰 # 缺点：代码重复，程序臃肿

---

提高版-函数字典#

Copy

# 功能函数略....... # 主程序 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就可以获取到值中的功能函数 # 优点：思路清晰，代码简洁，程序优美 # 缺点：需要手动创建一个函数字典

进阶版-自动生成函数字典#

Copy

# 装饰器 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 = {}，手动排除不需要的函数，功能函数少时提现不出它的优势 # 上面的情况是全部函数都在一个文件中； # 如果功能函数在其他文件中，笔触采用导入的方式运行，这种情况使用自动添加函数字典就优势明显。

---

高级版-跨文件自动添加#

这里使用简化版的软件开发目录规范，将不同功能的函数放在不同的文件中，并且所有的文件在一个文件夹下。

Copy

ATM/ |-- run.py # 程序启动文件 |-- atm.py # 主函数文件，即atm()函数 |--funcs.py # 存放 login()、register()、transfer()函数的文件 |--tools.py # 存放装饰器函数auto()、auto_append()函数的文件

run.py

Copy

from atm import atm # 导入atm.py下的atm()函数 if __name__ == '__main__': atm() # 运行atm()函数

atm.py

Copy

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

Copy

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

Copy

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()

Copy

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路径