利用阿里云官方python版SDK和刷新预热自动化脚本来打造一个不需要python环境的阿里云CDN刷新工具。
安装python,本人安装的是3.6.5,接下来使用下面的安装阿里云SDK和python打包工具pyinstaller:
pip install aliyun-python-sdk-core
pip install aliyun-python-sdk-cdn
pip install pyinstaller
将下面的代码保存为cdntool.py,代码来自阿里云官方刷新预热自动化脚本,已经修改适配python3并加入支持单条URL刷新:
# coding=utf-8
# __author__ = 'hanli.zyb'
'''Check Package'''
try:
import os
import sys
import getopt
import time
import json
from aliyunsdkcore.client import AcsClient
from aliyunsdkcore.acs_exception.exceptions import ClientException
from aliyunsdkcore.acs_exception.exceptions import ServerException
from aliyunsdkcdn.request.v20180510.RefreshObjectCachesRequest import RefreshObjectCachesRequest
from aliyunsdkcdn.request.v20180510.PushObjectCacheRequest import PushObjectCacheRequest
from aliyunsdkcdn.request.v20180510.DescribeRefreshTasksRequest import DescribeRefreshTasksRequest
from aliyunsdkcdn.request.v20180510.DescribeRefreshQuotaRequest import DescribeRefreshQuotaRequest
except Exception as e:
print(e)
sys.exit("[Error] Please pip install aliyun-python-sdk-cdn and aliyun-python-sdk-core ,please install now......")
class Refresh(object):
'''init func'''
def __init__(self):
self.lists = []
self.param = {}
'''
描述:调度的主函数
resP:检测入参结果,如果类型不是 bool 说明有报错
'''
def main(self, argv):
if len(argv) < 1:
sys.exit("\nusage: " + sys.argv[0] + " -h ")
try:
opts, _args = getopt.getopt(argv, "hi:k:n:r:u:t:a:o:")
except Exception as _e:
sys.exit("\nusage: " + sys.argv[0] + " -h ")
for opt, arg in opts:
if opt == '-h':
self.helps()
sys.exit()
elif opt == '-i':
self.param['-i'] = arg
elif opt == '-k':
self.param['-k'] = arg
elif opt == '-r':
self.param['-r'] = arg
elif opt == '-u':
self.param['-u'] = arg
elif opt == '-t':
self.param['-t'] = arg
elif opt == '-a':
self.param['-a'] = arg
elif opt == '-o':
self.param['-o'] = arg
elif opt == '-n':
self.param['-n'] = arg
else:
sys.exit("\nusage: " + sys.argv[0] + " -h ")
resP = self.doCheck(self.param)
if not isinstance(resP, bool):
sys.exit(resP)
try:
client = AcsClient(
self.param['-i'], self.param['-k'], 'cn-hangzhou')
except NameError:
sys.exit("[Error]: SDK module not detected")
for g in self.doProd(self.param):
self.lists = []
self.doRefresh(''.join(g), self.param['-t'], client)
'''
描述:检测入参数
'''
def doCheck(self, param):
try:
for key1 in ('-i', '-k', '-t'):
if not key1 in param.keys():
return "[Error]: {0} Must be by parameter".format(key1)
if not '-r' in param.keys() and not '-u' in param.keys():
return "[Error]: -r or -u Must be by parameter"
try:
if not '-n' in param:
self.param['-n'] = 50
if not (abs(int(param['-n'])) <= 100 and abs(int(param['-n'])) > 0):
return "[Error]: 0 < -n <= 100"
else:
self.param['-n'] = int(param['-n'])
except ValueError as e:
return "[Error]: -n Must be int Type ,{0}".format(str(e))
if not param['-t'] in ("push", "clear"):
return "[Error]: taskType Error"
if '-a' in param and '-o' in param:
return "[Error]: -a and -o cannot exist at same time"
if '-a' in param:
if not param['-a'] in ("domestic", "overseas"):
return "[Error]: Area value Error"
if param['-t'] == 'clear':
return "[Error]: -t must be push and 'clear' -o use together"
if '-o' in param:
if not param['-o'] in ("File", "Directory"):
return "[Error]: ObjectType value Error"
if param['-t'] == 'push':
return "[Error]: -t must be clear and 'push' -a use together"
except KeyError as e:
return "[Error]: Parameter {0} error".format(str(e))
return True
'''
描述:生成器切分文件,对每行文件进行处理 '\n'
gop:每次读取 URL 数量
'''
def doProd(self, params):
gop = params['-n']
mins = 1
maxs = gop
if '-r' in params:
with open(params['-r'], "r") as f:
for line in f.readlines():
if mins != maxs:
line = line.strip("\n") + "\n"
else:
line = line.strip("\n")
self.lists.append(line)
if mins >= maxs:
yield self.lists
mins = maxs
maxs = gop + maxs - 1
else:
mins += 1
if len(self.lists) > 0:
yield self.lists
if '-u' in params:
self.lists.append(params['-u'])
yield self.lists
'''
描述:刷新/预热任务
'''
def doRefresh(self, lists, types, client):
try:
if types == 'clear':
taskID = 'RefreshTaskId'
request = RefreshObjectCachesRequest()
if '-o' in self.param:
request.set_ObjectType(self.param['-o'])
elif types == 'push':
taskID = 'PushTaskId'
request = PushObjectCacheRequest()
if '-a' in self.param:
request.set_Area(self.param['-a'])
taskreq = DescribeRefreshTasksRequest()
request.set_accept_format('json')
request.set_ObjectPath(lists)
response = json.loads(client.do_action_with_exception(request))
print(response)
while True:
count = 0
taskreq.set_accept_format('json')
taskreq.set_TaskId(int(response[taskID]))
taskresp = json.loads(client.do_action_with_exception(taskreq))
print("[" + response[taskID] + "]" + "is doing... ...")
for t in taskresp['Tasks']['CDNTask']:
if t['Status'] != 'Complete':
count += 1
if count == 0:
break
else:
continue
time.sleep(5)
except Exception as e:
sys.exit("[Error]" + str(e))
'''
描述:帮助信息
'''
def helps(self):
print("\nscript options explain: \
\n\t -i <AccessKey> 访问阿里云凭证,访问控制台上可以获得; \
\n\t -k <AccessKeySecret> 访问阿里云密钥,访问控制台上可以获得; \
\n\t -r <filename> 文件名称,每行一条 URL,有特殊字符先做 URLencode,以 http/https 开头; \
\n\t -u <URL> 一条 URL,有特殊字符先做 URLencode,以 http/https 开头; \
\n\t -t <taskType> 任务类型 clear 刷新,push 预热; \
\n\t -n [int,[..100]] 可选项,每次操作文件数量,做多 100 条; \
\n\t -a [String,<domestic|overseas> 可选项,预热范围,不传是默认是全球;\
\n\t domestic 仅中国大陆; \
\n\t overseas 全球(不包含中国大陆); \
\n\t -o [String,<File|Directory>] 可选项,刷新的类型; \
\n\t File 文件刷新(默认值); \
\n\t Directory 目录刷新")
# TODO 入口
if __name__ == '__main__':
fun = Refresh()
fun.main(sys.argv[1:])
打包成EXE:
pyinstaller -F --hidden-import queue ./cdntool.py
工具支持参数:
D:\work\cdntool>cdntool.exe -h
script options explain:
-i <AccessKey> 访问阿里云凭证,访问控制台上可以获得;
-k <AccessKeySecret> 访问阿里云密钥,访问控制台上可以获得;
-r <filename> 文件名称,每行一条 URL,有特殊字符先做 URLencode,以 http/https 开头;
-u <URL> 一条 URL,有特殊字符先做 URLencode,以 http/https 开头;
-t <taskType> 任务类型 clear 刷新,push 预热;
-n [int,[..100]] 可选项,每次操作文件数量,做多 100 条;
-a [String,<domestic|overseas> 可选项,预热范围,不传是默认是全球;
domestic 仅中国大陆;
overseas 全球(不包含中国大陆);
-o [String,<File|Directory>] 可选项,刷新的类型;
File 文件刷新(默认值);
Directory 目录刷新
参考:
https://help.aliyun.com/document_detail/151829.html?spm=a2c4g.11186623.6.898.209e1612LlsyTt