#!/usr/bin/python
# -*- coding: UTF-8 -*-
import os
import codecs
import datetime
import msvcrt

# 待检测目录
check_dir_path = '.\代码目录'


# 写日志
def write_log(text):
    log_file = '.\检测结果.txt'
    with open(log_file, 'a+', encoding='UTF-8') as file:
        prefix = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") + ' '
        file.write(prefix + text + '\n')


# 主程序
def main():
    if not os.path.exists(check_dir_path):
        # 创建文件夹
        os.mkdir(check_dir_path)
        printout = '请把需要检测文件放置在代码目录下'
        write_log(printout)
        print(printout)
        return False
    # 定义文件数
    files = 0
    # 定义行数
    total_lines = 0
    for dir_path, dir_names, filenames in os.walk(check_dir_path):
        paths = [os.path.join(dir_path, names) for names in filenames]
        for path in paths:
            try:
                in_lines = 0
                with codecs.open(path, encoding='UTF-8') as infile:
                    lines = len(infile.readlines())
                    total_lines += lines
                    in_lines += lines
                printout = '检测成功：%s 共%s行' % (path, in_lines)
                write_log(printout)
                print(printout)
                files += 1
            except:
                printout = '检测异常：%s' % path
                write_log(printout)
                print(printout)
    printout = '检测结果如下：'
    write_log(printout)
    print(printout)
    printout = '共检测到%s个文件，共%s行' % (files, total_lines)
    write_log(printout)
    print(printout)


if __name__ == '__main__':
    # 执行程序
    main()
    # 按任意键退出
    print("脚本运行结束，结果已导出，按任意键退出。")
    ord(msvcrt.getch())