#!/usr/bin/env python2

##  Deprecated
##  Python 版本加密实现有问题，导致无法超过 999， 建议用 Javascript 版本

from Crypto.Cipher import AES
from pkcs7 import PKCS7Encoder
import time
import json
import base64
import requests


class WxCrypto(object):

    def __init__(self, key):
        self.key = key[:16]
        self.iv = key[:16]
        self.mode = AES.MODE_CBC

    def encrypt(self, text):
        aes = AES.new(self.key, self.mode, self.iv)
        encoder = PKCS7Encoder()
        pad_text = encoder.encode(text)
        cipher = aes.encrypt(pad_text)
        enc_cipher = base64.b64encode(cipher)
        return enc_cipher

    def decrypt(self, text):
        aes = AES.new(self.key, self.mode, self.iv)
        plain_text = aes.decrypt(base64.b64decode(text))
        encoder = PKCS7Encoder()
        plain_text = encoder.decode(plain_text)
        return plain_text


def update_score(session_id, score):
    headers = {
        'User-Agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 11_2_1 like Mac OS X) AppleWebKit/604.4.7 (KHTML, like Gecko) Mobile/15C153 MicroMessenger/6.6.1 NetType/WIFI Language/zh_CN',
        'Referer': 'https://servicewechat.com/wx7c8d593b2c3a7703/4/page-frame.html',
        'Accept-Language': 'zh-cn'
    }
    base_req = {
        'base_req': {
            'session_id': session_id,
            'fast': 1
        }
    }
    session = requests.Session()
    session.headers.update(headers)
    my_user_info_resp = session.post('https://mp.weixin.qq.com/wxagame/wxagame_getfriendsscore', json=base_req)
    if not my_user_info_resp.ok or not my_user_info_resp.json().get('my_user_info'):
        print(my_user_info_resp.json())
        raise Exception('something crash')
    times = my_user_info_resp.json()['my_user_info']['times']

    action_data = {
        'score': score,
        'times': times,
        'game_data': json.dumps({
            # 'seed': int(time.time()),
            # 'action': [[0.816, 1.09, False],
            #            [0.275, 2.21, True]],
            # 'musicList': [False, False]
        })
    }
    wx_crypto = WxCrypto(session_id[:16])
    action_data_cipher = wx_crypto.encrypt(json.dumps(action_data, separators=(',', ':')))
    data = {'action_data': action_data_cipher}
    data.update(base_req)
    result_resp = requests.post('https://mp.weixin.qq.com/wxagame/wxagame_settlement', json=data)
    print(result_resp.json())


session_id = 'xxxxxxxxxx'
update_score(session_id, score)