介绍

本文以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,程序结果界面如下,很完美对吧。通过上述操作,我们实现了自动将功能函数添加到字典中。

总结:自动将函数对象添加到字典的bug-LMLPHP

不过值得注意的有如下几点:

  • 我们是将字典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函数,结果。。。。。字典居然是空的

总结:自动将函数对象添加到字典的bug-LMLPHP

也就是说,我的这些操作,就是啥也没做。但我明明做了呀。

这个问题,我研究了半天愣是没想明白。后来跟着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路径

05-11 22:13