import os
import configparser
# 加載配置文件
class Config(object):
  def __init__(self, config_file='configDemo.ini'):
    self._path = os.path.join(os.getcwd(), config_file)
    if not os.path.exists(self._path):
      raise FileNotFoundError("No such file: config.ini")
    self._config = configparser.ConfigParser()
    self._config.read(self._path, encoding='utf-8-sig')
    self._configRaw = configparser.RawConfigParser()
    self._configRaw.read(self._path, encoding='utf-8-sig')
  def get(self, section, name):
    return self._config.get(section, name)
  def getRaw(self, section, name):
    return self._configRaw.get(section, name)

# 主程序入口
# 檢查是否存在要搶購(gòu)的端口，然后進(jìn)入循環(huán)掃描
if len(skuids) != 1:
  logger.info('請(qǐng)準(zhǔn)備一件商品')
skuId = skuids[0]
flag = 1
# 循環(huán)掃描該商品是否有貨，有庫(kù)存即會(huì)自動(dòng)下單，無(wú)庫(kù)存則休眠后繼續(xù)掃描
while (1):
  try:
    # 初始化校驗(yàn)
    if flag == 1:
      logger.info('當(dāng)前是V3版本')
      validate_cookies()  # 校驗(yàn)登陸狀態(tài)
      getUsername()    # 獲取登陸用戶信息
      select_all_cart_item()  # 全選購(gòu)物車(chē)
      remove_item()      # 刪除購(gòu)物車(chē)
      add_item_to_cart(skuId)  # 增加搶購(gòu)的商品
    # 檢測(cè)配置文件修改
    if int(time.time()) - configTime >= 60:
      check_Config()
    logger.info('第' + str(flag) + '次 ')
    # 計(jì)數(shù)器
    flag += 1
    # 檢查庫(kù)存模塊
    inStockSkuid = check_stock(checksession, skuids, area)
    # 自動(dòng)下單模塊
    V3AutoBuy(inStockSkuid)
    # 休眠模塊
    timesleep = random.randint(1, 3) / 10
    time.sleep(timesleep)
    # 校驗(yàn)是否還在登錄模塊
    if flag % 100 == 0:
      V3check(skuId)
  except Exception as e:
    print(traceback.format_exc())
    time.sleep(10)

# 校驗(yàn)登陸狀態(tài)
def validate_cookies():
  for flag in range(1, 3):
    try:
      targetURL = 'https://order.jd.com/center/list.action'
      payload = {
        'rid': str(int(time.time() * 1000)),
      }
      resp = session.get(url=targetURL, params=payload, allow_redirects=False)
      if resp.status_code == requests.codes.OK:
        logger.info('登錄成功')
        return True
      else:
        logger.info('第【%s】次請(qǐng)重新獲取cookie', flag)
        time.sleep(5)
        continue
    except Exception as e:
      logger.info('第【%s】次請(qǐng)重新獲取cookie', flag)
      time.sleep(5)
      continue
  message.sendAny('腳本登錄cookie失效了，請(qǐng)重新登錄')
  sys.exit(1)

def add_item_to_cart(sku_id):
  # 請(qǐng)求添加商品url
  url = 'https://cart.jd.com/gate.action'
  payload = {
    'pid': sku_id,
    'pcount': 1,
    'ptype': 1,
  }
  # 返回結(jié)果
  resp = session.get(url=url, params=payload)
  # 套裝商品加入購(gòu)物車(chē)后直接跳轉(zhuǎn)到購(gòu)物車(chē)頁(yè)面
  if 'https://cart.jd.com/cart.action' in resp.url:
    result = True
  else:
   # 普通商品成功加入購(gòu)物車(chē)后會(huì)跳轉(zhuǎn)到提示 "商品已成功加入購(gòu)物車(chē)！" 頁(yè)面
    soup = BeautifulSoup(resp.text, "html.parser")
    result = bool(soup.select('h3.ftx-02')) # [<h3 class="ftx-02">商品已成功加入購(gòu)物車(chē)！</h3>]
  if result:
    logger.info('%s 已成功加入購(gòu)物車(chē)', sku_id)
  else:
    logger.error('%s 添加到購(gòu)物車(chē)失敗', sku_id)

def submit_order(session, risk_control, sku_id, skuids, submit_Time, encryptClientInfo, is_Submit_captcha, payment_pwd,
         submit_captcha_text, submit_captcha_rid):
  # 提交端口的url
  url = 'https://trade.jd.com/shopping/order/submitOrder.action'
  # 提交參數(shù)
  data = {
    'overseaPurchaseCookies': '',
    'vendorRemarks': '[]',
    'submitOrderParam.sopNotPutInvoice': 'false',
    'submitOrderParam.trackID': 'TestTrackId',
    'submitOrderParam.ignorePriceChange': '0',
    'submitOrderParam.btSupport': '0',
    'riskControl': risk_control,
    'submitOrderParam.isBestCoupon': 1,
    'submitOrderParam.jxj': 1,
    'submitOrderParam.trackId': '9643cbd55bbbe103eef18a213e069eb0', # Todo: need to get trackId
    'submitOrderParam.needCheck': 1,
  }
  # 如果用到京豆會(huì)需要輸入支付密碼
  def encrypt_payment_pwd(payment_pwd):
    return ''.join(['u3' + x for x in payment_pwd])
  # 校驗(yàn)支付密碼
  if len(payment_pwd) > 0:
    data['submitOrderParam.payPassword'] = encrypt_payment_pwd(payment_pwd)
  # 請(qǐng)求報(bào)文頭
  headers = {
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.100 Safari/531.36",
    "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3",
    "Referer": "http://trade.jd.com/shopping/order/getOrderInfo.action",
    "Connection": "keep-alive",
    'Host': 'trade.jd.com',
  }
  # 訂單提交會(huì)嘗試兩次
  for count in range(1, 3):
    logger.info('第[%s/%s]次嘗試提交訂單', count, 3)
    try:
      # 可能會(huì)存在的校驗(yàn)碼
      if is_Submit_captcha:
        captcha_result = page_detail_captcha(session, encryptClientInfo)
        # 驗(yàn)證碼服務(wù)錯(cuò)誤
        if not captcha_result:
          logger.error('驗(yàn)證碼服務(wù)異常')
          continue
        data['submitOrderParam.checkcodeTxt'] = submit_captcha_text
        data['submitOrderParam.checkCodeRid'] = submit_captcha_rid
      # 提交訂單
      resp = session.post(url=url, data=data, headers=headers)
      resp_json = json.loads(resp.text)
      logger.info('本次提交訂單耗時(shí)[%s]毫秒', str(int(time.time() * 1000) - submit_Time))
      # 判斷是否提交成功
      if resp_json.get('success'):
        logger.info('訂單提交成功! 訂單號(hào)：%s', resp_json.get('orderId'))
        return True
      else:
        # 提交失敗返回的多種原因
        resultMessage, result_code = resp_json.get('message'), resp_json.get('resultCode')
        if result_code == 0:
          # self._save_invoice()
          if '驗(yàn)證碼不正確' in resultMessage:
            resultMessage = resultMessage + '(驗(yàn)證碼錯(cuò)誤)'
            logger.info('提交訂單驗(yàn)證碼[錯(cuò)誤]')
            continue
          else:
            resultMessage = resultMessage + '(下單商品可能為第三方商品，將切換為普通發(fā)票進(jìn)行嘗試)'
        elif result_code == 60077:
          resultMessage = resultMessage + '(可能是購(gòu)物車(chē)為空 或 未勾選購(gòu)物車(chē)中商品)'
        elif result_code == 60123:
          resultMessage = resultMessage + '(需要在payment_pwd參數(shù)配置支付密碼)'
        elif result_code == 60070:
          resultMessage = resultMessage + '(省份不支持銷(xiāo)售)'
          skuids.remove(sku_id)
          logger.info('[%s]類(lèi)型口罩不支持銷(xiāo)售', sku_id)
        logger.info('訂單提交失敗, 錯(cuò)誤碼：%s, 返回信息：%s', result_code, resultMessage)
        logger.info(resp_json)
        return False
    except Exception as e:
      print(traceback.format_exc())
      continue