一個得到類命名的方法

jeppeter

想得到一個類全名，但找了很多的方法，都沒有辦法直接得到，如

1. class AUnit(object):
   
2.         def GetMsg(self):
   
3.                 return clsname.GetCallerClassName(1)
   
4.         class AUnit2(object):
   
5.                 def GetMsg(self):
   
6.                         return clsname.GetCallerClassName(1)
   
7.                 class AUnit3(object):
   
8.                         def GetMsg(self):
   
9.                                 return clsname.GetCallerClassName(1)
   

復(fù)制代碼

在AUnit.GetMsg()函數(shù)調(diào)用，得到這個函數(shù)的調(diào)用類的時候，要返回的是

1. __main__.AUnit

復(fù)制代碼

當然這里假設(shè)是在__main__的模塊內(nèi)調(diào)用

如果是AUnit.AUnit2.GetMsg()調(diào)用，得到的是

1. __main__.AUnit.AUnit2

復(fù)制代碼

依此類推。

只好自己動手寫一個

1. #! python
   
2. import inspect
   
3. import types
   
4. import logging
   
5. def FuncMethodSearh(obj):
   
6.         if inspect.ismethod(obj):
   
7.                 return True
   
8.         if inspect.isfunction(obj):
   
9.                 return True
   
10.         if inspect.isclass(obj):
    
11.                 return True
    
12.         return False
    
13. 
    
14. 
    
15. def __IsCodeInClass(clsobj,codeobj):
    
16.         mns = inspect.getmembers(clsobj,predicate=FuncMethodSearh)
    
17.         ret = 0
    
18.         for m in mns:
    
19.                 #logging.info('%s %s'%(repr(m[1]),repr(codeobj)))
    
20.                 if type(m[1]) is types.MethodType or type(m[1]) is types.FunctionType or\
    
21.                   type(m[1]) is types.GeneratorType:
    
22.                         fobj = m[1].func_code
    
23.                         if fobj.co_name == codeobj.co_name  and \
    
24.                         fobj.co_firstlineno == codeobj.co_firstlineno:
    
25.                                 ret = 1
    
26.                                 break
    
27.         return ret
    
28. 
    
29. def __GetCodeObjClass(clsobj,codeobj,calllevel=0):
    
30.         if calllevel > 300:
    
31.                 raise
    
32.         ret = __IsCodeInClass(clsobj,codeobj)
    
33.         if ret > 0:
    
34.                 return clsobj.__name__
    
35.         mns = inspect.getmembers(clsobj,predicate=inspect.isclass)
    
36.         for m in mns:
    
37.                 # we do not recursive for the two  types
    
38.                 if m[0] != '__class__' and m[0] != '__base__':
    
39.                         #logging.info('%s %s'%(repr(m[1]),repr(codeobj)))
    
40.                         n = __GetCodeObjClass(m[1],codeobj,calllevel + 1)
    
41.                         if n:
    
42.                                 return clsobj.__name__+'.'+n
    
43.         return None
    
44. 
    
45. def GetCallerClassFullName(modobj,codeobj):
    
46.         '''
    
47.                 @modobj : module object area to search for code object
    
48.                 @codeobj : code object will find its class belong to
    
49. 
    
50.                 @return : it codeobj can not detect the name return module name
    
51.                               if codeobj is the function not belong to any of the class
    
52.                               return the module name
    
53.                               if codeobj is the method or function of class ,return whole class name including module name
    
54.         '''
    
55.         mns = inspect.getmembers(modobj,predicate=FuncMethodSearh)
    
56.         ret = 0
    
57.         for m in mns:
    
58.                 if type(m[1]) is types.MethodType or type(m[1]) is types.FunctionType or\
    
59.                   type(m[1]) is types.GeneratorType:
    
60.                         fobj = m[1].func_code
    
61.                         if fobj.co_name == codeobj.co_name  and \
    
62.                         fobj.co_firstlineno == codeobj.co_firstlineno:
    
63.                                 ret = 1
    
64.                                 break
    
65.         if ret > 0:
    
66.                 return modobj.__name__
    
67. 
    
68.         mns = inspect.getmembers(modobj,predicate=inspect.isclass)
    
69.         for m in mns:
    
70.                 n = __GetCodeObjClass(m[1],codeobj)
    
71.                 if n:
    
72.                         return modobj.__name__ +'.'+n
    
73.         return modobj.__name__
    
74.        
    
75. def GetCallerClassName(level=2):
    
76.         mn = ''
    
77.         stks = inspect.stack()
    
78.         if len(stks) > level:
    
79.                 frm = stks[level]
    
80.                 mm = inspect.getmodule(frm[0])
    
81.                 fc = frm[0].f_code
    
82.                 mn = GetCallerClassFullName(mm,fc)
    
83.         return mn
    

復(fù)制代碼

大家可以測試。