W jaki sposób uzyskać nazwę modułu definicji klasy obiektu zamiast nazwy modułu procesu tworzenia obiektu?

Question

W Pythonie 2.5, mam następujący kod w module o nazwie modtest.py:

def print_method_module(method): 
    def printer(self): 
     print self.__module__ 
     return method(self) 
    return printer 

class ModTest(): 

    @print_method_module 
    def testmethod(self): 
     pass 

if __name__ == "__main__": 
    ModTest().testmethod() 

Jednak gdy uruchamiam to, że wypisuje:

__main__ 

Jeśli utworzyć drugi plik nazywa modtest2.py i uruchom go:

import modtest 

if __name__ == "__main__": 
    modtest.ModTest().testmethod() 

ten wypisuje:

modtest 

Jak zmienić dekorator, aby zawsze drukować modtest, nazwę modułu, w którym zdefiniowano klasę?

Answer 1

Po bezpośrednim uruchomieniu pliku źródłowego Pythona, nazwa modułu tego pliku to __main__, nawet jeśli jest znana innej nazwie po wykonaniu innego pliku i zaimportowaniu go.

Prawdopodobnie chcesz zrobić to, co zrobiłeś w modtest2, i zaimportować moduł zawierający definicję klasy, zamiast bezpośrednio wykonywać ten plik. Można jednak uzyskać nazwę pliku głównego modułu, jak na przykład dla celów diagnostycznych:

def print_method_module(method): 
    def printer(self): 
     name = self.__module__ 
     if name == '__main__': 
      filename = sys.modules[self.__module__].__file__ 
      name = os.path.splitext(os.path.basename(filename))[0] 
     print name 
     return method(self) 
    return printer 

Answer 2

Zgaduję, że można użyć hackery sys._getframe(), aby uzyskać to, co chcesz.