12wang3
diff --git a/‎PyBeaner/0007/code_lines.py‎
Lines changed: 43 additions & 0 deletions b/‎PyBeaner/0007/code_lines.py‎
Lines changed: 43 additions & 0 deletions
@@ -0,0 +1,43 @@
+# coding=utf-8
+__author__ = 'PyBeaner'
+import os
+import fnmatch
+
+total_lines = 0
+code_lines = 0
+empty_lines = 0
+comment_lines = 0
+
+
+def count_line(line):
+    line = line.lstrip()
+    global comment_lines, empty_lines, total_lines, code_lines
+
+    total_lines += 1
+    if line.startswith("#"):
+        comment_lines += 1
+    elif not line:
+        empty_lines += 1
+    else:
+        code_lines += 1
+
+
+def scan_dir(directory, suffix="*.py"):
+    directory = os.path.abspath(directory)
+    print("Scanning files in %s ..." % directory)
+    for cur_dir, dirs, files in os.walk(directory):
+        for file in files:
+            if not fnmatch.fnmatch(file, suffix):
+                continue
+            file_path = os.path.join(cur_dir, file)
+            with open(file_path,errors="replace") as f:
+                for line in f:
+                    count_line(line)
+
+
+if __name__ == '__main__':
+    scan_dir(r"../..")
+    print("Total lines:%d" % total_lines)
+    print("Code lines:%d" % code_lines)
+    print("Empty lines:%d" % empty_lines)
+    print("Comment lines:%d" % comment_lines)