#-*-coding:utf-8-*-
'''
创建记事本，编写内容，查看内容，格式化笔记本，备份与恢复本地记事本，日志内容过滤
write()函数
writelines()函数
os 函数
copyfile()和move()函数
replace()函数   替换字符串
mkdir()函数      创建文件目录
makedirs()函数   可以创建文件多级目录
rmdir()函数       删除空目录
rmtree()函数      删除非空目录
'''
import os
import shutil
import re
import time
print('=============欢迎使用窗内网记事本=============')
con = True
while con:
    k = int(input('请输入您的操作\n1.【我要写日记】\n2.【查看往事】\n3.【格式化】\n4【退出】\n'))
    if (k == 1):
        times = 'd:\\ProjectFile\\'+time.strftime('%Y-%m-%d',time.localtime())
        if os.path.exists(times) == False:
            os.makedirs(times)
        files = times + '\mylog.log'
        o = open(files,'a+')
        content = input('请输入您需要记录的事情：\n')
        count = 0;
        for s in o.readlines():
            li = re.findall('窗内网',s)
            if len(li)>0:
                count=count+li.count('窗内网')
        th = int(input('查找到'+str(count)+'个可能替换的内容是否继续？\n确定输入1：，取消输入2：\n'))
        if th == 1:
            content = content.replace('窗内网','窗内网(http://www.itzcn.com)')
        o.write(content)
        o.close()
        print('=====================')
    elif(k == 2):
        fls = input('请输入记事本分类名称：\n')
        if os.path.exists('D:\ProjectFile\\'+fls+'\mylog.log'):
            print("日志内容：\n")
            o = open('D:\ProjectFile\mylog.log','a+')
            listcontent = o.readlines()
            for content in listcontent:
                print(content)
            o.close()
        else:
            print('分类名称不存在！')
        print('======================')
    elif(k == 3):
        gsh = int(input('您确定要格式化记事本吗？格式化后数据将会全部消失\n确定输入：1，取消输入2：\n'))
        if (gsh == 1):
            fls = input('请输入格式化记事本分类名称：\n')
            print('记事本正在格式化中......')
            if os.path.exists('D:\ProjectFile\mylog.log'):
                os.remove('D:\ProjectFile\mylog.log');
                print('记事本格式化成功')
                open('D:\ProjectFile\mylog.log','a+')
            else:
                print('记事本不存在！')
                print('==============================')
    elif(k == 4):
        bf = int(input('您确定要备份记事本吗？\n确定输入：1.取消输入2：\n'))
        if (bf == 1):
            fls = input('请输入备份记事本分类名称：\n')
            print('记事本正在备份中......')
            if os.path.exists('D:\\ProjectFile\\'+fls+'\mylog.log'):
                os.makedirs('D:\\ProjectFile\\'+fls+'\\bf')
                shutil.copyfile('D:\\ProjectFile\\mylog.log','D:\\ProjectFile\\bf_mylog.log')
                shutil.move('D:\\ProjectFile\\bf_mylog.log','D:\\ProjectFile\\bf_mylog.log')
                print('备份成功')
            else:
                print('备份记事本分类不存在！')
    elif(k == 5):
        hf = int(input('您确定要恢复记事本吗\n确定输入：1，取消输入2：\n'))
        if (hf == 1):
            print('记事本正在恢复中......')
            shutil.copyfile('D:\\ProjectFile\\bf_mylog.log','D:\\ProjectFile\\mylog.log')
    elif(k == 6):
        print('日记分类信息：')
        def ListDir(path,fun,par):
            for filespath in par:
                print(os.path.join(fun,filespath).strip('D:\ProjectFile'))
        if __name__ == '__main__':
            os.path.walk('D:\ProjectFile',ListDir,())
    else:
        print('欢迎下次使用')
        con=False