Python如何获取到当前函数名和通过字符串调用函数

Posted 2020-10-21

获取当前函数名:

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

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