update: add search results output to file

Author: repoog

README.md

@@ -41,7 +41,7 @@ USAGE:
 * v2.0 更新搜索设计和算法
 * v2.1 更新搜索结果输出展现
 * v2.2 优化部分代码，增加项目搜索进度条，解决代码输出BUG
-* v2.4 优化程序目录设计，优化源码实现，删除冗余代码
+* v2.4 优化程序目录设计，优化源码实现，增加默认文件输出
 
 ***
 ## Sensitive info scan tool of Github

config/Config.py

@@ -4,6 +4,9 @@
 FILE_DB = "pattern/file.db"
 INFO_DB = "pattern/info.db"
 
+# Output file default name
+OUTPUT_FILE = "gitprey.txt"
+
 # GitHub account config for searching
 USER_NAME = "repoog"
-PASSWORD = ""
+PASSWORD = "/*F0r0p3nS0urc3*/"

include/ColorPrint.py

@@ -1,6 +1,11 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
+try:
+    from config.Config import *
+except ImportError:
+    print "[!_!]ERROR INFO: Can't find Config file for searching."
+    exit()
 
 def error_print(string):
     # Print error information with red color
@@ -15,17 +20,29 @@ def info_print(string):
 def project_print(string):
     # Print project information with deep green color
     print "\033[1;36;40m%s\033[0m" % string
+    __output_result(string)
 
 
 def file_print(string):
     # Print file url with yellow color
     print "\033[1;33;40m%s\033[0m" % string
+    __output_result(string)
 
 
 def code_print(string):
     # Print code line with white color
     print "\033[1;37;40m%s\033[0m" % string
+    __output_result(string)
 
 
+def __output_result(string):
+    """
+    Output search results to text file
+    :param results: search results
+    :return: None
+    """
+    with open(OUTPUT_FILE, 'a+') as file_obj:
+        file_obj.write(string + '\r\n')
+
 if __name__ == "__main__":
     pass