import jieba

'''
项目配置
'''
# 每次从数据库获取的案子数量
PHHD_BATCH_SIZE = 10
# 没有查询到案子的等待时间(分钟)
SLEEP_MINUTES = 5
# 是否发送报错邮件
SEND_ERROR_EMAIL = True
# 处理批号（这里仅起声明作用）
BATCH_ID = ''

'''
别名配置
'''
# 使用别名中的value替换key。考虑到效率问题，只会替换第一个匹配到的key。
HOSPITAL_ALIAS = {
    '沐阳': ['沭阳'],
    '连水': ['涟水'],
    '唯宁': ['睢宁'],  # 雕宁
    '九〇四': ['904'],
    '漂水': ['溧水'],
}
DEPARTMENT_ALIAS = {
    '耳鼻喉': ['耳鼻咽喉'],
    '急症': ['急诊'],
}

'''
搜索过滤配置
'''
# 默认会过滤单字
HOSPITAL_FILTER = ['医院', '人民', '第一', '第二', '第三', '大学', '附属']

DEPARTMENT_FILTER = ['医', '伤', '西', '新']

'''
分词配置
'''
jieba.suggest_freq(('肿瘤', '医院'), True)
jieba.suggest_freq(('骨', '伤'), True)
jieba.suggest_freq(('感染', '性'), True)
jieba.suggest_freq(('胆', '道'), True)
jieba.suggest_freq(('脾', '胃'), True)

'''
出院记录缺页判断关键词配置
'''
DISCHARGE_KEY = {
    '入院诊断': ['入院诊断'],
    '入院情况': ['入院情况', '入院时情况', '入院时主要症状'],
    '入院日期': ['入院日期', '入院时间'],
    '诊疗经过': ['诊疗经过', '住院经过', '治疗经过'],
    '出院诊断': ['出院诊断'],
    '出院情况': ['出院情况', '出院时情况'],
    '出院日期': ['出院日期', '出院时间'],
    '出院医嘱': ['出院医嘱', '出院医瞩']
}


def get_batch_id():
    """
    获取处理批号
    :return: 处理批号
    """
    return BATCH_ID


def set_batch_id(batch_id):
    """
    修改处理批号哦
    :param batch_id: 新批号
    """
    global BATCH_ID
    BATCH_ID = batch_id