获取当前函数名:
应用环境: 某些时候, 为了简化和更好扩展程序,我们需要获取到当前运行的函数名字
方法1(不推荐)
import sys
def I_want_to_know_my_name():
print(sys._getframe().f_code.co_name) #你懂得,下划线开头的不应该使用的
方法2
import traceback
def who_am_i():
return traceback.extract_stack()[-2][2]
def I_want_to_know_my_name():
print(who_am_i())
方法3
import inspect
def who_am_i2():
return inspect.stack()[1][3]
def I_want_to_know_my_name():
print(who_am_i2())
通过字符串调用函数
通过反射或者exec来调用, 下面有类和函数的例子
类:
class foo(object):
def load(self):
f = getattr(self, "load_json")
f()
def load_json(self):
print("This is load_json module")
myfunc = foo()
myfunc.load()
函数(反射):
# 单模块情况下, 多模块就import module1, 然后getattr(module1....)
import sys
def bar():
print("This is bar")
def foo():
print("This is foo")
def check_str(stra):
mod = sys.modules[__name__] #获取当前模块对象
if hasattr(mod, stra):
getattr(mod, stra)()
else:
print("No this module")
check_str("bar") #你可能会问如果输入check_str会怎么样, 我会告诉你缺少变量并报错,不过最好还是在上面函数加判断避免这种情况出现
函数(exec):
def bar():
print("This is bar")
def foo(stra):
try:
exec(stra)
except:
print("No this module")
foo("bar()")
结合一起用的例子
假设对外提供一个load的函数, 通过一系列的判断, 我获取到一个ext_module_name的值, 通过这个值和"load", 我各种low 错了,是load. 如果以后需要扩展xml, gson, yaml等等, 只有写上对应的###_load就可以了
import traceback
ext_module_name = "_pickle_" #假设这个变量是之前一系列过程得到的结果
def who_am_i():
return traceback.extract_stack()[-2][2]
def _json_load():
print("json load")
def _pickle_load():
print("pickle load")
def load():
my_name = who_am_i()
try:
return exec("%s%s()" % (ext_module_name, my_name))
except Exception as e:
raise e
load()
原文地址:http://blog.51cto.com/11257471/2062914
时间: 2024-10-29 20:01:28