def resolve_vcode():
data = request.data
dct = json.loads(data)
if isinstance(dct, dict):
if dct['type'] == vcode.vcode_type:
vcode.resolve_vcode(dct['vcode'])
return get_success_response()
return get_error_response('post a wrong data')
def batch_wxid():
data = request.data
lst = json.loads(data)
if isinstance(lst, list):
sqlite_helper.batch_subscribe(lst)
return get_success_response().format()
else:
return get_error_response(data + 'is not a list').format()
def save_page():
subscribes = sqlite_helper.get_wxid_list()
print(subscribes)
try:
info_list = list()
for s in subscribes:
print("processing wxid=%s" % s['name'])
all_articles = sogou_api.get_articles_by_id(s['name'])
for a in all_articles:
info_list.append(a)
download_queue.resolve(info_list)
return get_success_response().format()
except Exception as e:
return get_error_response(e.message).format()
def stop():
download_queue.stop()
return get_success_response().format()
def start():
return get_success_response().format() if download_queue.start() else get_error_response('spider is running').format()
def remove_wxid(wxid):
wxid = wxid.strip()
sqlite_helper.unsubscribe(wxid)
return get_success_response().format()
def add_wxid(wxid):
try:
sqlite_helper.subscribe(wxid)
return get_success_response().format()
except Exception as e:
return get_error_response(e.message).format()