import base64
import json
import urllib.request

ENCODING = 'utf-8'


def get_img_base64(img_file):
    with open(img_file, 'rb') as infile:
        s = infile.read()
        return base64.b64encode(s).decode(ENCODING)


def predict(url, appcode, img_base64, kv_configure):
    param = {}
    param['image'] = img_base64
    if kv_configure is not None:
        param['configure'] = json.dumps(kv_configure)
    body = json.dumps(param)
    data = bytes(body, "utf-8")

    headers = {'Authorization': 'APPCODE %s' % appcode}
    request = urllib.request.Request(url=url, headers=headers, data=data)
    try:
        response = urllib.request.urlopen(request, timeout=10)
        return response.code, response.headers, response.read()
    except urllib.request.HTTPError as e:
        return e.code, e.headers, e.read()


def demo():
    AppKey = '204917800'
    AppSecret = 'rmIcdp5is0hM2e4KBC3Ned9aFGBYGU2F'
    AppCode = '3a9b1eeacf6a4a7c9b6f3157feb61fa7'

    url = 'https://form.market.alicloudapi.com/api/predict/ocr_table_parse'
    img_file = '1.png'
    # configure = {'side': 'face'}
    # 如果没有configure字段，configure设为None
    # configure = None
    configure = {"format": "html", "dir_assure": False, "line_less": True}

    img_base64data = get_img_base64(img_file)
    stat, header, content = predict(url, AppCode, img_base64data, configure)
    if stat != 200:
        print('Http status code: ', stat)
        print('Error msg in header: ', header['x-ca-error-message'] if 'x-ca-error-message' in header else '')
        print('Error msg in body: ', content)
        exit()
    result_str = content

    print(result_str.decode(ENCODING))
    # result = json.loads(result_str)


if __name__ == '__main__':
    demo()