def star_article():
try:
data=request.get_json()
user_id=data.get('user_id')
article_id=data.get('article_id')
Article=Meeting_article.query.filter_by(id=article_id).first()
Count=User_article.query.filter_by(user_id=user_id,title=Article.title).count()
print(Count)
if(Count!=0):
return responseError(Responses.EXIST_STAR)
userArticle=User_article(create_time=Article.create_time,meeting_name=Article.meeting_name,title=Article.title,
author=Article.author,abstract=Article.abstract,address=Article.address,
keyword=Article.keyword,user_id=user_id)
db.session.add(userArticle)
db.session.commit()
temp=User_article.query.filter_by(author=Article.author,abstract=Article.abstract,
address=Article.address,user_id=user_id).first()
newNote=Note(user_id=user_id,article_id=temp.id)
db.session.add(newNote)
db.session.commit()
return responseSuccess(Responses.OPERATION_SUCCESS)
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def register():
try:
data = request.get_json()
user_name = data.get("user_name")
user_account = data.get("user_account")
user_password = data.get("user_password")
user_repassword = data.get("user_repassword")
#验证两次密码是否相同
if user_password != user_repassword:
return responseError(Responses.NOT_SAME_PASSWORD)
#验证用户名是否已存在
if User.query.filter_by(username=user_name).count() != 0:
return responseError(Responses.EXIST_NAME)
#验证用户账号是否已存在
if User.query.filter_by(account=user_account).count() != 0:
return responseError(Responses.EXIST_ACCOUNT)
user=User(account=user_account,username=user_name,password=user_password)
db.session.add(user)
db.session.commit()
return responseSuccess(Responses.OPERATION_SUCCESS)
except Exception as e:
print(e)
db.session.rollback()
return responseError(Responses.PARAMETERS_ERROR)
def update_face():
try:
data = request.get_json()
user_id=data.get("user_id")
user_face=data.get("user_face")
user = User.query.filter_by(id=user_id).first()
user_avatar = "default.jpg"
if user.avatar != "default":
user_avatar = user.avatar
avatar_filename = random_filename(user_avatar)
user.avatar = avatar_filename
db.session.commit()
try:
avatar_path = url_for('static', filename="avatars/%s" % avatar_filename)
# 写入图片并保存
with open(avatar_path, 'wb') as w:
w.write(base64.b64decode(user_face))
# 获得修改后的头像的base64编码
with open(avatar_path, 'rb') as f:
stream = base64.b64encode(f.read())
avatar_base64_str = str(stream, encoding='utf-8')
return responseBody(data=[{'avatar': avatar_base64_str}])
except Exception as e:
print(e)
db.session.rollback()
return responseError(Responses.SAVE_FILE_FAIL)
except Exception as e:
print(e)
db.session.rollback()
return responseError(Responses.PARAMETERS_ERROR)
def login():
data = request.get_json()
account = data.get("account")
password = data.get("password")
try:
user = User.query.filter_by(account=account).first()
# 用户不存在
if not user:
return responseError(Responses.NO_USER_FOUND)
# 密码错误
if not user.password == password:
print(user.password)
print(password)
return responseError(Responses.INCORRECT_PASSWORD)
# 生成token
token = generate_token(user_id=user.id)
session["token"] = token
session["user_id"] = user.id
# 获取用户头像路径
user_avatar = "default.jpg"
# if user.avatar != "default":
# user_avatar = user.avatar
# avatar_path = url_for('static', filename="avatars/%s" % user_avatar)
# 用户账号，姓名，当前时间
account = user.account
username = user.username
if(user.gender==True):
userGender="男"
else:
userGender="女"
date = datetime.datetime.now()
# 保存登录时间
session["login_date"] = date
# 获取加入天数
day = date.day - user.create_time.day
# 获得头像base64编码
# with open(avatar_path, 'rb') as f:
# stream = base64.b64encode(f.read())
# avatar_base64_str = str(stream, encoding='utf-8')
# 返回数据
response_info = {
"name": username,
"userid": user.id,
"token":str(token),
"date": date.strftime("%Y-%m-%d %H:%M:%S"),
"day": day
}
return responseBody(data=response_info)
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def del_article():
try:
data=request.get_json()
article_id=data.get("article_id")
article=Meeting_article.query.filter_by(id=article_id).first()
if not article:
return responseError(Responses.NO_RECORD_FOUND)
db.session.delete(article)
db.session.commit()
return responseSuccess(Responses.OPERATION_SUCCESS)
except Exception as e:
print(e)
db.session.rollback()
return responseError(Responses.PARAMETERS_ERROR)
def search_title():
try:
data=request.get_json()
title=data.get("title")
page_info=data.get("page_info")
page_size = page_info.get("page_size")
page_num = page_info.get("page_num")
allPageNum = Meeting_article.query.filter(
Meeting_article.title.like("%" + title + "%")).count()
pagination=Meeting_article.query.filter(
Meeting_article.title.like("%"+title+"%"))\
.order_by(Meeting_article.create_time.desc()).paginate(per_page=page_num,page=page_size)
articles=pagination.items
response_info=[]
for i in articles:
response_info.append({
'id':i.id,
'title':i.title,
'create_time':i.create_time,
'meeting_name':i.meeting_name,
'authors':i.author,
'abstract':i.abstract,
'keyword':i.keyword,
'address':i.address
})
return responseBody(data={'data':response_info,'allPageNum':allPageNum})
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def getTop10():
try:
date=datetime.datetime.strptime('2020-01-01', '%Y-%m-%d')
Articles=Meeting_article.query.filter_by(create_time=date).all()
keywordDict={}
for i in Articles:
temp_list=list(i.keyword.split(';'))
for j in temp_list:
if (j not in keywordDict.keys()):
keywordDict[j]=1
else:
keywordDict[j]+=1
keywordDict['']=0
keywordDict['暂无关键词']=0
keyInfo=[]
valueInfo=[]
num=0
i=0
for k in sorted(keywordDict, key=keywordDict.__getitem__, reverse=True):
if(i<10):
keyInfo.append(k)
valueInfo.append(keywordDict[k])
num+=keywordDict[k]
i+=1
return responseBody(data=[{'key':keyInfo,'value':valueInfo,'num':num}])
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def getLineChart():
try:
Iccv_meeting=Meeting_article.query.filter_by(meeting_name='ICCV').all()
ICCV_list=[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
for i in Iccv_meeting:
date=datetime.datetime.strftime(i.create_time,'%Y-%m-%d')
if( int(2010) <= int(date[0:4]) and int( 2021)>=int(date[0:4]) ):
ICCV_list[(int(date[0:4])-2010)]=ICCV_list[(int(date[0:4])-2010)]+1
Cvpr_meeting=Meeting_article.query.filter_by(meeting_name='CVPR').all()
CVPR_list = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
for i in Cvpr_meeting:
date=datetime.datetime.strftime(i.create_time,'%Y-%m-%d')
if( int(2010) <= int(date[0:4]) and int( 2021)>=int(date[0:4]) ):
CVPR_list[(int(date[0:4])-2010)]=CVPR_list[(int(date[0:4])-2010)]+1
ECCV_list=[0,0,0,0,0,0,0,0,0,0,0,0]
Eccv_meeting = Meeting_article.query.filter_by(meeting_name='ECCV').all()
for i in Eccv_meeting:
date=datetime.datetime.strftime(i.create_time,'%Y-%m-%d')
if( int(2010) <= int(date[0:4]) and int( 2021)>=int(date[0:4]) ):
# print(int(date[0:4]))
ECCV_list[(int(date[0:4])-2010)]=ECCV_list[(int(date[0:4])-2010)]+1
return responseBody(data=[{'ECCV':ECCV_list,'CVPR':CVPR_list,'ICCV':ICCV_list}])
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def update_note():
try:
data = request.get_json()
user_id = data.get("user_id")
article_id = data.get("article_id")
note_info=data.get("note_info")
note = Note.query.filter_by(user_id=user_id, article_id=article_id).first()
if not note:
return responseError(Responses.NO_RECORD_FOUND)
note.note=note_info
db.session.commit()
return responseSuccess(Responses.OPERATION_SUCCESS)
except Exception as e:
print(e)
db.session.rollback()
return responseError(Responses.PARAMETERS_ERROR)
def update_infos():
try:
data = request.get_json()
user_id = int(data.get("user_id"))
# 列表分别为user_name,user_password,user_mobile,user_email,user_school,user_info,user_gender
user_list = (data.get("user_list"))
user = User.query.filter_by(id=user_id).first()
user.username = user_list[0]
user.password=user_list[1]
user.mobile = user_list[2]
user.email = user_list[3]
user.college = user_list[4]
user.info = user_list[5]
if(user_list[6]=="男"):
user.gender = True
else:
user.gender=False
db.session.commit()
return responseSuccess(Responses.OPERATION_SUCCESS)
except Exception as e:
print(e)
db.session.rollback()
return responseError(Responses.PARAMETERS_ERROR)
def logout():
try:
# 清空session
session["token"] = None
session["user_id"] = None
data = request.get_json()
user_id = int(data.get("user_id"))
return responseSuccess(Responses.OPERATION_SUCCESS)
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def get_note():
try:
data=request.get_json()
user_id=data.get("user_id")
article_id=data.get("article_id")
note=Note.query.filter_by(user_id=user_id,article_id=article_id).first()
note_info=note.note
return responseBody(data={'note_info':note_info})
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def search_byfile():
try:
data = request.get_json()
key_id = data.get('key_id')
file_name = data.get('file_name')
keyList = []
with open(r"F:\寒假作业2\结队作业2\Team_work\static\userupload\%s" % file_name, 'r')as f:
allKeyword = f.read()
keyList = allKeyword.split(';')
f.close()
return responseBody(data={'keyInfo':keyList[key_id-1]})
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def get_userInfo():
try:
data=request.get_json()
user_id=data.get('user_id')
user=User.query.filter_by(id=user_id).first()
gender="男"
if(user.gender==False):
gender="女"
user_info={'user_name':user.username,'user_gender':gender,'school':user.college,'password':user.password}
return responseBody(data=user_info)
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def upload_file():
try:
data=request.files.get('file')
# print(data.read())
filename = secure_filename(data.filename)
# 生成文件保存路径
save_path = r"F:\寒假作业2\结队作业2\Team_work\static\userupload\%s"%filename
# 保存文件
data.save(save_path)
List=[]
with open(save_path,'r')as f:
List=f.read().split(';')
f.close()
return responseBody(data={'keyList':List})
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
def getRisingSUN():
try:
childrens=[]
#ECCV
eccvArticles=Meeting_article.query.filter_by(meeting_name='ECCV').all()
eccvKeywordDict = {}
for i in eccvArticles:
temp_list = list(i.keyword.split(';'))
for j in temp_list:
if (j not in eccvKeywordDict.keys()):
eccvKeywordDict[j] = 1
else:
eccvKeywordDict[j] += 1
eccvKeywordDict[''] = 0
eccvKeywordDict['暂无关键词'] = 0
eccvNum=0
valueInfo = []
i = 0
for k in sorted(eccvKeywordDict, key=eccvKeywordDict.__getitem__, reverse=True):
if (i < 10):
valueInfo.append({'name':k,'value':eccvKeywordDict[k]})
eccvNum += eccvKeywordDict[k]
i+=1
childrens.append({'name':'EVVC','value':eccvNum,'children':valueInfo})
#ICCV
iccvArticles = Meeting_article.query.filter_by(meeting_name='ICCV').all()
iccvKeywordDict = {}
for i in iccvArticles:
temp_list = list(i.keyword.split(';'))
for j in temp_list:
if (j not in iccvKeywordDict.keys()):
iccvKeywordDict[j] = 1
else:
iccvKeywordDict[j] += 1
iccvKeywordDict[''] = 0
iccvKeywordDict['暂无关键词'] = 0
iccvNum = 0
valueInfo = []
i = 0
for k in sorted(iccvKeywordDict, key=iccvKeywordDict.__getitem__, reverse=True):
if (i < 10):
valueInfo.append({'name': k, 'value': iccvKeywordDict[k]})
iccvNum += iccvKeywordDict[k]
i += 1
childrens.append({'name': 'ICCV', 'value': iccvNum, 'children': valueInfo})
# CVPR
cvprArticles = Meeting_article.query.filter_by(meeting_name='CVPR').all()
cvprKeywordDict = {}
for i in cvprArticles:
temp_list = list(i.keyword.split(';'))
for j in temp_list:
if (j not in cvprKeywordDict.keys()):
cvprKeywordDict[j] = 1
else:
cvprKeywordDict[j] += 1
cvprKeywordDict[''] = 0
cvprKeywordDict['暂无关键词'] = 0
cvprNum = 0
valueInfo = []
i = 0
for k in sorted(cvprKeywordDict, key=cvprKeywordDict.__getitem__, reverse=True):
if (i < 10):
valueInfo.append({'name': k, 'value': cvprKeywordDict[k]})
cvprNum += cvprKeywordDict[k]
i += 1
childrens.append({'name': 'CVPR', 'value': cvprNum, 'children': valueInfo})
return responseBody(data=childrens)
except Exception as e:
print(e)
return responseError(Responses.PARAMETERS_ERROR)
