Update consult_func.py

consult_func.py

@@ -370,3 +370,40 @@ def keyword(
         if kmp(text,word)!=-1 and score>=0:
             return True
     return False
+
+
+#保存测试数据
+def save2(username,password,):
+    new_data2=[2,3,4,5.0,'该回答的分数',6,7.0,'该回答的文字',8.0,'该回答的文字','最终建议']
+    if (username=='' or password==''):
+        return '请填写完整信息'
+
+    user_login_df=pd.read_csv('./user_login/login.csv',encoding='gbk',converters = {'password':str})
+    user_login=user_login_df.set_index('username')['password'].T.to_dict()
+    user_data2_df=pd.read_csv('./user_data/data_2.csv',encoding='gbk',converters ={'senti_flt':float,'hobby_flt':float,'all_flt':float})
+
+    results=''
+    is_usn=(username in user_login)
+    if(is_usn):
+        pswd_crct=(password==user_login[username])
+    if(is_usn and pswd_crct):
+        results='登录成功，存储报告成功。'
+        user_data2_df.loc[len(user_data2_df)]=[username,datetime.date.today(),
+                                             datetime.datetime.now().strftime('%H:%M:%S')]+new_data2
+        user_data2_df.to_csv('./user_data/data_2.csv',encoding='gbk',index=False)
+    elif(is_usn):
+        results='密码错误！'
+    else:
+        user_login_df.loc[len(user_login_df)]=[username,password]
+        user_data2_df.loc[len(user_data2_df)]=[username,datetime.date.today(),
+                                             datetime.datetime.now().strftime('%H:%M:%S')]+new_data2
+        user_login_df.to_csv('./user_login/login.csv',encoding='gbk',index=False)
+        user_data2_df.to_csv('./user_data/data_2.csv',encoding='gbk',index=False)
+        results='账户不存在，已创建新的账户并保存数据。'
+
+        
+    return results
+
+#从作答项目获取存储内容
+def get_info():
+    pass