python通过方法名执行方法-Python如何获取到当前函数名和通过字符串调用函数

获取当前函数名:

应用环境: 某些时候, 为了简化和更好扩展程序,我们需要获取到当前运行的函数名字

方法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/eval来调用, 下面有类和函数的例子

类:

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/eval):

def bar():

print("This is bar")

def foo(stra):

try:

exec(stra) # 如果需要return 值, 则一定要用eval

except:

print("No this module")

foo("bar()")

结合一起用的例子

假设对外提供一个load的函数, 通过一系列的判断, 我获取到一个ext_module_name的值, 通过这个值和"load", 我各种low 错了,是load. 如果以后需要扩展xml, gson, yaml等等, 只有写上对应的###_load就可以了, 不用修改load, 这样可以满足开闭原则

import traceback

ext_module_name = "_pickle_" #假设这个变量是之前一系列过程得到的结果

def who_am_i():

return traceback.extract_stack()[-2][2]

def _json_load():

return "json load"

def _pickle_load():

return "pickle load"

def load():

my_name = who_am_i()

try:

return eval("%s%s()" % (ext_module_name, my_name))

except Exception as e:

raise e

print(load())