avatar

利用 Python + Flask 搭建软件更新服务

我也想给自己搞一个软件更新服务

说到软件更新相信大家肯定都不止一次两次见过了吧。在手机上,经常可以看到某某助手或者应用中心提示应用更新,而且会很醒目地告诉你:可以节约多少多少流量。在大家各自的PC上面也不止一次看到这种提示吧。

“检查到新版本x.x.x,当前版本是x.x.x,是否更新”

PC软件提示更新

Android软件提示更新

流程是怎么样的

这种软件更新是怎么做到的呢?
咱不知道大公司是怎么想的
抛开Android的不说,咱们本次讨论一下PC的软件在我的思路中,是如何做到的,怎么样?
其实,流程是十分简单的,大家也是一想就能明白。

我想是这样的

流程如下:

首先,咱们得先有一部远程服务器吧。用来存放软件更新的文件以及清单文件,以及搭建一个服务用以软件检测是否存在有新版本和下载新文件用
接着软件客户端,就是你的PC上面要有个服务去检测发现版本更新信息。可以在启动的时候,也可以在你点击软件检查更新的时候去检查版本信息。
点击更新后,下载清单文件,然后再比较一下清单文件和本地文件的差异,可以检测一下文件的MD5什么的。如果不一样的就下载更新,当然如果是不存在的,就更加得下载下来啦。
最后就是……你得有网,Internet,晓得吧,木得Internet谈自动更新(增量更新)都是“耍流氓”…

画个流程图看看

流程图如下:

最最最简单的流程

OK ,那么这个这么简单的流程咱们搞清楚了吗?

服务器得做什么好呢

服务器的任务

清楚后,咱们分析一下我们服务器担任的角色。我们今天先不管客户端啦(虽然考虑的时候应该一起考虑,小声bb)
服务器要起到几个作用:
1、提供网络服务,就是得有简单文件下载服务和应答服务。
2、得有个文件夹存放更新文件,里面放更新文件和清单文件。
3、在网络中任意一个位置都可以访问到,而不仅限于自己的那个路由器……

所以,我们瞬间就可以想起了Python中自带的那个网络服务。。。但是,用过的人都知道它实在是太简单,难以胜任我们的这个需求啊。所以,这个时候,我们可以选用Flask这个小辣椒啦。什么?你不知道Flask是什么?

Flask是个服务框架

就是这个火药筒
简单普及一波:其实Flask那个图片不是辣椒啦,人家是装火药的啦
基于他,我们可以很容易地快速搭建好我们自己定义的服务。
再扯一下,单纯用这个框架简单地写个服务还是可以能应付一般点的并发量的。。。太大的并发量的话,就得采用其他策略去对付了。所以,我们现在讨论的是不考虑高并发情况的。

想想代码的函数名和它该干嘛

下面我就简单解释一下各函数的写法思路吧。

download,接收filename的参数,表示下载的文件名。获取后,通过flak的内置方法返回所需下载文件即可
Getfile_md5,接收文件路径作为参数,通过文件路径读取文件,然后通过hashlib计算MD5然后返回
generate,没有参数,用于更新清单文件。(虽然感觉这个做法有点多余,因为可以多写一个独立的文件处理,管理员上传完文件后再执行一次就OK了D)
findFile,接收目录路径作为参数,递归查找目标目录里面的文件,如果是文件则计算MD5是目录则递归进去寻找文件。
check,没有参数,返回当前服务器版本号以及更新信息。

函数的目的就是这么简单。简简单单几十行就可以把这个服务给实现出来了。后面PC客户端在请求是否有新版本更新的时候,就可以用这个服务脚本来处理了。
当然,这里面还是有挺多没考虑到的问题的。比如被人家恶意下载导致服务器流量快速被消耗,高并发的状态下,它是否能顶的住?

Don’t BB , show me the code !

代码我直接就放上来吧。注释都有写作用的。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
# -*- coding: utf-8 -*-
# @Time : 4/1/2019 19:27
# @Author : MARX·CBR
# @File : __init__.py.py
import pickle

from flask import request, jsonify, send_from_directory, abort, Flask
import os
import hashlib
import json

app = Flask(__name__)
allfile = []
md5_list = []
updateList = {}
directory = os.getcwd()


# 下载文件服务
@app.route("/<filename>", methods=['GET'])
def download(filename):
if request.method == "GET":
if os.path.isfile(os.path.join('updateFiles', filename)):
return send_from_directory('updateFiles', filename, as_attachment=True)
abort(404)


# 计算文件MD5
def Getfile_md5(filename):
if not os.path.isfile(filename):
return
myHash = hashlib.md5()
f = open(filename, 'rb')
while True:
b = f.read(8096)
if not b:
break
myHash.update(b)
f.close()
return myHash.hexdigest()


# 计算生成新的清单文件
@app.route("/generateNewConfig", methods=['GET'])
def generate():
findFile(directory + '/updateFiles/')
file_md5_list = json.dumps(updateList)
print(file_md5_list)
with open('./updateFiles/listFile', 'wb') as f:
pickle.dump(updateList, f)
return_data = {
'Statu': 'success',
}
return jsonify(return_data)
# file_md5_list=json.load(updateList)


# 找到更新文件目录里面的文件以及文件夹、递归寻找
def findFile(path):
fsinfo = os.listdir(path)
for fn in fsinfo:
temp_path = os.path.join(path, fn)
if not os.path.isdir(temp_path):
print('文件路径: {}'.format(temp_path))
fm = Getfile_md5(temp_path)
print(fn)
fn = temp_path.replace(directory + "/updateFiles/", '')
updateList[fn] = fm
else:
findFile(temp_path)


# 检查更新版本,该部分尚未够,完善。可以考虑为管理员远程上传文件的时候
# 将更新说明以json格式一同上传到服务器中,更新时直接读取即可
@app.route("/checkUpdate", methods=['GET'])
def check():
if request.method == "GET":
return_data = {
'Version': '0.0.1',
'Msg': '更新文件,修复初始化卡顿bug\n增加文件预下载功能',
}
return jsonify(return_data)


# 首页Hello
@app.route("/", methods=['GET'])
def hello():
if request.method == "GET":
return "Hello MARXCBR"


if __name__ == "__main__":
app.run(host='0.0.0.0', port=8080) # 运行,指定监听地址为 0.0.0.0:8080
# 服务器端运行可以让所有地方访问到

大概就这样吧,后续会给出软件更新服务之客户端怎么玩的文章。咱们试试从头到尾自己实现一个简单的软件更新服务,哈哈哈。

要是感觉不错的话,就留个言评论一下下呀,欢迎各位大哥指正
文章作者: MarxCBR
文章链接: https://www.marxcbr.cn/archives/296eec40.html
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 MarxCBR的小屋
打赏
  • 微信
    微信
  • 支付宝
    支付宝

评论
2