import requests

link = requests.get('https://www.shanbay.com/api/v1/vocabtest/category/?_=1593431637392')
#先用requests下载链接
js_link = link.json()
#解析下载到的内容
bianhao = int(input('''请输入你选择的词库编号，按enter确认
1，GMAT  2，考研  3，高考  4，四级  5，六级
6，英专  7，托福  8，GRE  9，雅思  10，任意
>'''))
#让用户选择自己想测的词库，输入数字编号。int()来转换数据类型
ciku = js_link['data'][bianhao-1][0]
#利用用户输入数字编号，获取题库代码。
test = requests.get('https://www.shanbay.com/api/v1/vocabtest/vocabularies/?category=' + ciku)
#下载用户测试的50个单词
words = test.json()
#对test进行解析
danci = []
#新增list，用于统计用户认识的单词
words_knows =[]
#创建空列表，用于启示录用户认识的单词
not_knows = []
#创建一个空的列表，用于记录用户不认识的单词
print('测试现在开始。如果你认识这个单词，请输入Y，否则直接按enter：')
n = 0
for x in words['data']:
    n = n+1
    print('\n第'+str(n)+'个：'+x['content'])
    answer = input('认识请敲Y，否则敲ENTER：')
    #让用户输入是否认识
    if answer == 'Y':
        danci.append(x['content'])
        words_knows.append(x)
        #如果认识，就把单词追加进列表
    else:
        not_knows.append(x)
        #如果不认识，追加进列表

print('\n在上述'+str(len(words['data']))+'个单词中，有'+str(len(danci))+'个是你觉得自己认识的，它们是：')
print(danci)

print('现在我们来检测一下，你有没有真正掌握它们：')
wrong_words = []
#空列表，用于记录错的单词
right_num = 0
#用于记录对单词的数量
for y in words_knows:
    print('\n\n'+'A:'+y['definition_choices'][0]['definition'])
    print('B:'+y['definition_choices'][1]['definition'])
    print('C:'+y['definition_choices'][2]['definition'])
    print('D:'+y['definition_choices'][3]['definition'])
    xuanze = input('请选择单词\"'+y['content']+'\"的正确翻译（输入字母即可）：')
    dic = {'A':y['definition_choices'][0]['rank'],'B':y['definition_choices'][1]['rank'],'C':y['definition_choices'][2]['rank'],'D':y['definition_choices'][3]['rank']}
    #创建字典，搭建起ABCD和四个rank值的映射关系
    if dic[xuanze] == y['rank']:
        right_num += 1
    else:
        wrong_words.append(y)

print('现在，到了公布成绩的时刻：')
print('在'+str(len(words['data']))+'个'+js_link['data'][bianhao-1][1]+'词汇当中，你认识其中'+str(len(danci))+'个，实际掌握'+str(right_num)+'个，错误'+str(len(wrong_words))+'个。')
#例：在50个托福单词中，你认识X个，实际掌握X个，错误X个

save = input('是否打印并保存你的错词集？填入Y或N：')
#询问用户，是否要打印并保存错题集
if save == 'Y':
    f = open('错题集.txt','a+')
    print('你记错的单词有：')
    f.write('你记错的单词有：\n')
    m = 0
    for z in wrong_words:
        m += 1
        print(z['content'])
        f.write(str(m)+'.'+z['content']+'\n')
    print('你不认识的单词有：')
    f.write("你没记住的单词有：\n")
    s = 0
    for x in not_knows:
        s += 1
        print(x['content'])
        f.write(str(s)+'.'+x['content']+'\n')
    print('错词和没记住的词已保存至当前文件目录下，下次见！')
else:
    print('下次见')