一个得到类命名的方法

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)
   

复制代码

在AUnit.GetMsg()函数调用，得到这个函数的调用类的时候，要返回的是

1. __main__.AUnit

复制代码

当然这里假设是在__main__的模块内调用

如果是AUnit.AUnit2.GetMsg()调用，得到的是

1. __main__.AUnit.AUnit2

复制代码

依此类推。

只好自己动手写一个

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
    

复制代码

大家可以测试。