# coding: utf-8
import string
from .__version__ import __version__


# 模型版本只对应到第二层，第三层的改动表示模型兼容。
# 如: __version__ = '1.2.*'，对应的 MODEL_VERSION 都是 '1.2.0'
MODEL_VERSION = '.'.join(__version__.split('.', maxsplit=2)[:2]) + '.0'

EMB_MODEL_TYPES = [
    'conv',  # seq_len == 35, deprecated
    'conv-lite',  # seq_len == 69
    'conv-lite-s',  # seq_len == 35
    'densenet',  # seq_len == 70, deprecated
    'densenet-lite',  # seq_len == 70
    'densenet-s',  # seq_len == 35
    'densenet-lite-s',  # seq_len == 35
]
SEQ_MODEL_TYPES = ['lstm', 'gru', 'fc']

root_url = (
    'https://static.einplus.cn/cnocr/%s'
    % MODEL_VERSION
)
# name: (epochs, url)
AVAILABLE_MODELS = {
    'conv-lite-fc': (25, root_url + '/conv-lite-fc.zip'),
    'densenet-lite-gru': (39, root_url + '/densenet-lite-gru.zip'),
    'densenet-lite-fc': (40, root_url + '/densenet-lite-fc.zip'),
    'densenet-lite-s-gru': (35, root_url + '/densenet-lite-s-gru.zip'),
    'densenet-lite-s-fc': (40, root_url + '/densenet-lite-s-fc.zip'),
}

# 候选字符集合
NUMBERS = string.digits + string.punctuation
ENG_LETTERS = string.digits + string.ascii_letters + string.punctuation