本文實(shí)例講述了Python實(shí)現(xiàn)統(tǒng)計(jì)代碼行的方法。分享給大家供大家參考，具體如下：

參加光榮之路測(cè)試開發(fā)班已三月有余，吳總上課也總問“ 咱們的課上了這么多次了大家實(shí)踐了多少行代碼了？”。這里是一個(gè)一臉懵逼的表情。該怎么統(tǒng)計(jì)呢？一個(gè)個(gè)文件數(shù)當(dāng)然不可取，能用代碼解決的事咱們堅(jiān)決不動(dòng)手。最近在網(wǎng)上刷題時(shí)也正好遇到有這么一道題，所以決定擼一擼。

題目：有個(gè)目錄，里面是你自己寫過的程序，統(tǒng)計(jì)一下你寫過多少行代碼。包括空行和注釋，但是要分別列出來。

首先分析一下思路捋一下大象裝冰箱的步驟，從一個(gè)給定的目錄統(tǒng)計(jì)該目錄下所有的代碼行大致需要以下7個(gè)步驟：

1. 遍歷該目錄下所有的文件。
2. 判斷文件是否以“.py”結(jié)尾。（以python代碼為例）
3. 打開.py文件（切忌勿用W+，W+會(huì)清空文件內(nèi)容）
4. 循環(huán)讀取文件的每一行
5. 判斷每一行的內(nèi)容：
                  (a) 注釋: 以#開頭。
                  (b) 注釋：以三引號(hào)開頭結(jié)束。
                  (c) 空行：除空白字符無其他。
                  (d) 代碼行：除空白字符之后還剩下其他字符。
6. 判斷是否為文件末尾,
7. 關(guān)閉文件, 返回結(jié)果.

解題思路捋清楚之后剩下的就是將各模塊的代碼像搭積木一樣搭起來就完事了（示例代碼在本文最后）：

5~9： 導(dǎo)入OS，定義 code_lines_count 函數(shù)并接收一個(gè) path 形式參數(shù)，聲明了三個(gè)變量分別用于統(tǒng)計(jì)代碼行，注釋行和空行
10~13： 遍歷os.walk獲取到的file 對(duì)象，然后將文件的后綴名利用splitext函數(shù)分割然后使用列表索引 [1] 取得文件后綴名，并判斷是否是以“.py”結(jié)尾。（此處也可以使用listdir, 但listdir只能取單層目錄下的文件， 并且需要單獨(dú)判斷取得的元素是文件還是文件夾，較麻煩）
第12行定義了一個(gè)file_abs_path的變量并賦值文件的絕對(duì)路徑，是因?yàn)橄旅娴拇a會(huì)多次使用，不必每次都使用so.path.join(xx,xx)。
14~18：對(duì)于上一個(gè)步驟獲取到的以“.py”結(jié)尾的文件 利用with 方式打開（使用with可省去關(guān)閉文件的代碼），對(duì)于打開的文件使用While True 循環(huán)的使用readline()去讀取文件的每一行并賦值給line 變量。
19~39：該段代碼用于對(duì)上一步驟readline()取得的行做判斷是 代碼行，空行還是注釋行。
19~21：如果line為空，表示取到文件末尾，此時(shí)break while 循環(huán)，繼續(xù)files 中的下一個(gè)文件操作。
22~24：使用strip()對(duì)readline()取得的行做去空白處理，如果經(jīng)過處理后是以“#”開頭則表示這行是一個(gè)注釋行 此時(shí)對(duì)comm_lines 做加1 操作。
25~33：使用strip()對(duì)readline()取得的行做去空白處理，如果經(jīng)過處理后是以三個(gè)單引號(hào)或者三個(gè)雙引號(hào)開頭則表示此處為一個(gè)多行注釋的開始，然后判斷該行的三引號(hào)數(shù)量如果為1則表示注釋分多行， 否則注釋為一行（一對(duì)三引號(hào)在同一行），對(duì)于注釋為多行情況使用while 循環(huán)得讀取接下來的行，并且沒讀一行對(duì) comm_lines 做加1 操作，如果讀到某一行存在三引號(hào)則判定注釋結(jié)束，break 當(dāng)層while循環(huán)（此處只考慮了比較規(guī)范的注釋）
34~36：如果讀到的行做過strip()之后非空且不是注釋，則是一個(gè)代碼行， 并對(duì)code_lines做加1操作。
37~39：如果上述條件都不滿足，則判斷為一個(gè)空行，并對(duì)space_lines 做加1操作。
41：返回統(tǒng)計(jì)到的代碼行，注釋行和空行。
43：測(cè)試代碼下圖是運(yùn)行的一個(gè)實(shí)例

#_*_coding:utf-8_*_

# 統(tǒng)計(jì)代碼行數(shù), 空行, 注釋.

import os
def code_lines_count(path):
  code_lines = 0
  comm_lines = 0
  space_lines = 0
  for root,dirs,files in os.walk(path):
    for item in files:
      file_abs_path = os.path.join(root,item)
      postfix = os.path.splitext(file_abs_path)[1]
      if postfix == '.py':
        #print 'Start: ',file_abs_path
        with open(file_abs_path) as fp:
          while True:
            line = fp.readline()
            if not line:
              #print 'break here,%r' %line
              break
            elif line.strip().startswith('#'):
              #print '1, here',line
              comm_lines += 1
            elif line.strip().startswith("'''") or line.strip().startswith('"""'):
              comm_lines += 1
              if line.count('"""') ==1 or line.count("'''") ==1:
                while True:
                  line = fp.readline()
                  #print '4, here',line
                  comm_lines += 1
                  if ("'''" in line) or ('"""' in line):
                    break
            elif line.strip():
              #print '5, here',line
              code_lines += 1
            else:
              #print '6, here',line
              space_lines +=1
        #print 'Done',file_abs_path
  return code_lines,comm_lines,space_lines
#test
print "Code lines: %d\nComments lines: %d\nWhiteSpace lines: %d" %code_lines_count(r'D:\exercises')

--碼代碼本質(zhì)和打游戲一樣， 都是打怪升級(jí)穿裝備。

更多關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題：《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》、《Python入門與進(jìn)階經(jīng)典教程》及《Python文件與目錄操作技巧匯總》

希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。

最新評(píng)論