import base64

 import os

 def jiami():

     load=input('输入加密的txt文件的路径：')

     #将路径文件加密保存

     #对应utf-8的base64编码

     f=open(load,encoding='utf-8')

     #这里要使用‘utf-8’格式打开，否则会报错

     content=f.read()

     content1=content.encode(encoding='utf-8')

     content2=base64.b64encode(content1)

     #这一句，将字符串进行加密

     f.close()

     #下面以写入的方式将content2写入原文本，并且替换

     with open(load, 'wb+') as f:

     #以二进制的方式打开

     #bytes类型数据需要以二进制形式打开写入

        f.write(content2)

     print('加密成功')

 def jiemi():

     load1=input('输入解密的txt文件的路径：')

     #将路径文件解密

     f=open(load1,encoding='utf-8')

     content=f.read()

     content1=base64.b64decode(content)

     with open(load1, 'wb+') as f:

         f.write(content1)

     print('解密成功')

 def choice():

     num=input('加密文本输入1/解密文本输入2：')

     if int(num) == 1:

         jiami()

     elif int(num) == 2:

         jiemi()

     else:

         print('输入错误')

 while 1==1:

     if __name__ == '__main__':

         choice()