應用環境: 某些時候, 為了簡化和更好擴展程序,我們需要獲取到當前運行的函數名字

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

Python如何獲取到當前函數名和通過字符串調用函數