Complete 0006

DIYgod · DIYgod · commit 22180281793d · 2015-04-03T00:14:37.000+08:00
diff --git a/DIYgod/0006/important_word.py b/DIYgod/0006/important_word.py
@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+
+import re
+import os
+
+# Get all files in designated path
+def get_files(path):
+    filepath = os.listdir(path)
+    files = []
+    for fp in filepath:
+        fppath = path + '/' + fp
+        if(os.path.isfile(fppath)):
+            files.append(fppath)
+        elif(os.path.isdir(fppath)):
+            files += get_files(fppath)
+    return files
+
+# Get the most popular word in designated files
+def get_important_word(files):
+    worddict = {}
+    for filename in files:
+        f = open(filename, 'rb')
+        s = f.read()
+        words = re.findall(r'[a-zA-Z0-9]+', s)
+        for word in words:
+            worddict[word] = worddict[word] + 1 if word in worddict else 1
+    wordsort = sorted(worddict.items(), key=lambda e:e[1], reverse=True)
+    return wordsort
+
+if __name__ == '__main__':
+    files = get_files('.')
+    print files
+    print get_important_word(files)
diff --git a/DIYgod/0006/test/test.txt b/DIYgod/0006/test/test.txt
@@ -0,0 +1,2 @@
+important important important important important important important important important important important important important important important important important important important 
+

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+important important important important important important important important important important important important important important important important important important important`
	`2`	`+`