优化cdn判断

This commit is contained in:
Jing Ling
2020-07-26 14:40:10 +08:00
parent 07f436929e
commit fe06fd5c42
2 changed files with 9 additions and 10 deletions
+2 -2
View File
@@ -4,8 +4,6 @@
"x-via", "x-via",
"x-ser", "x-ser",
"x-cf1", "x-cf1",
"webcache",
"chinacache",
"cache", "cache",
"x-cache", "x-cache",
"x-cached", "x-cached",
@@ -14,6 +12,8 @@
"x-cache-status", "x-cache-status",
"x-cache-hits", "x-cache-hits",
"cc_cache", "cc_cache",
"webcache",
"chinacache",
"x-req-id", "x-req-id",
"x-requestid", "x-requestid",
"cf-request-id", "cf-request-id",
+7 -8
View File
@@ -23,12 +23,15 @@ def check_cname_count(cnames):
def check_cname_keyword(cname): def check_cname_keyword(cname):
for keyword in cdn_cname_keyword.keys(): names = cname.lower().split(',')
if keyword in cname: for name in names:
return True for keyword in cdn_cname_keyword.keys():
if keyword in name:
return True
def check_header_key(header): def check_header_key(header):
header = set(map(lambda x: x.lower(), header.keys()))
for key in cdn_header_key: for key in cdn_header_key:
if key in header: if key in header:
return True return True
@@ -50,13 +53,9 @@ def check_cdn(data):
for index, item in enumerate(data): for index, item in enumerate(data):
cname = item.get('cname') cname = item.get('cname')
if cname: if cname:
cnames = cname.split(',')
if check_cname_count(cnames):
data[index]['cdn'] = 1
continue
if check_cname_keyword(cname): if check_cname_keyword(cname):
data[index]['cdn'] = 1 data[index]['cdn'] = 1
continue break
header = item.get('header') header = item.get('header')
if header: if header:
header = json.loads(header) header = json.loads(header)