最近组内准备整顿代码,领导让我写个简单的python脚本分析代码中注释的行数和无效注释。因为这个需求不是很急,所以我想把简单的事情做复杂点。于是就写了一个用VC内嵌Python,并通过模拟按键和发消息去控制其他软件的工具。(转载请指明出处)
作为一个程序员,总是希望自己写的东西别人能用上且喜欢去用。因为python更新很快,往往两个版本中存在一些语法或者实现的改动。其实最讨厌的就是语法变动了,像2.X的print到3.X时就是print()了。我本意是希望做个大家都能用上的,于是我决定用VC内嵌一个python引擎去完成相关工作。
首先说一下环境准备,这个网上一大堆,主要是下载并配置好pythonXX_d.lib、pythonXX.lib、python.dll等几个文件。然后一切就可以开始了。
python分析的过程也很简单, AnalyzeFloder.py是对一个文件夹进行遍历,并做相关统计;AnalyzeFileComment.py是分析每个文件并且筛选出注释;CommentUseOrNo.py是分析每个注释,看看是否是无用注释;AnalyzeReport.py做相关统计和序列化、反序列化工作。 AFCUI.py 是对上述功能的封装。这里说一下对无用注释的定义,我们定义如果注释内容为符合VC编码风格的代码即认为是无效代码。在现实开发中,特别是人员流动性特别大的公司,新人做需求时往往不敢去碰那些没有任何注释的代码,于是他们就直接把这些似乎不用了的代码注释掉。这样长久下来,“无用注释”的量就会很大了。
代码语言:javascript复制 # a(x)
# a (x)
# a (x)
# x = x
# x x
# x x,y y
self.InitNoUseRe(r'[a-zA-Z](w)*([ t])*([ws]*|([ws]*(,[ws]*)))')
# }
# }a;
self.InitNoUseRe(r'}[ trn]*([a-zA-z] ;)?[ trn]*)
self.InitNoUseRe(r'^([ rtn]*)({[ rtn]*))
self.InitNoUseRe(r'[a-zA-Z] [ t]*[=><]{1,2}([ t]*[a-z0-9A-Z] [ t]*))
self.InitNoUseRe(r'#define[ t][a-zA-Z]*([ rtn]*))
self.InitNoUseRe(r'return[ t][a-zA-Z]*(;[ rtn]*))
self.InitNoUseRe(r'[a-zA-Z](w)* (;[ rtn]*))
self.InitNoUseRe(r'[a-zA-Z](w)*--(;[ rtn]*))
self.InitNoUseRe(r'^([ rtn]*continue)(;[ rtn]*))
self.InitNoUseRe(r'^([ rtn]*break)(;[ rtn]*))
self.InitNoUseRe(r'case (w) (:[ rtn]*))
self.InitNoUseRe(r'^([ rtn]*[a-zA-Z](w)*[ rtn] [_a-zA-Z](w)*)((;|{|([w*]);)[ rtn]*))
self.InitNoUseRe(r'^([ rtn]*))
self.InitNoUseRe(r'^([ rtn]*(struct|class))([ rtn]* [a-zA-Z](w)*)(({|;)?[ rtn]*))
以上是我定义的无效代码的正则表达式,因为正则也是拿来用用,没有好好研究过,所以写的不好,凑合着用了,从目前的效果看,还是可以的。
其次,对于分析出来的数据,我将其序列化到一个文件中,这样以后要是加载之前分析过的工程,就可以直接反序列化那个文件就行了,不用重复分析。
代码语言:javascript复制 def GetFileInfoMapFromDump(self,filepath):
with open(filepath, 'rb') as f:
data = pickle.load(f)
return data
def DumpFileInfoMap(self,filemap,filepath):
with open(filepath, 'wb') as f:
pickle.dump(filemap, f, pickle.HIGHEST_PROTOCOL)
再次,说下python和VC的通信。python可以直接和VC通信,但是我返回的数据是一个复杂的结构,我一直也没想到一个简便的办法去解决,于是,我就自定义一种格式,让python在处理完数据后,将数据保存为我定义的格式,然后把文件名返回给VC,VC去读并分析这个文件,从而得到数据。
代码语言:javascript复制def GenerateVcDataFile(self,fileinfomap,vcdatefilepath):
vcdatafileobj = open( vcdatefilepath, 'w ')
for key in fileinfomap:
strvcdata = None
strvcdata = key
index = 0
for value in fileinfomap[key]:
if index == 3:
for linenum in value:
strvcdata = strvcdata '|' str(linenum)
else:
strvcdata = strvcdata '|' str(value)
index = index 1
strvcdata = strvcdata '|n'
vcdatafileobj.write(strvcdata)
if None != vcdatafileobj:
vcdatafileobj.close()
return vcdatefilepath
对应的VC解析代码我放在CAnalyzeVcData中,代码就不贴的(没严格省代码)。
python的大致流程就是如此,VC要是想执行Python的脚本,就如下了
代码语言:javascript复制CString CManageTask::AnanlyzeCodeFloder( const CString& cstrCodeFloderPath,
const CString& cstrProjectFloderPath )
{
CString cstrVcDataPath = L"";
Py_Initialize();
// 进行初始化python
PyObject* pModule = NULL;
// 函数对象
PyObject* pFunc = NULL;
do
{
// 找到模块对象指针
pModule = PyImport_ImportModule("AFCUI");
if ( NULL == pModule )
{
ASSERT(pModule);
break;
}
// 从模块对象中获取指定函数对象指针
pFunc = PyObject_GetAttrString( pModule, "AFloderFunc" );
if ( NULL == pFunc )
{
ASSERT(pFunc);
break;
}
std::string strProjectPath = ConvertCstrint2String(cstrProjectFloderPath);
std::string strCodeFloderPath = ConvertCstrint2String(cstrCodeFloderPath);
// 构造参数
PyObject* pargslist = Py_BuildValue("(s,s)", strCodeFloderPath.c_str(), strProjectPath.c_str());
char* pchvcdatapath = NULL;
// 调用函数
PyObject* pstr = PyEval_CallObject(pFunc, pargslist);
if ( NULL == pstr )
{
ASSERT(pstr);
break;
}
PyArg_Parse(pstr, "s", &pchvcdatapath);
if ( NULL != pchvcdatapath )
{
cstrVcDataPath = CA2W(pchvcdatapath);
}
} while (0);
Py_Finalize();
return cstrVcDataPath;
}
如果对于分析出的数据,为了方便用户查看,我得打开源代码并定位到那个位置。我想过很多方法,想过自己做RichEdit,但是我对界面实在不感冒,于是懒的情绪就上来了。但是我对“玩”还是很有感觉的,我决定盗用微软的Notepad.exe,还修改了Notepad.exe的资源,成为我定制版的东东。然后通过模拟按键和发消息等方法去让它去打开源代码并定位位置。
但是使用中,我发现有点不爽。因为我发现,notepad.exe不可以显示行。UE呢还要注册,于是选用Notepadplusplus了。Notepadplusplus还支持多标签页,让我感觉很兴奋,超出我的预期。
贴一段控制notepadplusplus的代码。
代码语言:javascript复制DWORD CManageTask::CreateNotePad()
{
STARTUPINFO si;
PROCESS_INFORMATION pi;
ZeroMemory( &pi, sizeof(pi) );
ZeroMemory( &si, sizeof(si) );
si.cb = sizeof(si);
CString cstrNNotepadPath = m_strFloder L"\Notepadplusplus\notepad .exe";
CreateProcess( cstrNNotepadPath,
L"", NULL, NULL, FALSE, NULL, NULL,NULL , &si, &pi);
return pi.dwProcessId;
}
RECT CManageTask::GetNotePadRect()
{
RECT rc;
m_pAnalyzeCodeDlg->GetWindowRect(&rc);
m_pAnalyzeCodeDlg->ClientToScreen(&rc);
RECT nrc;
nrc.left = rc.left;
nrc.right = nrc.left rc.right - rc.left;
nrc.top = rc.top rc.bottom - rc.top;
nrc.bottom = nrc.top rc.bottom - rc.top;
m_pAnalyzeCodeDlg->ScreenToClient( &nrc );
return nrc;
}
HWND CManageTask::GetNotePadFilePathEdit( HWND hwndP )
{
if ( NULL == hwndP )
{
return NULL;
}
HWND hComboBoxEx32 = ::FindWindowEx( hwndP, NULL, L"ComboBoxEx32", NULL );
HWND hComboBox = ::FindWindowEx( hComboBoxEx32 , NULL, L"ComboBox", NULL );
HWND hEdit = ::FindWindowEx( hComboBox , NULL, L"Edit", NULL );
return hEdit;
}
HWND CManageTask::GetNotePadOkButton( HWND hwndP )
{
if ( NULL == hwndP )
{
return NULL;
}
HWND hOKButton = ::FindWindowEx( hwndP, NULL, L"Button", NOTEPADOPENFILENAME2 );
return hOKButton;
}
HWND CManageTask::GetNotePadOpenFileDialog()
{
HWND hAfter = NULL;
HWND h = ::FindWindowEx( NULL, hAfter, NULL, NOTEPADOPENFILENAME );
DWORD dwPID = 0;
while ( NULL != h )
{
h = ::FindWindowEx( NULL, hAfter, NULL, NOTEPADOPENFILENAME );
GetWindowThreadProcessId( h, &dwPID );
if ( dwPID == m_dwThreadNotePadID )
{
return h;
}
else
{
hAfter = h;
}
}
return NULL;
}
HWND CManageTask::GetNotePadEdit( HWND hwndP )
{
if ( NULL == hwndP )
{
return NULL;
}
HWND hEidt = ::FindWindowEx( hwndP, NULL, L"Scintilla", NULL);
return hEidt;
}
VOID CManageTask::OpenNewFile( const CString& cstrFilePath )
{
HWND hwndP = NULL;
while ( NULL == hwndP )
{
::SetForegroundWindow( m_hwndNotePad );
::SetActiveWindow( m_hwndNotePad );
::SetFocus( m_hwndNotePad );
{
keybd_event( VK_CONTROL, MapVirtualKey(VK_CONTROL, 0), 0, 0 );
keybd_event( 0x4F, 0, 0, 0 );
keybd_event( 0x4F, 0, KEYEVENTF_KEYUP, 0 );
keybd_event( VK_CONTROL, MapVirtualKey(VK_CONTROL, 0), KEYEVENTF_KEYUP, 0 );
}
Sleep(500);
hwndP = GetNotePadOpenFileDialog();
}
HWND hEdit = GetNotePadFilePathEdit(hwndP);
HWND hOKButton = GetNotePadOkButton(hwndP);
CString strTmp = cstrFilePath;
if ( NULL != hEdit )
{
::SendMessage( hEdit, WM_SETTEXT, 0, (LPARAM)strTmp.GetBuffer() );
strTmp.ReleaseBuffer();
if ( NULL != hOKButton )
{
::SendMessage( hOKButton, WM_LBUTTONDOWN, 0, 0);
::SendMessage( hOKButton, WM_LBUTTONUP, 0, 0);
}
}
}
于是大功告成了。
目前还有很多没有完善的地方,比如字符串的转码(程序放在中文目录下有转码问题,出在python回传结果的时候)和规则(规则不全面)。还有很多可以“玩”的功能,比如对单个文件的重新分析。目前我思路都想好了,也预留了接口。我想通过FindFirstChangeNotification监控指定目录文件,如果有改动,就调用AFCUI.py中
代码语言:javascript复制 def AnalyzeCodeFilePathFunc(self,codefilepath):
analyzefile = CAnalyzeFloder()
analyzefile.AnalyzeFile(codefilepath)
fileinfomap = analyzefile.GetCodeFloderInfo()
AnalyzeReport(fileinfomap)
(需要封装)。然后VC得到当个文件的可以解析的数据,并更新内存和界面。因为之前玩过FindFirstChangeNotification(我毕设就是玩它的),所以我就没多大兴趣去做这个功能了。
还有这次MFC代码,我尝试了下界面和逻辑分离的原则,很好玩,代码也清晰很多,所有的逻辑都在CManageTask类中。