交作业了，简单的文件加密解密脚本

真空拔罐器

简单的文件加密解密，鱼油们多多指点！

真空拔罐器

优化了一下~

1. import os
   
2. import easygui as g
   
3. import random as r
   
4. import pickle as p
   
5. import sys
   
6. 
   
7. #定义一个文件类
   
8. class File:
   
9.     dict = {}
   
10.     def __init__(self,path):
    
11.         self.dir = os.path.dirname(path)
    
12.         self.name = os.path.splitext(os.path.basename(path))[0]
    
13.         self.ext = os.path.splitext(os.path.basename(path))[1]
    
14.         self.newname = ''.join(r.sample('qwertyuiopasdfghjklzxcvbnm&^$@#%1234567890',8))#改变文件名，障眼法
    
15.     def getDict(self):
    
16.         return self.dict     
    
17.     def setDict(self,ext):#将相关信息存入字典便于索引
    
18.         self.dict = {'dir' : self.dir,'name' : self.name,'ext' : self.ext,'newname' : self.newname,'newext' : ext}
    
19.     def delDict(self):
    
20.         del self.dict
    
21.     x = property(getDict,setDict,delDict)
    
22. def en_decryption(path):#加解密函数
    
23.     with open(path,'rb+') as f:
    
24.         old = f.read()
    
25.         new = old[::-1]      
    
26.         f.seek(0)
    
27.         f.write(new)#反转文件使其无法被识别,同样操作即可解密
    
28. try:
    
29.     with open('table.pkl','rb') as s:
    
30.         filelist = list(p.load(s))
    
31. except EOFError:
    
32.     filelist = []
    
33. if g.passwordbox('请输入密码','文件加密解密') == '12345nlx':
    
34.     while True:
    
35.         if g.boolbox('选择要进行的操作','文件加密',choices = ('加密','解密')):
    
36.             try:
    
37.                 path = g.fileopenbox('请选择文件','文件加密')
    
38.                 f = File(path)
    
39.                 os.chdir(f.dir)
    
40.             except TypeError:
    
41.                 break
    
42.             f.x = g.choicebox('请选择加密文件的拓展名','文件加密',choices = ('.txt','.mp4','.DLLs','.Doc','.mp3'))
    
43.             if f.x == None:#选择拓展名伪装
    
44.                 break
    
45.             else:
    
46.                 c = '判断'
    
47.                 for each in filelist:#判断文件是否已加密
    
48.                     if f.x['name'] == each['newname']:
    
49.                         c = g.choicebox('文件已加密，是否重新选择文件？','文件加密',choices = ('是','否'))
    
50.                         break
    
51.                 if c == '是':
    
52.                     continue
    
53.                 elif c == '否':
    
54.                     break           
    
55.                 elif c == '判断':
    
56.                     g.msgbox('文件加密中......','文件加密')
    
57.                     en_decryption(path)#加密
    
58.                     os.rename(os.path.basename(path),f.x['newname'] + f.x['newext'])
    
59.                     os.chdir(sys.path[0])
    
60.                     filelist.append(f.dict)#将加密文件存入已加密列表
    
61.                 with open('table.pkl','wb')  as s:
    
62.                     p.dump(filelist,s)#存入pickle
    
63.                 if g.boolbox('加密完成，是否退出程序','文件加密'):
    
64.                     break
    
65.         else :
    
66.             if filelist == []:
    
67.                 g.textbox('提示','文件解密',text = '未找到加密文件')
    
68.                 if  g.boolbox('是否退出程序','文件解密'):
    
69.                     break
    
70.                 else:
    
71.                     continue
    
72.             else:
    
73.                 try:
    
74.                     for each in g.multchoicebox('搜索到已加密文件如下，请选择要解密的文件','文件解密',choices = filelist):
    
75.                         for file in filelist:
    
76.                             if each == str(file):
    
77.                                 g.textbox('文件解密','文件解密',text = '文件%s解密中，请不要退出程序......'%(file['name'] + file['ext']))
    
78.                                 os.chdir(file['dir'])
    
79.                                 try:
    
80.                                     en_decryption(file['newname'] + file['newext'])#解密
    
81.                                     os.rename(file['newname'] + file['newext'],file['name'] + file['ext'])#重命名
    
82.                                     
    
83.                                     filelist.remove(file)#解密后将文件从已加密列表中删除
    
84.                                 except FileNotFoundError:
    
85.                                     g.msgbox('检测到文件路径已改变，请自行选择加密文件','文件解密')#路径检测
    
86.                                     try:
    
87.                                         path = g.fileopenbox('请选择文件','文件解密')
    
88.                                         f = File(path)
    
89.                                         for file in filelist:
    
90.                                             if f.name == file['newname']:
    
91.                                                 os.chdir(f.dir)
    
92.                                                 g.textbox('文件解密','文件解密',text = '文件%s解密中，请不要退出程序......'%(file['name'] + file['ext']))
    
93.                                                 en_decryption(file['newname'] + file['newext'])#解密
    
94.                                                 os.rename(file['newname'] + file['newext'],file['name'] + file['ext'])#重命名
    
95.                                                 filelist.remove(file)
    
96.                                     except TypeError:
    
97.                                         break
    
98.                                 break
    
99.                 except TypeError:
    
100.                     break
     
101.                 os.chdir(sys.path[0])
     
102.                 with open('table.pkl','wb')  as s:
     
103.                     p.dump(filelist,s)#存入pickle
     
104.                 if g.boolbox('解密完成，是否退出程序','文件解密'):
     
105.                     break
     
106. else:
     
107.     g.msgbox('密码错误！程序退出','文件加密解密')
     
108. g.msgbox('谢谢使用','文件加密解密')
     
109.    
     
110. 
     
111.    
     
112.    
     
113. 
     

复制代码