from dataclasses import dataclass
from typing import List


@dataclass
class Type:
    image_type: int
    image_field: List


@dataclass
class RequestConfig:
    url: str
    token: str

@dataclass
class Configs:
    request: RequestConfig
    type: Type



cet_local_config = RequestConfig(url='http://192.168.199.249:18050/ocr_system/cet', token='')
cet_test_config = RequestConfig(
    url='http://aihub-test.digitalyili.com/aiSquare/openApi/reasoning-services/rlocrxm/cettest/cet',
    token='9679c2b3-b90b-4029-a3c7-f347b4d242f7')
cet_sb_config = RequestConfig(
    url='http://aihub.digitalyili.com/aiSquare/openApi/reasoning-services/rlocrxm/cettest/cet',
    token='dcae8cc6-0e49-4db8-a2d2-94ef84da3636')

CET_CONFIGS = {
    'local': cet_local_config,
    'test': cet_test_config,
    'sb': cet_sb_config
}

# regbook
regbook_local_config = RequestConfig(url='http://192.168.199.249:18020/ocr_system/regbook', token='')
regbook_test_config = RequestConfig(
    url='http://aihub-test.digitalyili.com/aiSquare/openApi/reasoning-services/rlocrxm/hkbsb/regbook',
    token='9679c2b3-b90b-4029-a3c7-f347b4d242f7')
regbook_sb_config = RequestConfig(
    url='http://aihub.digitalyili.com/aiSquare/openApi/reasoning-services/rlocrxm/hkbsb/regbook',
    token='dcae8cc6-0e49-4db8-a2d2-94ef84da3636')

REGBOOK_CONFIGS = {
    'local': regbook_local_config,
    'test': regbook_test_config,
    'sb': regbook_sb_config
}

OCR_CONFIGS = {
    'cet': CET_CONFIGS,
    'regbook': REGBOOK_CONFIGS
}