def getList(page, token, pid=0):
userId = getUserId(token)
if userId['code'] == 0:
lists = Topic.objects.filter(check__in=[1, 3],
user_delete=None,
system_delete=None)
if pid:
lists = lists.filter(pid=pid)
count = lists.count()
listdata = list(
lists.order_by('-id').values('id', 'pid', 'name', 'headimg',
'title', 'content', 'imgs', 'top',
'visit', 'create_time'))
# listdata.sort(key=lambda keys: keys['id'], reverse=True)
data = listdata[(page - 1) * 10:page * 10]
for i in data:
i['desca'] = i['content'][:30]
i['category'] = Category.objects.get(id=i['pid']).name
i['imgs'] = eval(i['imgs']) if i['imgs'] else None
i['likeCount'] = Like.objects.filter(tid=i['id'],
cancel=False).count()
i['isLiked'] = True if Like.objects.filter(
tid=i['id'], cancel=False, uid=userId['data']['id']) else False
del i['content']
del i['pid']
return {
'code': 0,
'msg': '获取第' + str(page) + '页的数据成功',
'count': count,
'data': data
}
else:
return userId
def doAuth(usr, pwd, token):
userId = getUserId(token)
if userId['code'] == 0:
userId = userId['data']['id']
# 判断学号是否已经被绑定
user = MyUser.objects.filter(student_id=usr)
if user and user[0].id != userId:
return {'code': 1, 'msg': '学号已被绑定！'}
usr = str(usr)
pwd = str(pwd)
pwd = str(b64encode(bytes(pwd, encoding='utf-8')))
url = 'http://jxglstu.hfut.edu.cn:7070/appservice/home/appLogin/login.action'
data = {"username": usr, "password": pwd[2:-1], "identity": "0"}
try:
r = post(url, data=data, timeout=3).text
except:
return {'code': -2, 'msg': '发生错误，请检查网络！如非网络错误请联系管理员！'}
try:
code = loads(r)['code']
if code == 200:
data = loads(r)['obj']['business_data']
saveInfo(data, token)
return {
'code': 0,
'msg': '认证身份成功！个人信息已被更新！',
}
elif code == 1001:
return {'code': -1, 'msg': '学号或密码错误！'}
except:
return {'code': -3, 'msg': '学校闭网中，目前无法完成身份认证！'}
else:
return userId
def sresult(token, word, page):
userId = getUserId(token)
if userId['code'] == 0:
res = Topic.objects.filter(check__in=[1, 3],
user_delete=None,
system_delete=None).filter(
Q(title__contains=word)
| Q(content__contains=word))
count = res.count()
listdata = list(
res.values('id', 'name', 'headimg', 'title', 'content', 'imgs',
'top', 'like', 'dislike', 'visit', 'create_time'))
listdata.sort(key=lambda keys: keys['id'], reverse=True)
data = listdata[(page - 1) * 10:page * 10]
print(data)
for i in data:
i['desca'] = i['content'][:30]
del i['content']
return {
'code': 0,
'msg': '获取关键词"' + str(word) + '",第' + str(page) + '页的数据成功',
'count': count,
'data': data
}
else:
return userId
def getUserInfo(token):
userId = getUserId(token)
if userId['code'] == 0:
userId = userId['data']['id']
else:
exit(0)
user = MyUser.objects.get(id=userId)
info = dict()
info['id'] = user.id
info['wechatname'] = user.wechatname
info['student_id'] = user.student_id
info['auth'] = user.auth
info['name'] = user.name
info['department'] = user.department
info['major'] = user.major
info['adminclass'] = user.adminclass
info['dormitory'] = user.dormitory
info['email'] = user.email
info['phone'] = user.phone
info['qq'] = user.qq
info['nickname'] = user.nickname
info['headimg'] = user.headimg
info['sign'] = user.sign
info['gender'] = user.gender
info['birthday'] = user.birthday
return {
'code': 0,
'msg': '获取个人资料成功',
'data': info
}
def setUserInfo(data, token):
userId = getUserId(token)
if userId['code'] == 0:
userId = userId['data']['id']
else:
exit(0)
d = {k: v for k, v in data.items() if v}
MyUser.objects.filter(id=userId).update(**d)
def setWechatInfo(wechatname, headimg, token):
userId = getUserId(token)
if userId['code'] == 0:
userId = userId['data']['id']
else:
exit(0)
d = {
'wechatname': wechatname,
'headimg': headimg
}
user = MyUser.objects.filter(id=userId).update(**d)
def getCate(token):
userId = getUserId(token)
if userId['code'] == 0:
category = Category.objects.all()
return {
'code': 0,
'msg': '获取分类成功',
'count': category.count(),
'data': list(category.values())
}
else:
return userId
def my_topic(token):
userId = getUserId(token)
if userId['code'] == 0:
uid = userId['data']['id']
topics = list(
Topic.objects.filter(uid_id=uid,
user_delete=None,
system_delete=None).values())
for topic in topics:
topic['category'] = Category.objects.get(id=topic['pid_id']).name
return {'code': 0, 'msg': '获得话题成功', 'data': {'topics': topics}}
else:
return userId
def add(tid, content, token):
userId = getUserId(token)
if userId['code'] == 0:
if not Topic.objects.filter(
id=tid, user_delete=None, system_delete=None):
return {'code': -1, 'msg': '话题不存在'}
timenow = time.localtime()
now = time.strftime("%Y-%m-%d %H:%M:%S", timenow)
uid = userId['data']['id']
Comment.objects.create(uid_id=uid,
pid=tid,
content=content,
create_time=now)
return {'code': 0, 'msg': '新增评论成功'}
else:
return userId
def getDetail(token, id):
userId = getUserId(token)
if userId['code'] == 0:
detail = Topic.objects.get(id=id)
detail.visit += 1
detail.save()
d = dict()
d['id'] = detail.id
d['category'] = detail.pid.name
d['name'] = detail.name
d['headimg'] = detail.headimg
d['title'] = detail.title
d['content'] = detail.content
d['imgs'] = eval(detail.imgs) if detail.imgs else None
d['likeCount'] = Like.objects.filter(tid=detail.id,
cancel=False).count()
d['isLiked'] = True if Like.objects.filter(
tid=detail.id, cancel=False, uid=userId['data']['id']) else False
d['visit'] = detail.visit
d['create_time'] = detail.create_time
comments = Comment.objects.filter(pid=detail.id)
comment_list = []
for comment in comments:
comment_list.append({
'id': comment.id,
'name': comment.uid.name + ' ' + comment.uid.adminclass,
'headimg': comment.uid.headimg,
'create_time': comment.create_time,
'conetnt': comment.content
})
d['comment'] = comment_list
if detail.pid.needreply:
reply = Reply.objects.get(tid=id)
r = dict()
r['id'] = reply.id
r['rname'] = reply.rname
r['content'] = reply.content
r['create_time'] = reply.create_time
data = {'detail': d, 'reply': r}
else:
data = {
'detail': d,
}
return {'code': 0, 'msg': '获取话题详细内容成功', 'data': data}
else:
return userId
def saveInfo(data, token):
userId = getUserId(token)
if userId['code'] == 0:
userId = userId['data']['id']
else:
exit(0)
d = {
'email': data['account_email'],
'adminclass': data['adminclass_name'],
'birthday': data['birthday'],
'department': data['depart_name'],
'gender': data['gender'],
'major': data['major_name'],
'phone': data['mobile_phone'],
'student_id': data['user_code'],
'name': data['user_name'],
'auth': 1
}
user = MyUser.objects.filter(id=userId).update(**d)
def addTopic(token, data):
timenow = localtime()
day = strftime("%Y-%m-%d", timenow)
time = strftime("%Y-%m-%d %H:%M:%S", timenow)
userId = getUserId(token)
if userId['code'] == 0:
userId = userId['data']['id']
if Topic.objects.filter(uid=userId,
create_time__contains=day,
system_delete__isnull=True):
return {'code': -1, 'msg': '今天已经发表过话题了，每人每天仅限一条哦！'}
d = dict()
user = MyUser.objects.get(id=userId)
if data['hide']:
d['name'] = user.nickname + ' ' + user.major
d['headimg'] = ''
else:
d['name'] = user.name + ' ' + user.major
d['headimg'] = user.headimg
d['title'] = data['title']
d['pid'] = data['category']
d['uid'] = userId
d['content'] = data['content']
try:
d['imgs'] = eval(data['imgs'])
except:
d['imgs'] = data['imgs'].split(
',') if data['imgs'] else data['imgs']
d['hide'] = data['hide']
d['create_time'] = time
print(d)
topic = Topic.objects.create(**d)
topic.save()
return {'code': 0, 'msg': '新增话题成功！'}
else:
return userId
def like(token, tid):
userId = getUserId(token)
if userId['code'] == 0:
uid = userId['data']['id']
if not Topic.objects.filter(
id=tid, user_delete=None, system_delete=None):
return {'code': -1, 'msg': '话题不存在'}
like = Like.objects.filter(uid=uid, tid=tid)
if like:
cancel = not like[0].cancel
like.update(cancel=cancel)
if cancel:
msg = '取消点赞成功'
else:
msg = '点赞成功'
else:
Like.objects.create(uid=uid, tid=tid, cancel=False)
msg = '点赞成功'
return {'code': 0, 'msg': msg}
else:
return userId