diff --git a/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py b/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py index 1d41639e..69f0d165 100644 --- a/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py +++ b/dsSchoolBuddy/ElasticSearch/Utils/EsSearchUtil.py @@ -614,25 +614,25 @@ class EsSearchUtil: seg_set = set(seg_list) # 检查是否已有学生信息,如果没有则从ES加载 - if user_id not in student_info: + if user_id not in self.student_info: # 从ES加载学生信息 info_from_es = self.get_student_info_from_es(user_id) if info_from_es: - student_info[user_id] = info_from_es + self.student_info[user_id] = info_from_es logger.info(f"从ES加载用户 {user_id} 的信息: {info_from_es}") else: - student_info[user_id] = {} + self.student_info[user_id] = {} # 提取并更新年级信息 grade_found = False - for grade, keywords in GRADE_KEYWORDS.items(): + for grade, keywords in self.GRADE_KEYWORDS.items(): for keyword in keywords: if keyword in seg_set: - if 'grade' not in student_info[user_id] or student_info[user_id]['grade'] != grade: - student_info[user_id]['grade'] = grade + if 'grade' not in self.student_info[user_id] or self.student_info[user_id]['grade'] != grade: + self.student_info[user_id]['grade'] = grade logger.info(f"提取到用户 {user_id} 的年级信息: {grade}") # 保存到ES - self.save_student_info_to_es(user_id, student_info[user_id]) + self.save_student_info_to_es(user_id, self.student_info[user_id]) grade_found = True break if grade_found: @@ -646,11 +646,11 @@ class EsSearchUtil: if match: grade_num = match.group(1) grade = f"{grade_num}年级" - if 'grade' not in student_info[user_id] or student_info[user_id]['grade'] != grade: - student_info[user_id]['grade'] = grade + if 'grade' not in self.student_info[user_id] or self.student_info[user_id]['grade'] != grade: + self.student_info[user_id]['grade'] = grade logger.info(f"通过正则提取到用户 {user_id} 的年级信息: {grade}") # 保存到ES - self.save_student_info_to_es(user_id, student_info[user_id]) + self.save_student_info_to_es(user_id, self.student_info[user_id]) except Exception as e: logger.error(f"提取学生信息失败: {str(e)}", exc_info=True) diff --git a/dsSchoolBuddy/ElasticSearch/Utils/__pycache__/EsSearchUtil.cpython-310.pyc b/dsSchoolBuddy/ElasticSearch/Utils/__pycache__/EsSearchUtil.cpython-310.pyc index c2e27783..fb2d085c 100644 Binary files a/dsSchoolBuddy/ElasticSearch/Utils/__pycache__/EsSearchUtil.cpython-310.pyc and b/dsSchoolBuddy/ElasticSearch/Utils/__pycache__/EsSearchUtil.cpython-310.pyc differ