python 程序自动升级_python3--Linux文件自动更新小程序

1 #!/usr/bin/python3

2 #coding = utf-8

3

4 #author = warlock921

5

6 importos7 importsys8 importtime9 importparamiko10

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 defprint_point():22

23 for i in range(100):24 percent = 1.0*i/100*100

25 print('完成进度：\033[1;31m %.2s %s \033[0m'%(str(percent),'%'),end='\r')26 time.sleep(0.08)27

28

29 defget_foward_dir():30 returnos.getcwd()31

32 #用于读取更新文档目录的方法

33 defread_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 ifvision_flag :48 print("\033[1;31m 注意： \033[0mUpdateFile文件夹下不存在可更新文件！按回车键退出！")49 #print(dirlist)

50 input("")51 exit()52

53 dirlist_length =len(dirlist)54 #print(dirlist_length)

55 list_num =056 #使用循环取出需要更新的目录路径及文件名称列表

57 while list_num <58>

59 update_dir = dirlist[list_num][0][1::]60

61 ifvision_flag:62 print('-'*50)63 print("需要更新的目录%d：\033[4;31m %s \033[0m"%((list_num+1),(ServerDir+update_dir+os.sep)))64

65 update_file = dirlist[list_num][-1]66 ifvision_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 returnpath_file_list272

73 #用于显示已更新过的文档目录

74 defread_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 =083 while list_num2 <84 update_dir2="dirlist2[list_num2][0][1::]85" print update_file2="dirlist2[list_num2][-1]88" list_num2>

90

91

92 defread_bak_dir():93 pass

94

95 defdo_update_file():96 vision_flag =False97 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 inpath_file_list.items():101 print("已更新：%s \033[1;33m %s \033[0m"%(("..."+path_dir[13::]),path_file))102 time.sleep(0.5)103

104 defdo_backup_file():105 print_point()106 print("\033[1;33m 备份服务器文件完成！ \033[0m")107

108 #def into_server_need2update_file(need_update_path,need_update_file):

109 #pass

110

111 defrestart_server():112 pass

113

114 defmain():115 #update_flag用于表示文件是否已经更新过

116 update_flag =True117 #vision_flag用于表示是否显示函数里的打印语句

118 vision_flag =True119

120 list_x =[]121

122 #定义当前目录路径

123 forward_dir =get_foward_dir()124

125 #循环遍历当前文件夹,并将当前目录下的文件夹(不包括文件)压入列表

126 for x inos.listdir():127 ifos.path.isdir(x):128 list_x.append(x)129

130 #判断是否存在UpdateFile文件

131 if ServerUpdateFileDir inlist_x:132 #print("命中\033[1;31m %s \033[0m文件夹！"%ServerUpdateFileDir)

133 #操作更换工作目录，进入需要更新的文件夹

134 os.chdir(("."+os.sep+ServerUpdateFileDir))135 else:136 #如果不存在UpdateFile文件，则让用户选择是否建立该文件夹

137 print("\033[1;31m UpdateFile文件夹不存在，请先建立UpdateFile文件夹，否则无法更新！ \033[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 whileTrue:153

154 os.system('clear')155 print("**************** 更新程序 v1.01 *****************")156 #字体打印颜色变更语法：\033[1;33;40m %s \033[0m

157 print("命中\033[1;31m %s \033[0m文件夹！"%ServerUpdateFileDir)158 print(("当前工作目录：\033[1;33;40m %s \033[0m")%forward_dir)159 print("当前工作的文件夹：%s"%get_foward_dir())160

161 #判断是否已经更新过

162 ifupdate_flag:163 #未更新则执行下面语句

164 read_update_dir(vision_flag)165 print('*'*50)166 update_usr_input = input("\033[1;31m 注意 \033[0m是否开始更新文件：\n 1.开始更新 \n 2.不更新只备份文件 \n 3.退出更新程序 \n 请选择(数字键1/2/3，默认3)：")167 if update_usr_input == "1":168 print("\033[1;31m 开始更新，请勿进行其他操作... \033[0m")169

170 #先备份

171 do_backup_file()172 #后更新

173 do_update_file()174

175 print("更新成功完成！")176 update_flag =False177 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("\033[1;31m 注意： \033[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()

84>58>