code count

code_count.py

@@ -0,0 +1,19 @@
+import os
+
+def code_count(path):
+    print path
+    file_count = 0
+    line_count = 0
+
+    for root, dirs, files in os.walk(path):
+        for file in files:
+            ext = file[file.find('.'):]
+            if ext in ['.cpp', '.c', '.hpp', '.h']:
+                file_count += 1
+                file_path = root + '\\' + file
+                fp = open(file_path)
+                line_count += len(fp.readlines())
+                fp.close()
+
+    print file_count
+    print line_count

main.py

@@ -1,4 +1,5 @@
 import os
+from code_count import code_count
 
 source_path = './data/'
 database = {}
@@ -58,4 +59,5 @@ def parse_file(filename):
         if os.path.isfile(filename):
             parse_file(filename)
 
-    print database
+    #print database
+    code_count(source_path)