bls_plugin.py 38 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910
  1. import re
  2. import os
  3. import uuid
  4. import base64
  5. import time
  6. import json
  7. import random
  8. import string
  9. from datetime import datetime, timedelta
  10. from pathlib import Path
  11. from urllib.parse import urlparse, parse_qs, urlencode
  12. from typing import Dict, List, Optional, Any, Callable
  13. from curl_cffi import requests, const
  14. from bs4 import BeautifulSoup
  15. from cryptography.hazmat.primitives import serialization, hashes
  16. from cryptography.hazmat.primitives.asymmetric import padding
  17. from cryptography.hazmat.backends import default_backend
  18. # 框架依赖
  19. from vs_plg import IVSPlg
  20. from vs_types import VSPlgConfig, VSQueryResult, VSBookResult, AvailabilityStatus, NotFoundError, PermissionDeniedError, RateLimiteddError, SessionExpiredOrInvalidError, BizLogicError
  21. from toolkit.vs_cloud_api import VSCloudApi
  22. class BlsPlugin(IVSPlg):
  23. """
  24. BLS 签证预约插件 (精简版)
  25. """
  26. def __init__(self, group_id: str):
  27. self.group_id = group_id
  28. self.config: Optional[VSPlgConfig] = None
  29. self.free_config: Dict[str, Any] = {}
  30. self.logger = None
  31. self.session: Optional[requests.Session] = None
  32. # 运行时状态
  33. self.book_params: Dict = {}
  34. self.is_healthy = True
  35. # OCR 服务地址默认值
  36. self.local_service_url = ""
  37. def get_group_id(self) -> str:
  38. return self.group_id
  39. def set_log(self, logger: Callable[[str], None]) -> None:
  40. self.logger = logger
  41. def set_config(self, config: VSPlgConfig):
  42. self.config = config
  43. self.free_config = config.free_config or {}
  44. # 从配置中读取 OCR 服务地址,如果没有则使用默认
  45. if self.free_config.get("local_service_url"):
  46. self.local_service_url = self.free_config["local_service_url"]
  47. def health_check(self) -> bool:
  48. return self.is_healthy
  49. def create_session(self):
  50. self.session = requests.Session(
  51. proxy=self._get_proxy_url(),
  52. impersonate="chrome124",
  53. curl_options={
  54. const.CurlOpt.MAXAGE_CONN: 1800,
  55. const.CurlOpt.VERBOSE: self.config.debug
  56. }
  57. )
  58. domain = self.free_config.get("domain")
  59. if not domain:
  60. raise NotFoundError(message="Required field [domain] in free config")
  61. # 1.1 获取登录页 & 解析参数
  62. login_url = f"https://{domain}/Global/account/login"
  63. headers = {
  64. 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/131.0.0.0 Safari/537.36',
  65. 'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8'
  66. }
  67. resp = self._perform_request('GET', login_url, headers=headers)
  68. if self.config.debug:
  69. self._save_debug_html(resp.text, prefix="Bls_Login_Page")
  70. soup = BeautifulSoup(resp.text, 'html.parser')
  71. form_data = self._extract_hidden_fields(soup)
  72. real_user = None
  73. real_pass = None
  74. # 解析动态 ID (UserId1, Password1 等)
  75. for inp in soup.find_all('input'):
  76. name = inp.get('name', '')
  77. if inp.has_attr('required'):
  78. if 'UserId' in name:
  79. real_user = name
  80. elif 'Password' in name:
  81. real_pass = name
  82. # 解析 data 参数 (用于验证码)
  83. data_val = self._extract_js_var(resp.text, "iframeOpenUrl", r"data=([^']+)")
  84. # 1.2 处理验证码
  85. captcha_token = self._solve_bls_captcha(data_val)
  86. # 1.3 提交登录
  87. submit_url = f"https://{domain}/Global/account/loginsubmit"
  88. payload = form_data
  89. payload["X-Requested-With"] = "XMLHttpRequest"
  90. payload["CaptchaData"] = captcha_token
  91. # 填入账号密码
  92. payload[real_user] = self.config.account.username
  93. payload[real_pass] = self.config.account.password
  94. login_resp = self._perform_request('POST', submit_url, data=payload, headers=headers)
  95. if login_resp.json()['success']:
  96. return
  97. raise BizLogicError(message='Login failed')
  98. # =========================================================================
  99. # 2. 查询流程 (Query)
  100. # =========================================================================
  101. def query(self) -> VSQueryResult:
  102. res = VSQueryResult()
  103. domain = self.free_config.get("domain")
  104. # 2.1 签证类型验证
  105. url_vtv = f"https://{domain}/Global/bls/visatypeverification"
  106. headers = {
  107. 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/131.0.0.0 Safari/537.36',
  108. 'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8'
  109. }
  110. resp = self._perform_request('GET', url_vtv, headers=headers)
  111. if self.config.debug:
  112. self._save_debug_html(resp.text, prefix="Bls_Visatypeverification_Page")
  113. self._check_resp_is_session_expired_or_invalid('APPLICATION PROCESS', resp)
  114. form_vtv = self._extract_hidden_fields(BeautifulSoup(resp.text, 'html.parser'))
  115. captcha_token = self._solve_bls_captcha()
  116. form_vtv['CaptchaData'] = captcha_token
  117. form_vtv["X-Requested-With"] = "XMLHttpRequest"
  118. vtv_resp = self._perform_request('POST', f"https://{domain}/Global/bls/VisaTypeVerification", data=form_vtv, headers=headers)
  119. if not vtv_resp.json()['success']:
  120. raise BizLogicError(message='Submit VisaTypeVerification Failed')
  121. # 2.2 签证类型选择
  122. return_url = vtv_resp.json()['returnUrl'] # 包含 data=xxx
  123. data_val = re.search(r"data=([^&]+)", return_url).group(1)
  124. url_vt = f"https://{domain}/Global/bls/visatype?data={data_val}"
  125. vt_resp = self._perform_request('GET', url_vt, headers=headers)
  126. if self.config.debug:
  127. self._save_debug_html(resp.text, prefix="Bls_Visatype_Page")
  128. self._check_resp_is_session_expired_or_invalid('APPLICATION PROCESS', resp)
  129. # 这里需要极其复杂的 JS 变量提取 (JS Arrays -> Match Name -> Get ID)
  130. vt_payload = self._construct_visatype_payload(vt_resp.text, BeautifulSoup(vt_resp.text, 'html.parser'))
  131. res.city = self.free_config.get('city', '')
  132. res.country = self.free_config.get('country', '')
  133. res.visa_type = self.free_config.get('visa_type', '')
  134. res.routing_key = self.free_config.get('routing_key', '')
  135. vt_res = self._perform_request('POST', f"https://{domain}/Global/bls/VisaType", data=vt_payload, headers=headers)
  136. if not vt_res.json()['success']:
  137. if not vt_res.json()['available']:
  138. res.success = False
  139. res.availability_status = AvailabilityStatus.NoneAvailable
  140. return res
  141. # 2.3 获取预约参数
  142. final_url = vt_res.json()['returnUrl']
  143. q_params = parse_qs(urlparse(final_url).query)
  144. self.book_params = {k: v[0] for k, v in q_params.items()}
  145. # 2.4 查询日历
  146. url_ma = f"https://{domain}/Global/blsAppointment/ManageAppointment?{urlencode(self.book_params)}"
  147. resp_ma = self._perform_request('GET', url_ma, headers=headers)
  148. if self.config.debug:
  149. self._save_debug_html(resp.text, prefix="Bls_ManageAppointment_Page")
  150. self._check_resp_is_session_expired_or_invalid('APPLICATION PROCESS', resp)
  151. avail_str = self._extract_js_var(resp_ma.text, "var availDates", r"var availDates =(.*?);")
  152. if avail_str:
  153. avail_json = json.loads(avail_str)
  154. # 提取日期
  155. dates = [x['DateText'] for x in avail_json['ad'] if x['SingleSlotAvailable']]
  156. if dates:
  157. res.success = True
  158. res.availability_status = AvailabilityStatus.Available
  159. res.earliest_date = dates[0]
  160. for d in dates:
  161. da = VSQueryResult.DateAvailability()
  162. da.date = d
  163. da.times = []
  164. time_slot = VSQueryResult.DateAvailability.TimeSlot(time="00:00", label="Available")
  165. da.times.append(time_slot)
  166. res.availability.append(da)
  167. else:
  168. res.success = False
  169. res.availability_status = AvailabilityStatus.NoneAvailable
  170. return res
  171. raise BizLogicError(message='Query page not found required field [var availDates]')
  172. def book(self, slot_info: VSQueryResult, user_inputs: Dict) -> VSBookResult:
  173. res = VSBookResult()
  174. domain = self.free_config.get("domain")
  175. headers = {
  176. 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/131.0.0.0 Safari/537.36',
  177. 'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8'
  178. }
  179. # 3.1 获取 Manage Page (为了 Token 和 JS 变量)
  180. url_ma = f"https://{domain}/Global/blsAppointment/ManageAppointment?{urlencode(self.book_params)}"
  181. resp_ma = self._perform_request('GET', url_ma, headers=headers)
  182. ma_soup = BeautifulSoup(resp_ma.text, 'html.parser')
  183. ma_form = self._extract_hidden_fields(ma_soup)
  184. req_token = ma_form.get('__RequestVerificationToken')
  185. # 3.2 上传照片
  186. if 'passport_image_url' not in user_inputs:
  187. raise NotFoundError()
  188. photo_bytes = requests.get(user_inputs['passport_image_url']).content
  189. boundary = "----WebKitFormBoundary" + "".join(random.choices(string.ascii_letters + string.digits, k=16))
  190. upload_headers = {
  191. "content-type": f"multipart/form-data; boundary={boundary}",
  192. "requestverificationtoken": req_token,
  193. "x-requested-with": "XMLHttpRequest",
  194. }
  195. body = (f"--{boundary}\r\nContent-Disposition: form-data; name=\"file\"; filename=\"photo.jpg\"\r\n"
  196. f"Content-Type: image/jpeg\r\n\r\n").encode("utf-8") + photo_bytes + f"\r\n--{boundary}--\r\n".encode("utf-8")
  197. up_resp = self.session.post(f"https://{domain}/Global/query/UploadProfileImage", headers=upload_headers, data=body)
  198. if up_resp.status_code !=200:
  199. raise BizLogicError(message='Upload Passport Image failed')
  200. ma_form['ApplicantPhotoId'] = up_resp.json()['fileId']
  201. # 3.3 邮箱 OTP 流程
  202. data_val = self._extract_js_var(resp_ma.text, "win.iframeOpenUrl", r"data=([^&]+)")
  203. # 发送 OTP
  204. headers["X-Requested-With"] = "XMLHttpRequest"
  205. self._perform_request('GET', f"https://{domain}/Global/blsappointment/SendAppointmentVerificationCode?code={data_val}", headers=headers)
  206. # 读取 OTP (Wait 30s max)
  207. otp_code = self._read_otp_email(wait_sec=30)
  208. # 验证 OTP
  209. verify_payload = {
  210. "Code": otp_code,
  211. "Value": ma_form.get('EmailCode'),
  212. "Id": ma_form.get('Id')
  213. }
  214. headers['requestverificationtoken'] = req_token
  215. v_resp = self._perform_request('POST', f"https://{domain}/Global/blsappointment/VerifyEmail", data=verify_payload, headers=headers)
  216. headers.pop('requestverificationtoken')
  217. if not v_resp.json().get('success'):
  218. raise BizLogicError(message='Email verification failed')
  219. ma_form['EmailVerified'] = 'True'
  220. ma_form['EmailVerificationCode'] = otp_code
  221. # 3.4 锁定时间 (简单随机)
  222. target_date = slot_info.earliest_date
  223. # Query Slots in Day
  224. slot_url = f"https://{domain}/Global/blsappointment/GetAvailableSlotsByDate"
  225. # 构造复杂的 query params... 省略部分非关键参数
  226. slot_params = {
  227. "appointmentDate": target_date,
  228. "locationId": ma_form.get("LocationId"),
  229. "categoryId": ma_form.get("AppointmentCategoryId"),
  230. "visaType": ma_form.get("VisaType"),
  231. "visaSubType": ma_form.get("VisaSubTypeId"),
  232. "applicantCount": 1,
  233. "dataSource": ma_form.get("DataSource"),
  234. "missionId": ma_form.get("MissionId")
  235. }
  236. headers['requestverificationtoken'] = req_token
  237. slots_resp = self._perform_request('POST', slot_url, params=slot_params, headers=headers)
  238. headers.pop('requestverificationtoken')
  239. slots_data = sorted(slots_resp.json(), key=lambda x: -x["Count"]) # 选剩余最多的
  240. if not slots_data or slots_data[0]['Count'] <= 0:
  241. self._log('Available slot times not found')
  242. res.success = False
  243. return res
  244. target_time = slots_data[0]['Name']
  245. ma_form['ServerAppointmentDate'] = target_date
  246. ma_form['AppointmentDetailsList'] = '[]'
  247. # 这里的 key 是动态的 ID,需重新解析 ID
  248. date_id = re.search(r'AppointmentDate(\d+)', str(ma_soup)).group(1)
  249. slot_id = re.search(r'AppointmentSlot(\d+)', str(ma_soup)).group(1)
  250. ma_form[f'AppointmentDate{date_id}'] = target_date
  251. ma_form[f'AppointmentSlot{slot_id}'] = target_time
  252. # 3.5 再次验证码 & 提交 ManageAppointment
  253. captcha_token = self._solve_bls_captcha(data_val)
  254. ma_form['CaptchaData'] = captcha_token
  255. final_ma_resp = self._perform_request('POST', f"https://{domain}/Global/BLSAppointment/ManageAppointment", data=ma_form, headers=headers)
  256. appt_model_id = final_ma_resp.json().get('model', {}).get('Id')
  257. if not appt_model_id:
  258. raise NotFoundError(message='Appointment model id not found')
  259. # 3.6 填写申请表 (VisaAppointmentForm)
  260. # 获取页面 -> 解析 JS 变量 -> 映射 UserInfo -> 提交
  261. # 这里逻辑较深,核心是映射。简化为提交一个空的 applicants JSON,实际需完整映射。
  262. # 假设 _fill_applicant_form 做了这些工作
  263. self._submit_final_form(appt_model_id, user_inputs, self.book_params, req_token)
  264. # 成功,返回 Liveness 链接
  265. Liveness_page = f"https://{domain}/Global/BlsAppointment/livenessView?id={appt_model_id}"
  266. session_data = self._save_http_session(Liveness_page)
  267. res.success = True
  268. res.account = self.config.account.username
  269. res.session_id = session_data['session_id']
  270. res.book_date = target_date
  271. res.book_time = target_time
  272. self._log(f"Book Success. Liveness URL: {res.payment_link}")
  273. return res
  274. def _log(self, message):
  275. if self.logger:
  276. self.logger(f'[BlsPlugin] [{self.group_id}] {message}')
  277. def _get_proxy_url(self):
  278. # 构造代理
  279. proxy_url = ""
  280. if self.config.proxy.ip:
  281. s = self.config.proxy
  282. if s.username:
  283. proxy_url = f"{s.scheme}://{s.username}:{s.password}@{s.ip}:{s.port}"
  284. else:
  285. proxy_url = f"{s.scheme}://{s.ip}:{s.port}"
  286. return proxy_url
  287. def _save_debug_html(self, content: str, prefix: str = "debug"):
  288. save_dir = "debug_pages"
  289. if not os.path.exists(save_dir):
  290. os.makedirs(save_dir)
  291. timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
  292. filename = f"{save_dir}/{prefix}_{timestamp}.html"
  293. with open(filename, "w", encoding="utf-8") as f:
  294. f.write(content)
  295. self._log(f"HTML saved to: {filename}")
  296. def _perform_request(self, method, url, headers=None, data=None, json_data=None, params=None):
  297. """
  298. 统一 HTTP 请求封装,严格复刻 C++ 逻辑:
  299. 1. 发送 OPTIONS 请求
  300. 2. 发送实际请求
  301. """
  302. resp = self.session.request(method, url, headers=headers, data=data, json=json_data, params=params, timeout=30)
  303. if self.config.debug:
  304. self._log(f'[perform request] Response={resp.text}\nMethod={method}, Url={url}, Data={data}, JsonData={json_data}, Params={params}')
  305. if resp.status_code == 200:
  306. return resp
  307. elif resp.status_code == 401:
  308. self.is_healthy = False
  309. raise SessionExpiredOrInvalidError()
  310. elif resp.status_code == 403:
  311. raise PermissionDeniedError()
  312. elif resp.status_code == 429:
  313. self.is_healthy = False
  314. raise RateLimiteddError()
  315. else:
  316. raise BizLogicError(message=f"HTTP Error {resp.status_code}: {resp.text[:100]}")
  317. def _solve_bls_captcha(self, data='') -> Optional[str]:
  318. """
  319. 验证码处理:获取图片 -> 调用远程 OCR 服务 -> 提交验证
  320. """
  321. domain = self.free_config.get("domain")
  322. url = f"https://{domain}/Global/NewCaptcha/GenerateCaptcha"
  323. if data: url = f"https://{domain}/Global/CaptchaPublic/GenerateCaptcha?data={data}"
  324. headers = {
  325. 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/131.0.0.0 Safari/537.36',
  326. 'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8'
  327. }
  328. resp = self._perform_request("GET", url, headers=headers)
  329. if self.config.debug:
  330. self._save_debug_html(resp.text, prefix="Bls_Captcha_Page")
  331. self._check_resp_is_session_expired_or_invalid('Please select all boxes with number', resp)
  332. soup = BeautifulSoup(resp.text, 'html.parser')
  333. resp = requests.post(
  334. f'{self.local_service_url}/browser/visable_captchas',
  335. data=resp.text,
  336. headers={"Content-Type": "text/plain"},
  337. timeout=10
  338. )
  339. result = resp.json()
  340. if result.get('status') != 'success':
  341. raise BizLogicError(message='Broswer task failed')
  342. numbers = result['data']['number']
  343. image_ids = result['data']['image_ids']
  344. selected_ids = []
  345. for sid in image_ids:
  346. div = soup.find("div", id=sid)
  347. img = div.find("img")
  348. src = img.get("src")
  349. base64_data = src.split("base64,", 1)[1]
  350. img_bytes = base64.b64decode(base64_data)
  351. ocr_resp = requests.post(
  352. f'{self.local_service_url}/predict/bls?model=pytorch',
  353. data=img_bytes,
  354. headers={"Content-Type": "application/octet-stream"},
  355. timeout=5
  356. )
  357. if ocr_resp.status_code == 200:
  358. res_json = ocr_resp.json()
  359. ocr_res = res_json.get('data', '').replace('$', '')[:3]
  360. self._log(f'ocr captcha id={sid} result={ocr_res}, target={numbers}')
  361. if ocr_res == numbers:
  362. selected_ids.append(sid)
  363. else:
  364. raise BizLogicError(message='Captcha server response error')
  365. if not selected_ids:
  366. raise BizLogicError(message='Captcha selected ids is empty')
  367. # 3. 提交选中结果
  368. self._log(f'select_ids={selected_ids}')
  369. form = self._extract_hidden_fields(soup)
  370. form['SelectedImages'] = ",".join(selected_ids)
  371. submit_url = f"https://{domain}/Global/{'CaptchaPublic' if data else 'NewCaptcha'}/SubmitCaptcha"
  372. headers["X-Requested-With"] = "XMLHttpRequest"
  373. resp = self._perform_request('POST', submit_url, headers=headers, data=form)
  374. j = resp.json()
  375. if j.get('success'):
  376. if data:
  377. return resp.json()['captcha']
  378. else:
  379. return resp.json()['cd']
  380. else:
  381. # 存盘所有错误验证码后续进行数据分析
  382. self._log('Captcha Selection Invalid, Saving important data to data/bls_captcha')
  383. for img in soup.select("img.captcha-img"):
  384. src = img.get("src", "")
  385. if not src.startswith("data:image"):
  386. continue
  387. b64 = src.split("base64,", 1)[1]
  388. with open(f'data/bls_captcha/{uuid.uuid4().hex}.jpg', "wb") as fp:
  389. fp.write(base64.b64decode(b64))
  390. raise BizLogicError(message="Sovle captcha failed")
  391. def _extract_hidden_fields(self, soup) -> Dict:
  392. params = {}
  393. form = soup.find("form")
  394. if form:
  395. for inp in form.find_all("input"):
  396. name = inp.get("name")
  397. if name: params[name] = inp.get("value", "")
  398. else:
  399. self._log('Form element not found')
  400. return params
  401. def _extract_js_var(self, html, context, pattern):
  402. # 简单正则提取
  403. if context in html:
  404. match = re.search(pattern, html)
  405. if match: return match.group(1)
  406. return ""
  407. def _construct_visatype_payload(self, html: str, soup: BeautifulSoup) -> Optional[Dict]:
  408. """
  409. 构造 VisaType 提交参数 (对应原代码 parse_visatype_form)
  410. """
  411. # 基础表单参数 (__RequestVerificationToken 等)
  412. params = self._extract_hidden_fields(soup)
  413. # 提取页面中的 JS 数据变量
  414. def get_js_data(var_name):
  415. try:
  416. # 匹配 var name = [...]; 结构
  417. pattern = f"var {var_name}\\s*=\\s*(.*?);"
  418. match = re.search(pattern, html, re.DOTALL)
  419. if match:
  420. return json.loads(match.group(1))
  421. except Exception as e:
  422. self._log(f"Failed to parse JS var {var_name}: {e}")
  423. return []
  424. # 读取配置
  425. query_selector = self.free_config.get("query_selector", {})
  426. cfg_jur = query_selector.get("jurisdiction")
  427. cfg_loc = query_selector.get("location")
  428. cfg_type = query_selector.get("visa_type")
  429. cfg_subtype = query_selector.get("visa_subtype")
  430. cfg_cat = query_selector.get("appointment_category")
  431. jur_value = None
  432. loc_value = None
  433. type_value = None
  434. subtype_value = None
  435. cat_value = None
  436. resp = requests.post(
  437. f'{self.local_service_url}/browser/visatype_visable',
  438. data=html,
  439. headers={"Content-Type": "text/plain"},
  440. timeout=10
  441. )
  442. result = resp.json()
  443. if result.get('status') != 'success':
  444. raise BizLogicError(message='Broswer task failed')
  445. jur_id = result['data']['jur_id']
  446. loc_id = result['data']['loc_id']
  447. type_id = result['data']['type_id']
  448. subtype_id = result['data']['subtype_id']
  449. cat_id = result['data']['cat_id']
  450. jurisdiction_list = get_js_data("jurisdictionData")
  451. location_list = get_js_data("locationData")
  452. visa_type_list = get_js_data("visaIdData")
  453. visa_subtype_list = get_js_data("visasubIdData")
  454. app_category_list = get_js_data("AppointmentCategoryIdData")
  455. # 4. 匹配 Value
  456. # (A) Appointment Category
  457. for item in app_category_list:
  458. if item.get("Name") == cfg_cat:
  459. cat_value = item.get("Id")
  460. break
  461. # (B) Jurisdiction (如果配置了)
  462. if cfg_jur and jurisdiction_list:
  463. for item in jurisdiction_list:
  464. if item.get("Name") == cfg_jur:
  465. jur_value = item.get("Id")
  466. break
  467. # (C) Location
  468. for item in location_list:
  469. if item.get("Name") == cfg_loc:
  470. loc_value = item.get("Id")
  471. break
  472. # (D) Visa Type (需匹配 LocationId)
  473. if loc_value:
  474. for item in visa_type_list:
  475. # 比较 Name 和 LocationId
  476. if item.get("Name") == cfg_type and str(item.get("LocationId")) == str(loc_value):
  477. type_value = item.get("Id")
  478. break
  479. # (E) Visa SubType (需匹配 VisaType Value)
  480. if type_value:
  481. for item in visa_subtype_list:
  482. # BLS 逻辑: visasubIdData 中的 Value 字段对应 VisaTypeId
  483. if item.get("Name") == cfg_subtype and str(item.get("Value")) == str(type_value):
  484. subtype_value = item.get("Id")
  485. break
  486. # 5. 构造动态参数 & 校验
  487. if not cat_value:
  488. raise NotFoundError(message=f"Config: AppCategory '{cfg_cat}' not found")
  489. params[f"AppointmentCategoryId{cat_id}"] = cat_value
  490. if cfg_jur:
  491. if not jur_value:
  492. raise NotFoundError(message=f"Config: Jurisdiction '{cfg_jur}' not found")
  493. params[f"JurisdictionId{jur_id}"] = jur_value
  494. if not loc_value:
  495. raise NotFoundError(message=f"Config: Location '{cfg_loc}' not found")
  496. params[f"Location{loc_id}"] = loc_value
  497. if not type_value:
  498. raise NotFoundError(message=f"Config: VisaType '{cfg_type}' not found for Loc '{cfg_loc}'")
  499. params[f"VisaType{type_id}"] = type_value
  500. if not subtype_value:
  501. raise NotFoundError(message=f"Config: VisaSubType '{cfg_subtype}' not found")
  502. params[f"VisaSubType{subtype_id}"] = subtype_value
  503. # 固定参数
  504. for k in list(params.keys()):
  505. if k.startswith("AppointmentFor"):
  506. params[k] = "Individual"
  507. # 6. 构造 ResponseData (行为轨迹模拟)
  508. # BLS 后端会校验这个字段,模拟用户选择下拉框的时间间隔
  509. response_data = []
  510. current_time = datetime.utcnow()
  511. def add_trace(prefix, val_id):
  512. nonlocal current_time
  513. # 模拟 1-3 秒的操作间隔
  514. duration = random.randint(1000, 3000)
  515. gap = random.randint(500, 1500)
  516. start = current_time
  517. end = start + timedelta(milliseconds=duration)
  518. # BLS 时间格式: 2023-10-27T10:00:00.123Z
  519. fmt = "%Y-%m-%dT%H:%M:%S.%f"
  520. response_data.append({
  521. "Id": f"{prefix}{val_id}",
  522. "Start": start.strftime(fmt)[:-3] + "Z",
  523. "End": end.strftime(fmt)[:-3] + "Z",
  524. "Total": duration,
  525. "Selected": True
  526. })
  527. current_time = end + timedelta(milliseconds=gap)
  528. # 按顺序添加轨迹
  529. add_trace("AppointmentCategoryId", cat_id)
  530. if jur_id: add_trace("JurisdictionId", jur_id)
  531. add_trace("Location", loc_id)
  532. add_trace("VisaType", type_id)
  533. add_trace("VisaSubType", subtype_id)
  534. params["ResponseData"] = json.dumps(response_data)
  535. params["X-Requested-With"] = "XMLHttpRequest"
  536. return params
  537. def _submit_final_form(self, model_id: str, user_inputs: Dict, book_params: Dict, token: str):
  538. """
  539. 提交最终签证申请表 (VisaAppointmentForm)
  540. 对应原代码的: get_visa_appointment_form_html -> parse -> fix_data -> submit
  541. """
  542. domain = self.free_config.get("domain")
  543. headers = {
  544. 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/131.0.0.0 Safari/537.36',
  545. 'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8'
  546. }
  547. # 1. 获取表单页面 (为了提取 JS 变量映射表)
  548. url_get = f"https://{domain}/Global/BlsAppointment/VisaAppointmentForm?appointmentId={model_id}"
  549. # 构造 Referer
  550. ref_query = urlencode(book_params)
  551. referer = f"Global/blsAppointment/ManageAppointment?{ref_query}"
  552. headers['X-Requested-With'] = "XMLHttpRequest"
  553. headers['Referer'] = f"https://{domain}/{referer}"
  554. resp = self._perform_request('GET', url_get, headers=headers)
  555. headers.pop['X-Requested-With']
  556. headers.pop['Referer']
  557. html = resp.text
  558. soup = BeautifulSoup(resp.text, 'html.parser')
  559. # 2. 提取基础隐藏域 (包含 __RequestVerificationToken 等)
  560. form_data = self._extract_hidden_fields(soup)
  561. # 3. 提取下拉菜单数据源 (JS Variables)
  562. # BLS 的页面里有很多 var countryData = [...]; 这种数据
  563. def get_list(name):
  564. val = self._extract_js_var(html, f"var {name}", rf"var {name}\s*=\s*(.*?);")
  565. return json.loads(val) if val else []
  566. # 提取关键数据源
  567. country_data = get_list("countryData")
  568. gender_data = get_list("genderData")
  569. marital_data = get_list("maritalStatusData")
  570. occupation_data = get_list("occupationData")
  571. # passport_type_data = get_list("passportTypeData") # 通常默认 Ordinary
  572. # 4. 辅助函数:根据文本找 ID
  573. def find_id(data_list, text_val, default=None):
  574. if not text_val: return default
  575. text_val = str(text_val).lower().strip()
  576. for item in data_list:
  577. if str(item.get("Name")).lower() == text_val:
  578. return item.get("Id")
  579. return default
  580. # 5. 准备日期 (YYYY-MM-DD)
  581. # uinfo 中的日期可能是不同格式,需统一
  582. def fmt_date(d_str):
  583. try:
  584. # 尝试解析常见格式
  585. for fmt in ["%Y-%m-%d", "%d/%m/%Y", "%d-%m-%Y"]:
  586. try:
  587. return datetime.strptime(d_str, fmt).strftime("%Y-%m-%d")
  588. except: pass
  589. except: pass
  590. return d_str # 原样返回 fallback
  591. dob = fmt_date(user_inputs.get("birthday", ""))
  592. ppt_issue = fmt_date(user_inputs.get("passport_issue_date", ""))
  593. ppt_expiry = fmt_date(user_inputs.get("passport_expiry_date", ""))
  594. # 自动计算行程日期 (如果未提供,默认一个月后)
  595. try:
  596. travel_date = (datetime.now() + timedelta(days=30)).strftime("%Y-%m-%d")
  597. except: travel_date = ""
  598. # 6. 构造申请人详细数据对象 (JSON)
  599. # 注意:这里的字段名必须严格匹配 BLS 后端实体定义
  600. applicant_detail = {
  601. "ApplicantSerialNo": "1",
  602. "ApplicantId": form_data.get("applicantId", "0"), # 从页面隐藏域提取
  603. "Id": form_data.get("applicantId", "0"),
  604. "ParentId": form_data.get("Id", model_id), # 关联的 Appointment ID
  605. # 基本信息
  606. "FirstName": user_inputs.get("first_name", ""),
  607. "SurName": user_inputs.get("last_name", ""),
  608. "LastName": user_inputs.get("last_name", ""),
  609. "SurnameAtBirth": user_inputs.get("last_name", ""), # 默认同名
  610. "GenderId": find_id(gender_data, user_inputs.get("gender"), "1"), # 默认 Male
  611. "MaritalStatusId": find_id(marital_data, user_inputs.get("marital_status", "Single"), "1"),
  612. "ServerDateOfBirth": dob,
  613. # 国籍/出生地
  614. "PlaceOfBirth": user_inputs.get("place_of_birth", "-"),
  615. "CountryOfBirthId": find_id(country_data, user_inputs.get("nationality"), "0"),
  616. "NationalityAtBirthId": find_id(country_data, user_inputs.get("nationality"), "0"),
  617. "NationalityId": find_id(country_data, user_inputs.get("nationality"), "0"),
  618. # 护照信息
  619. "PassportType": "Ordinary Passport", # 默认
  620. "PassportNo": user_inputs.get("passport_no", ""),
  621. "ServerPassportIssueDate": ppt_issue,
  622. "ServerPassportExpiryDate": ppt_expiry,
  623. "IssuePlace": user_inputs.get("place_of_issue", "-"),
  624. "IssueCountryId": find_id(country_data, user_inputs.get("nationality"), "0"),
  625. # 联系方式 (必填占位符)
  626. "HomeAddressLine1": "-",
  627. "HomeAddressCity": "-",
  628. "HomeAddressPostalCode": "-",
  629. "HomeAddressContactNumber": user_inputs.get("phone", "-"),
  630. "HomeAddressCountryId": find_id(country_data, user_inputs.get("nationality"), "0"),
  631. "EmployerName": "-",
  632. "EmployerAddress": "-",
  633. # 职业
  634. "CurrentOccupationId": find_id(occupation_data, user_inputs.get("occupation", "Others"), "20"),
  635. # 行程信息 (部分写死为常规旅游)
  636. "PurposeOfJourneyId": "Tourism",
  637. "MemberStateDestinationId": "Spain",
  638. "MemberStateFirstEntryId": "Spain",
  639. "NumberOfEntriesRequested": "Multiple Entries",
  640. "IntendedStayDuration": "5",
  641. "ServerTravelDate": travel_date,
  642. "ServerIntendedDateOfArrival": travel_date,
  643. "ServerIntendedDateOfDeparture": travel_date, # 简化
  644. # 费用承担
  645. "CostCoveredById": "By the Applicant himself / herself",
  646. "MeansOfSupportId": "Cash",
  647. # 杂项
  648. "IsMinor": False,
  649. "IsVisaIssuedBefore": False,
  650. "BlsInvitingAuthority": "1", # 这里的 1 通常代表 "No" 或者特定枚举
  651. "PreviousFingerPrintStatus": "2", # 2 通常代表 No
  652. # 邀请人信息 (旅游通常填酒店或空)
  653. "InvitingAuthorityName": "-",
  654. "InvitingAddress": "-",
  655. "InvitingCity": "-",
  656. "InvitingEmail": "no-reply@example.com"
  657. }
  658. # 7. 更新表单数据
  659. # ApplicantsDetailsList 需要是一个 JSON 字符串
  660. form_data['ApplicantsDetailsList'] = json.dumps([applicant_detail])
  661. # 补全其他可能需要的字段
  662. form_data['PreviousFingerPrintStatus_0'] = "2"
  663. form_data['BlsInvitingAuthority_0'] = "1"
  664. form_data["X-Requested-With"] = "XMLHttpRequest"
  665. # 8. 提交
  666. # 注意:提交地址通常和 manage appointment 相同,或者是特定的 Save 接口
  667. # 根据你的原代码,是 Global/BLSAppointment/ManageAppointment
  668. url_post = f"https://{domain}/Global/BLSAppointment/ManageAppointment"
  669. # Headers 需要 Token
  670. headers = {
  671. 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/131.0.0.0 Safari/537.36',
  672. 'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
  673. "Referer": f"https://{domain}/{referer}",
  674. "X-Requested-With": "XMLHttpRequest",
  675. "requestverificationtoken": token
  676. }
  677. # 这里的 form_data['params'] 逻辑在 _extract_hidden_fields 可能会有差异
  678. # 确保 form_data 是扁平的字典
  679. submit_resp = self._perform_request('POST', url_post, data=form_data, headers=headers)
  680. if submit_resp.json().get('success'):
  681. self._log("Final Form Submitted Successfully.")
  682. return True
  683. raise BizLogicError(message='Submit application form failed')
  684. def _read_otp_email(self, wait_sec: int = 60) -> str:
  685. """
  686. 读取 BLS 的 OTP 邮件
  687. """
  688. master_email = "visafly666@gmail.com"
  689. recipient = self.config.account.username
  690. sender = "Info@blsinternational.com"
  691. subject_keywords = "BLS"
  692. body_keywords = "verification code"
  693. # 设置时间起点 (UTC)
  694. now_utc = datetime.utcnow()
  695. formatted_utc_time = now_utc.strftime("%Y-%m-%d %H:%M:%S")
  696. self._log(f"Waiting for OTP from {sender}...")
  697. # 轮询查收, 每 5 秒查一次
  698. attempts = wait_sec // 5
  699. for i in range(attempts):
  700. # 调用云端接口获取邮件内容
  701. # expiry=300 表示邮件有效搜索窗口为 5 分钟
  702. content_out = VSCloudApi.Instance().fetch_mail_content(
  703. master_email,
  704. sender,
  705. recipient,
  706. subject_keywords,
  707. body_keywords,
  708. formatted_utc_time,
  709. 300
  710. )
  711. # 正则匹配 6 位数字验证码
  712. match = re.search(r'\b\d{6}\b', content_out)
  713. if match:
  714. otp = match.group(0)
  715. self._log("OTP code found: {otp}")
  716. return otp
  717. # 等待下一次轮询
  718. time.sleep(5)
  719. if i % 2 == 0:
  720. self._log("OTP not received yet, retrying...")
  721. # 超时处理
  722. raise NotFoundError(f"OTP email not found within {wait_sec}s")
  723. def _check_resp_is_session_expired_or_invalid(self, keyword, resp) -> bool:
  724. """
  725. 检测是否发生了 Session 过期
  726. """
  727. # 1. 检查最终 URL 是否包含登录页特征
  728. # 这里的判断依据是你提供的日志:Redirect to /Global/Account/LogIn
  729. if "/Account/LogIn" in resp.url or "/Account/Login" in resp.url:
  730. self.is_healthy = False
  731. raise SessionExpiredOrInvalidError()
  732. # 2. (备用) 如果 _perform_request 禁止了重定向,检查 302 Location
  733. if resp.status_code == 302:
  734. location = resp.headers.get("Location", "")
  735. if "/Account/LogIn" in location or "/Account/Login" in location:
  736. self.is_healthy = False
  737. raise SessionExpiredOrInvalidError()
  738. resp_text = resp.text
  739. if not resp_text:
  740. self.is_healthy = False
  741. raise SessionExpiredOrInvalidError()
  742. if keyword not in resp_text:
  743. if 'your session has expired, please login again.' in resp_text.lower():
  744. self.is_healthy = False
  745. raise SessionExpiredOrInvalidError()
  746. def _save_http_session(self, page_url):
  747. """
  748. 提取 cookies, local_storage, 存入 VSCloudApi
  749. """
  750. cookies_dict = {}
  751. # 方式 1: curl_cffi 的 cookies 对象通常支持 get_dict()
  752. if hasattr(self.session.cookies, "get_dict"):
  753. cookies_dict = self.session.cookies.get_dict()
  754. else:
  755. # 方式 2: 迭代 (兼容标准 CookieJar)
  756. for c in self.session.cookies:
  757. cookies_dict[c.name] = c.value
  758. cookies_str = json.dumps(cookies_dict)
  759. # 简单生成 SessionID hash
  760. ua_str = self.user_agent or "unknown_ua"
  761. raw = cookies_str + ua_str + page_url
  762. session_id = hashes.Hash(hashes.SHA256(), backend=default_backend())
  763. session_id.update(raw.encode())
  764. sid = session_id.finalize().hex()
  765. proxy_str = ""
  766. if self.config.proxy.ip:
  767. proxy_str = f"{self.config.proxy.scheme}://"
  768. if self.config.proxy.username:
  769. proxy_str += f"{self.config.proxy.username}:{self.config.proxy.password}@"
  770. proxy_str += f"{self.config.proxy.ip}:{self.config.proxy.port}"
  771. return VSCloudApi.Instance().create_http_session(
  772. sid, cookies_str, "", ua_str, proxy_str, page_url
  773. )