Karim
/
EnglishPal
forked from mrlan/EnglishPal
1
0
Fork 0
Code Pull Requests 1 Activity

1、添加误删的IELTS单词标签 

2、简化评级流程（快很多，几乎与网站相同）
3、删除了原先用于提取词根的函数
Bug476-ZhangWeiHao-YuHuangtao
俞黄焘 2023-05-18 20:36:31 +08:00
parent c4378e73cd
commit a83f91a88d
1 changed files with 5 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
app/difficulty.py
View File

@ -24,7 +24,7 @@ def convert_test_type_to_difficulty_level(d):
:return:
"""
result = {}
L = list(d.keys()) # in dic, we have test types (e.g., CET4,CET6,BBC) for each word
L = list(d.keys()) # in d, we have test types (e.g., CET4,CET6,BBC) for each word
for k in L:
if 'CET4' in d[k]:
@ -33,30 +33,13 @@ def convert_test_type_to_difficulty_level(d):
result[k] = 5
elif 'CET6' in d[k] or 'GRADUATE' in d[k]:
result[k] = 6
elif 'OXFORD5000' in d[k]:
elif 'OXFORD5000' in d[k] or 'IELTS' in d[k]:
result[k] = 7
elif 'BBC' in d[k]:
result[k] = 8
return result # {'apple': 4, ...}
def simplify_the_words_dict(d):
"""
用于把保存了词库中评级后的词新建一个以词根为键、以同词根的最低等级为值
"""
stem = snowballstemmer.stemmer('english')
result = {}
for k in d: # j 在字典中
temp = stem.stemWord(k) # 提取j得词根
if temp not in result: # 如果这个词根不在结果字典中，则以词根为键、以dic中的等级作为值添加
result[temp] = d[k]
else: # 如果这个词在结果词典中，则比较一下单词的难度等级是否最小
if result[temp] > d[k]:
result[temp] = d[k]
return result
def get_difficulty_level_for_user(d1, d2):
"""
@ -65,14 +48,13 @@ def get_difficulty_level_for_user(d1, d2):
在d2的后面添加单词，没有新建一个新的字典
"""
d2 = convert_test_type_to_difficulty_level(d2) # 根据d2的标记评级{'apple': 4, 'abandon': 4, ...}
d2_simplified = simplify_the_words_dict(d2) # 提取d2的词根 {'appl': 4, 'abandon': 4, ...}
stem = snowballstemmer.stemmer('english')
for k in d1: # 用户的词
if k in d2: # 如果用户的词以原型的形式存在于词库d2中
continue # 无需评级，跳过
elif stem.stemWord(k) in d2_simplified: # 如果用户的词的词根存在于词库d2的词根库中
d2[k] = d2_simplified[k] # 按照词根进行评级
elif stem.stemWord(k) in d2: # 如果用户的词的词根存在于词库d2的词根库中
d2[k] = d2[stem.stemWord(k)] # 按照词根进行评级
break
else:
d2[k] = 3 # 如果k的词根都不在，那么就当认为是3级