def rk_create(self, im_string, timeout=60):
"""
im: 图片字节
im_type: 题目类型
"""
params = {
'typeid': 6113,
'timeout': timeout,
}
params.update(self.base_params)
files = {'image': ('a.jpg', im_string)}
while True:
try:
r = requests.post('http://api.ruokuai.com/create.json',
data=params,
files=files,
headers=self.headers)
#
Log.v("使用若快进行验证码识别")
data = r.json()
break
except requests.RequestException:
Log.w("提交若快打码请求出现问题， 正在重试中...")
continue
Log.d(data)
return data
def filter_by_config(self):
status = self.query()
if not status:
Log.e("未能获取乘客信息, 请重试")
PassengerData.raw_data = self.passengers
PassengerData.get_final_data()
data = PassengerData.find_people_by_names(Config.basic_config.ticket_people_list)
if len(data) == 0:
Log.e("乘客信息未在账号中找到, 请检查")
return False
if len(data) != len(Config.basic_config.ticket_people_list):
Log.w("乘客信息配置中包含错误, 已经滤除错误乘客")
self.config_passengers = data
return True
def send_requests(session, urlmapping_obj, params=None, data=None, **kwargs):
session.headers.update(urlmapping_obj.headers)
if urlmapping_obj.method.lower() == 'post':
session.headers.update(
{"Content-Type": r'application/x-www-form-urlencoded; charset=UTF-8'}
)
else:
session.headers.pop("Content-Type", None)
try:
Log.d("请求 url {url}".format(url=urlmapping_obj.url))
try:
response = session.request(method=urlmapping_obj.method,
url=urlmapping_obj.url,
params=params,
data=data,
timeout=10,
# allow_redirects=False,
**kwargs)
except requests.RequestException as e:
Log.w(e)
Log.w("请求{0}异常 ".format(urlmapping_obj.url))
raise ResponseError
if params:
Log.d("{url} Get 参数 {data}".format(url=urlmapping_obj.url,
data=params))
if data:
Log.d("{url} Post 参数 {data}".format(url=urlmapping_obj.url,
data=data))
Log.d("返回response url {url}".format(url=response.url))
if response.status_code == requests.codes.ok:
if 'xhtml+xml' in response.headers['Content-Type']:
data = response.text
root = ET.fromstring(data)
result = {v.tag: v.text for v in root.getchildren()}
return result
if 'json' in response.headers['Content-Type']:
result = response.json()
Log.d("{url} 返回值 {result}".format(url=urlmapping_obj.url,
result=result))
return result
# other type
return response.text
else:
Log.w(response.url)
Log.w(response.status_code)
Log.w("返回状态码有问题")
except Exception as e:
Log.e(e)
return None
def _get_submit_token(self):
html = send_requests(LOGIN_SESSION, self.URLS['getExtraInfo'])
Log.v("获取token中....")
result = re.findall(r"var globalRepeatSubmitToken = '(.*)'", html)
ticket_passenger_info = re.findall(r'var ticketInfoForPassengerForm=(.*);', html)
if result:
self.token = result[0]
if ticket_passenger_info:
try:
self.ticket_passenger_info = json.loads(ticket_passenger_info[0].replace("'", "\""))
except TypeError:
Log.w("获取submit info失败")
return False
if self.token and self.ticket_passenger_info:
Log.v("成功获取token与以及车次信息")
return True
else:
return False
def raw_data(self):
count = 10
while count:
try:
r = requests.get(STATION_URL)
if r.status_code == requests.codes.ok:
raw_data = city_re.findall(r.text)
if raw_data:
# remove empty line data.
data = map(lambda x: x.strip(), raw_data[0].split("@"))
Log.v("获取车站信息成功")
return list(filter(lambda x: bool(x), data))
else:
count -= 1
Log.w("无法读取车站信息，重试中")
continue
except requests.RequestException:
count -= 1
Log.w("获取车站信息失败，重试获取中")
continue
if count <= 0:
raise ResponseError
def get_captcha_image(session,
urlmapping_obj,
params=None,
data=None,
**kwargs):
"""
xml data example:
<HashMap>
<result_message>生成验证码成功</result_message>
<result_code>0</result_code>
<image>imagedata<image>
</HashMap>
format result data.
"""
session.headers.update(urlmapping_obj.headers)
try:
response = session.request(
method=urlmapping_obj.method,
url=urlmapping_obj.url,
params=params,
data=data,
timeout=10,
# allow_redirects=False,
**kwargs)
except requests.RequestException as e:
Log.w(e)
Log.w("请求{0}异常 ".format(urlmapping_obj.url))
raise ResponseError
if response.status_code == requests.codes.ok:
if 'xhtml+xml' in response.headers['Content-Type']:
data = response.text
root = ET.fromstring(data)
message = root.find('result_message').text
code = root.find('result_code').text
image = root.find('image').text
return {"result_message": message, "code": code, 'image': image}
elif 'json' in response.headers['Content-Type']:
return response.json()
else:
Log.w(response.url)
Log.w(response.status_code)
raise ResponseCodeError
else:
Log.w(response.url)
Log.w(response.status_code)
raise ResponseCodeError
def send_requests(session, urlmapping_obj, params=None, data=None, **kwargs):
session.headers.update(urlmapping_obj.headers)
if urlmapping_obj.method.lower() == 'post':
session.headers.update({
"Content-Type":
r'application/x-www-form-urlencoded; charset=UTF-8'
})
else:
session.headers.pop("Content-Type", None)
if urlmapping_obj.type.lower(
) == 'cdn' and CdnStorage.status and CdnStorage.result:
# use cdn to check ticket
cdn_ip = CdnStorage.choose_one().ip
urlmapping_obj.url = urlmapping_obj.url.replace(
parse_url(urlmapping_obj.url).host, cdn_ip)
Log.v("当前正在使用CDN IP 为{0}".format(cdn_ip))
try:
Log.d("请求 url {url}".format(url=urlmapping_obj.url))
try:
if urlmapping_obj.type.lower() == 'cdn':
# only query data disable warning
urllib3.disable_warnings(
urllib3.exceptions.InsecureRequestWarning)
response = session.request(
method=urlmapping_obj.method,
url=urlmapping_obj.url,
params=params,
data=data,
timeout=10,
# allow_redirects=False,
verify=False,
**kwargs)
else:
response = session.request(method=urlmapping_obj.method,
url=urlmapping_obj.url,
params=params,
data=data,
timeout=10,
**kwargs)
except requests.RequestException as e:
Log.w(e)
Log.w("请求{0}异常 ".format(urlmapping_obj.url))
raise ResponseError
if params:
Log.d("{url} Get 参数 {data}".format(url=urlmapping_obj.url,
data=params))
if data:
Log.d("{url} Post 参数 {data}".format(url=urlmapping_obj.url,
data=data))
Log.d("返回response url {url}".format(url=response.url))
if response.status_code == requests.codes.ok:
if 'xhtml+xml' in response.headers['Content-Type']:
data = response.text
root = ET.fromstring(data)
result = {v.tag: v.text for v in root.getchildren()}
return result
if 'json' in response.headers['Content-Type']:
result = response.json()
Log.d("{url} 返回值 {result}".format(url=urlmapping_obj.url,
result=result))
return result
# other type
return response.text
else:
Log.w(response.url)
Log.w(response.status_code)
Log.w("返回状态码有问题")
except Exception as e:
Log.e(e)
return None
