当前位置: 澳门新濠3559 > 操作系统 > 正文

python3--Linux文件自动更新小程序

时间:2019-11-16 22:36来源:操作系统
1 #!/usr/bin/python3 2 #coding = utf-8 3 4 #author = warlock921 5 6 import os 7 import sys 8 import time 9 import paramiko 10 11 ServerDir='' 12 ServerUpdateFileDir = 'UpdateFile' 13 ServerBakDir = "BackupFile" 14 15 if os.name == 'posix':
  1 #!/usr/bin/python3
  2 #coding = utf-8
  3 
  4 #author = warlock921
  5 
  6 import os
  7 import sys
  8 import time
  9 import paramiko
 10 
 11 ServerDir=''
 12 ServerUpdateFileDir = 'UpdateFile'
 13 ServerBakDir = "BackupFile"
 14 
 15 if os.name == 'posix':
 16         ServerDir = "你需要更新的文件根目录"
 17 elif os.name == 'nt':
 18         ServerDir = "你需要更新的文件根目录"
 19         os.sep = '/'
 20 
 21 def print_point():
 22 
 23     for i in range(100):
 24         percent = 1.0*i/100*100
 25         print('完成进度:33[1;31m %.2s %s 33[0m'%(str(percent),'%'),end='r')
 26         time.sleep(0.08)
 27 
 28 
 29 def get_foward_dir():
 30     return os.getcwd()
 31 
 32 #用于读取更新文档目录的方法
 33 def read_update_dir(vision_flag):
 34     dirlist = []
 35     path_file_list1 = {}
 36     path_file_list2 = {}
 37 
 38     #定义需要更新的目录及文件列表
 39     #使用os.walk()函数实现遍历当前路径下的所有目录结构,输出为列表,列表值为tuple(元组)
 40     for i in os.walk("."):
 41         #print(dirlist)
 42         #将非空的目录及非顶层的目录保存到列表中
 43         if i[-1] != [] and i[0] != ".":
 44             dirlist.append(i)
 45             #print(dirlist)
 46     if dirlist == []:
 47         if vision_flag :
 48             print("33[1;31m 注意: 33[0mUpdateFile文件夹下不存在可更新文件!按回车键退出!")
 49         #print(dirlist)
 50         input("")
 51         exit()
 52 
 53     dirlist_length = len(dirlist)
 54     #print(dirlist_length)
 55     list_num = 0
 56     #使用循环取出需要更新的目录路径及文件名称列表
 57     while list_num < dirlist_length:
 58 
 59         update_dir = dirlist[list_num][0][1::]
 60         
 61         if vision_flag:
 62             print('-'*50)
 63             print("需要更新的目录%d:33[4;31m %s 33[0m"%((list_num+1),(ServerDir+update_dir+os.sep)))
 64 
 65         update_file = dirlist[list_num][-1]
 66         if vision_flag:
 67             print("需要更新的文件列表%d:%s"%((list_num+1),update_file))
 68         list_num+=1
 69         #这里使用了Python3中字典的特性,将每次循环取出来的字典,再update进一个新的字典,这样才能在字典中保存另一个字典
 70         path_file_list2.update(path_file_list1.fromkeys(update_file,(ServerDir+update_dir+os.sep)))
 71     return path_file_list2
 72 
 73 #用于显示已更新过的文档目录
 74 def read_updated_dir():
 75     dirlist2 = []
 76     for j in os.walk("."):
 77         if j[-1] != [] and j[0] != ".":
 78             dirlist2.append(j)
 79             #print(dirlist2)
 80     dirlist2_length = len(dirlist2)
 81     #print(dirlist_length)
 82     list_num2 = 0
 83     while list_num2 < dirlist2_length:
 84         update_dir2 = dirlist2[list_num2][0][1::]
 85         print('-'*50)
 86         print("已更新的目录%d:33[1;33m %s 33[0m"%((list_num2+1),(ServerDir+update_dir2)))
 87         update_file2 = dirlist2[list_num2][-1]
 88         print("已更新的文件列表%d:33[1;31m %s 33[0m"%((list_num2+1),update_file2))
 89         list_num2+=1
 90 
 91 
 92 def read_bak_dir():
 93     pass
 94 
 95 def do_update_file():
 96     vision_flag = False
 97     path_file_list=read_update_dir(vision_flag)
 98     #pfl_len=len(path_file_list)
 99     #遍历字典path_file_list,取出其中的key,value用于显示和更新
100     for path_file,path_dir in path_file_list.items():
101         print(" 已更新:%s 33[1;33m %s 33[0m"%(("..."+path_dir[13::]),path_file))
102         time.sleep(0.5)
103 
104 def do_backup_file():
105     print_point()
106     print("33[1;33m 备份服务器文件完成! 33[0m")
107 
108 #def into_server_need2update_file(need_update_path,need_update_file):
109     #pass
110 
111 def restart_server():
112     pass
113 
114 def main():
115     #update_flag用于表示文件是否已经更新过
116     update_flag = True
117     #vision_flag用于表示是否显示函数里的打印语句
118     vision_flag = True
119 
120     list_x = []
121 
122     #定义当前目录路径
123     forward_dir = get_foward_dir()
124 
125     #循环遍历当前文件夹,并将当前目录下的文件夹(不包括文件)压入列表
126     for x in os.listdir():
127         if os.path.isdir(x):
128             list_x.append(x)
129 
130     #判断是否存在UpdateFile文件
131     if ServerUpdateFileDir in list_x:
132         #print("命中33[1;31m %s 33[0m文件夹!"%ServerUpdateFileDir)
133         #操作更换工作目录,进入需要更新的文件夹
134         os.chdir(("."+os.sep+ServerUpdateFileDir))
135     else:
136         #如果不存在UpdateFile文件,则让用户选择是否建立该文件夹
137         print("33[1;31m UpdateFile文件夹不存在,请先建立UpdateFile文件夹,否则无法更新! 33[0m")
138         mkdir_input=input("是否建立UpdateFile文件夹?(y/n):")
139         if mkdir_input == 'y' or mkdir_input == 'Y':
140             os.system('mkdir UpdateFile')
141             print("已经建立UpdateFile文件夹!")
142             os.chdir(("."+os.sep+ServerUpdateFileDir))
143             input('')
144         elif mkdir_input == 'n' or mkdir_input == 'N':
145             print("不建立UpdateFile文件夹,无法进行更新,系统退出!")
146             input('')
147             exit()
148         else:
149             pass
150     
151     #开始循环,更新程序主程序
152     while True:
153 
154         os.system('clear')
155         print("****************  更新程序  v1.01   *****************")
156         #字体打印颜色变更语法:33[1;33;40m %s 33[0m
157         print("命中33[1;31m %s 33[0m文件夹!"%ServerUpdateFileDir)
158         print(("当前工作目录:33[1;33;40m %s 33[0m")%forward_dir)
159         print("当前工作的文件夹:%s"%get_foward_dir())
160 
161         #判断是否已经更新过
162         if update_flag:
163             #未更新则执行下面语句
164             read_update_dir(vision_flag)
165             print('*'*50)
166             update_usr_input = input("33[1;31m 注意 33[0m是否开始更新文件:n 1.开始更新 n 2.不更新只备份文件 n 3.退出更新程序 n 请选择(数字键1/2/3,默认3):")
167             if update_usr_input == "1":
168                 print("33[1;31m 开始更新,请勿进行其他操作... 33[0m")
169                 
170                 #先备份
171                 do_backup_file()
172                 #后更新
173                 do_update_file()
174 
175                 print("更新成功完成!")
176                 update_flag = False
177                 input("")
178                 continue
179             elif update_usr_input == "2":
180                 do_backup_file()
181                 input("")
182                 continue
183             elif update_usr_input == "3":
184                 os.system('clear')
185                 print("系统退出!")
186                 exit()
187             else:
188                 os.system('clear')
189                 print("系统退出!")
190                 exit()
191         else:
192             #已更新过则执行下面的代码
193             print('*'*50)
194             update_usr_input = input("33[1;31m 注意: 33[0m网站已经更新完成:n 1.查看更新过的目录 n 2.退出更新程序 n 请选择(数字键1/2,默认2):")
195             if update_usr_input == "1":
196                 read_updated_dir()
197                 input("")
198                 continue
199             elif update_usr_input == "2":
200                 os.system('clear')
201                 print("系统退出!")
202                 exit()
203             else:
204                 os.system('clear')
205                 print("系统退出!")
206                 exit()
207 
208 
209 if __name__ == '__main__':
210     main()

 

编辑:操作系统 本文来源:python3--Linux文件自动更新小程序

关键词: