def delete(self):
obj = Topics.get_by_uid(self.input.uid)
for tmp_obj in TopicAppends.filter_by_query(topic_id=obj.id):
tmp_obj.update(available=0)
for tmp_obj in Comments.filter_by_query(topic_id=obj.id):
tmp_obj.update(available=0)
obj.update(available=0)
def post(self):
obj = Topics.get_by_uid(self.input.uid)
if self.input.action in ['up', 'down']:
topic_up_down_obj = TopicUpDown.filter_by_query(
topic_id=obj.id,
user_id=session['user_info']['id'],
show_deleted=True).first()
if topic_up_down_obj:
topic_up_down_obj.available = 1
else:
topic_up_down_obj = TopicUpDown()
topic_up_down_obj.topic_id = obj.id
topic_up_down_obj.user_id = session['user_info']['id']
if self.input.action == 'up':
topic_up_down_obj.action = True
else:
topic_up_down_obj.action = False
topic_up_down_obj.save()
return {
'up_count':
TopicUpDown.filter_by_query(topic_id=obj.id, action=True).count(),
'down_count':
TopicUpDown.filter_by_query(topic_id=obj.id, action=False).count()
}
def post(self):
obj = Topics.get_by_uid(self.input.uid)
if obj.user_id != session['user_info']['id']:
raise LogicError('你不可以追加别人的帖子哦')
append_obj = TopicAppends.create_by_uid()
append_obj.topic_id = obj.id
append_obj.content = self.input.content
append_obj.save()
def post(self):
obj = Topics.get_by_uid(self.input.uid)
commend_obj = Comments.create_by_uid()
commend_obj.topic_id = obj.id
commend_obj.content = self.input.comment
commend_obj.user_id = session['user_info']['id']
obj.last_reply_user_id = session['user_info']['id']
obj.last_reply_time = self.get_datetime_now()
obj.save()
commend_obj.save()
def get(self):
if self.input.uid:
obj = Comments.get_by_uid(self.input.uid)
data = obj.to_dict()
data['user'] = Users.get_by_id(
data['user_id']).to_dict(remove_fields_list=['password'])
data['topic'] = Topics.get_by_id(data['topic_id']).to_dict()
else:
data = Comments.filter_by_query()
if self.input.tab_id:
data.filter_by_query(tab_id=self.input.tab_id)
elif self.input.sub_tab_id:
data.filter_by_query(sub_tab_id=self.input.sub_tab_id)
if self.input.sk_ and self.input.sv_:
data = data.filter(
getattr(Comments,
self.input.sk_).like(f'%{self.input.sv_}%'))
if self.input.odb_ and self.input.odt_ in ['asc', 'desc']:
data = data.order_by(
getattr(getattr(Comments, self.input.odb_),
self.input.odt_)())
else:
data = data.order_by(Comments.time_create.desc())
total = None
if self.input.page and self.input.page_size:
total = data.count()
data = data.offset(
(int(self.input.page) - 1) *
int(self.input.page_size)).limit(int(self.input.page_size))
data = [_t.to_dict() for _t in data]
user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(
i['user_id'] for i in data)), Users.available == 1)
}
topic_dict = {
i.id: i.to_dict()
for i in Topics.query.filter(
Topics.id.in_(set(
i['topic_id'] for i in data)), Topics.available == 1)
}
for _d in data:
_d['user'] = user_dict[_d['user_id']]
_d['topic'] = topic_dict[_d['topic_id']]
if total is not None:
data = {'total': total, 'list': data}
return data
def post(self):
obj = Topics.get_by_uid(self.input.uid)
# 自己不能给自己感谢
if session['user_info']['id'] == obj.user_id:
raise LogicError('自己不能感谢自己哦')
if not TopicThank.filter_by_query(
topic_id=obj.id, user_id=session['user_info']['id']).first():
thank_obj = TopicThank()
thank_obj.topic_id = obj.id
thank_obj.user_id = session['user_info']['id']
thank_obj.save()
def put(self):
obj = Topics.get_by_uid(self.input.uid)
if self.input.title:
obj.title = self.input.title
if self.input.content:
obj.content = self.input.content
obj.content_length = len(self.input.content)
if self.input.sub_tab_id:
sub_tab_obj = SubTabs.get_by_id(self.input.sub_tab_id)
obj.sub_tab_id = sub_tab_obj.id
obj.tab_id = sub_tab_obj.tab_id
obj.save()
def post(self):
obj = Topics.create_by_uid()
obj.title = self.input.title
sub_tab_obj = SubTabs.get_by_id(self.input.sub_tab_id)
obj.sub_tab_id = self.input.sub_tab_id
tab_obj = Tabs.get_by_id(sub_tab_obj.tab_id)
obj.tab_id = tab_obj.id
if self.input.content:
obj.content = self.input.content
obj.content_length = len(self.input.content)
obj.user_id = session['user_info']['id']
obj.save()
def post(self):
obj = Topics.get_by_uid(self.input.uid)
if session['user_info']['id'] == obj.user_id:
raise LogicError('自己不能收藏自己的主题哦')
topic_fav_obj = TopicFav.filter_by_query(
topic_id=obj.id,
user_id=session['user_info']['id'],
show_deleted=True).first()
if self.input.action == 'add':
if topic_fav_obj:
topic_fav_obj.available = 1
else:
topic_fav_obj = TopicFav()
topic_fav_obj.topic_id = obj.id
topic_fav_obj.user_id = session['user_info']['id']
topic_fav_obj.save()
elif self.input.action == 'cal':
if topic_fav_obj:
topic_fav_obj.available = 0
topic_fav_obj.save()
def post(self):
obj = Topics.get_by_uid(self.input.uid)
obj.view_count += 1
obj.save()
def get(self):
data = Topics.get_by_uid(self.input.uid).to_dict()
data['user'] = Users.get_by_id(data['user_id']).to_dict()
if data['last_reply_user_id']:
data['last_reply_user'] = Users.get_by_id(
data['last_reply_user_id']).to_dict()
data['tab'] = Tabs.get_by_id(data['tab_id']).to_dict()
data['sub_tab'] = SubTabs.get_by_id(data['sub_tab_id']).to_dict()
data['appends'] = [
i.to_dict()
for i in TopicAppends.filter_by_query(topic_id=data['id'])
]
comment_objs = Comments.filter_by_query(topic_id=data['id'])
data['comment_count'] = comment_objs.count()
if data['comment_count'] < 100:
comment_list = [i.to_dict() for i in comment_objs]
for index, comment in enumerate(comment_list):
comment['index'] = index + 1
else:
page_size = 50
page = int(self.input.page) if self.input.page else 1
comment_objs = comment_objs.offset(
(page - 1) * page_size).limit(page_size)
comment_list = [i.to_dict() for i in comment_objs]
for index, comment in enumerate(comment_list):
comment['index'] = (page - 1) * page_size + index + 1
comment_user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(
i['user_id'] for i in comment_list)), Users.available == 1)
}
for comment in comment_list:
comment['user'] = comment_user_dict[comment['user_id']]
comment['is_thank'] = False
data['up_count'] = TopicUpDown.filter_by_query(topic_id=data['id'],
action=True).count()
data['down_count'] = TopicUpDown.filter_by_query(topic_id=data['id'],
action=False).count()
data['is_fav'] = False
data['is_thank'] = False
if session.get('user_info'):
if TopicFav.filter_by_query(
topic_id=data['id'],
user_id=session['user_info']['id']).first():
data['is_fav'] = True
if TopicThank.filter_by_query(
topic_id=data['id'],
user_id=session['user_info']['id']).first():
data['is_thank'] = True
comment_thank_objs = CommentThank.query.filter(
CommentThank.comment_id.in_(
set([i['id'] for i in comment_list])),
CommentThank.available == 1)
comment_thank_id_list = [i.comment_id for i in comment_thank_objs]
for comment in comment_list:
if comment['id'] in comment_thank_id_list:
comment['is_thank'] = True
data['comments'] = comment_list
return data
def get(self):
data = dict()
user_info = Users.get_by_query(username=self.input.username).to_dict(remove_fields_list=['password'])
data['user'] = user_info
if user_info['privacy_level'] == 1 and not session.get('is_login'):
raise LogicError('没有权限访问此用户的信息哦')
if user_info['privacy_level'] == 2:
if not session.get('user_info') or session.get('user_info')['id'] != user_info['id']:
raise LogicError('没有权限访问此用户的信息哦')
topic_objs = Topics.filter_by_query(user_id=user_info['id'])
topic_count = topic_objs.count()
data['total'] = topic_count
if topic_count > 100:
page = int(self.input.page) if self.input.page else 1
page_size = 50
topic_objs = topic_objs.order_by(
Topics.time_create.desc()
).order_by(Topics.id.desc()).offset((page - 1) * page_size).limit(page_size)
else:
topic_objs = topic_objs.order_by(Topics.time_modify.desc()).order_by(Topics.id.desc())
topic_list = [_t.to_dict() for _t in topic_objs]
last_reply_user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(i['last_reply_user_id'] for i in topic_list)),
Users.available == 1
)
}
tab_dict = {
i.id: i.to_dict()
for i in Tabs.query.filter(Tabs.id.in_(set(i['tab_id'] for i in topic_list)), Tabs.available == 1)
}
sub_tab_dict = {
i.id: i.to_dict()
for i in SubTabs.query.filter(
SubTabs.id.in_(set(i['sub_tab_id'] for i in topic_list)),
SubTabs.available == 1
)
}
comment_count_dict = dict()
for c in Comments.query.filter(
Comments.topic_id.in_(set(i['id'] for i in topic_list)),
Comments.available == 1
):
if comment_count_dict.get(c.topic_id):
comment_count_dict[c.topic_id] += 1
else:
comment_count_dict[c.topic_id] = 1
up_down_dict_list = defaultdict(list)
for up_down_obj in TopicUpDown.query.filter(
TopicUpDown.topic_id.in_(set(i['id'] for i in topic_list)),
TopicUpDown.available == 1
):
up_down_dict_list[up_down_obj.topic_id].append(up_down_obj.action)
for _t in topic_list:
if _t['last_reply_user_id']:
_t['last_reply_user'] = last_reply_user_dict[_t['last_reply_user_id']]
_t['tab'] = tab_dict[_t['tab_id']]
_t['sub_tab'] = sub_tab_dict[_t['sub_tab_id']]
_t['comment_count'] = comment_count_dict.get(_t['id'], 0)
# 获取up和down数量
if up_down_dict_list.get(_t['id']):
_t['up_count'] = len([i for i in up_down_dict_list[_t['id']] if i is True])
_t['down_count'] = len([i for i in up_down_dict_list[_t['id']] if i is False])
else:
_t['up_count'] = 0
_t['down_count'] = 0
data['list'] = topic_list
return data
def get(self):
data = dict()
user_info = Users.get_by_query(username=self.input.username).to_dict(remove_fields_list=['password'])
user_info['fav_user_count'] = UserFavUser.filter_by_query(user_id=user_info['id']).count()
user_info['be_fav_user_count'] = UserFavUser.filter_by_query(fav_user_id=user_info['id']).count()
data['user'] = user_info
data['is_open'] = True
data['is_fav'] = False
if session.get('is_login'):
if UserFavUser.filter_by_query(user_id=session['user_info']['id'], fav_user_id=user_info['id']).first():
data['is_fav'] = True
comment_list = [
i.to_dict()
for i in Comments.filter_by_query(user_id=user_info['id']).order_by(Comments.time_create.desc())[:10]
]
comment_topic_dict = {
i.id: i.to_dict()
for i in Topics.query.filter(
Topics.id.in_(set([i['topic_id'] for i in comment_list])),
Topics.available == 1
)
}
comment_topic_list = [i for i in comment_topic_dict.values()]
user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(i['user_id'] for i in comment_topic_list)),
Users.available == 1
)
}
tab_dict = {
i.id: i.to_dict()
for i in Tabs.query.filter(
Tabs.id.in_(set(i['tab_id'] for i in comment_topic_list)),
Tabs.available == 1
)
}
sub_tab_dict = {
i.id: i.to_dict()
for i in SubTabs.query.filter(
SubTabs.id.in_(set(i['sub_tab_id'] for i in comment_topic_list)),
SubTabs.available == 1
)
}
for topic in comment_topic_list:
comment_topic_dict[topic['id']]['user'] = user_dict[topic['user_id']]
comment_topic_dict[topic['id']]['tab'] = tab_dict[topic['tab_id']]
comment_topic_dict[topic['id']]['sub_tab'] = sub_tab_dict[topic['sub_tab_id']]
for comment in comment_list:
comment['is_thank'] = False
comment['topic'] = comment_topic_dict[comment['topic_id']]
if session.get('user_info'):
comment_thank_id_list = [i.comment_id for i in
CommentThank.filter_by_query(user_id=session['user_info']['id'])]
for comment in comment_list:
if comment['id'] in comment_thank_id_list:
comment['is_thank'] = True
data['comments'] = comment_list
if user_info['privacy_level'] == 1 and not session.get('is_login'):
data['is_open'] = False
data['topics'] = []
return data
if user_info['privacy_level'] == 2:
if not session.get('user_info') or session.get('user_info')['id'] != user_info['id']:
data['is_open'] = False
data['topics'] = []
return data
topic_list = [
i.to_dict()
for i in Topics.filter_by_query(user_id=user_info['id']).order_by(Topics.time_create.desc())[:10]
]
last_reply_user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(i['last_reply_user_id'] for i in topic_list)),
Users.available == 1
)
}
tab_dict = {
i.id: i.to_dict()
for i in Tabs.query.filter(Tabs.id.in_(set(i['tab_id'] for i in topic_list)), Tabs.available == 1)
}
sub_tab_dict = {
i.id: i.to_dict()
for i in SubTabs.query.filter(
SubTabs.id.in_(set(i['sub_tab_id'] for i in topic_list)),
SubTabs.available == 1
)
}
comment_count_dict = dict()
for c in Comments.query.filter(
Comments.topic_id.in_(set(i['id'] for i in topic_list)),
Comments.available == 1
):
if comment_count_dict.get(c.topic_id):
comment_count_dict[c.topic_id] += 1
else:
comment_count_dict[c.topic_id] = 1
up_down_dict_list = defaultdict(list)
for up_down_obj in TopicUpDown.query.filter(
TopicUpDown.topic_id.in_(set(i['id'] for i in topic_list)),
TopicUpDown.available == 1
):
up_down_dict_list[up_down_obj.topic_id].append(up_down_obj.action)
for _t in topic_list:
if _t['last_reply_user_id']:
_t['last_reply_user'] = last_reply_user_dict[_t['last_reply_user_id']]
_t['tab'] = tab_dict[_t['tab_id']]
_t['sub_tab'] = sub_tab_dict[_t['sub_tab_id']]
_t['comment_count'] = comment_count_dict.get(_t['id'], 0)
# 获取up和down数量
if up_down_dict_list.get(_t['id']):
_t['up_count'] = len([i for i in up_down_dict_list[_t['id']] if i is True])
_t['down_count'] = len([i for i in up_down_dict_list[_t['id']] if i is False])
else:
_t['up_count'] = 0
_t['down_count'] = 0
data['topics'] = topic_list
return data
def get(self):
if self.input.uid:
obj = Topics.get_by_uid(self.input.uid)
data = obj.to_dict()
data['user'] = Users.get_by_id(
data['user_id']).to_dict(remove_fields_list=['password'])
if data['last_reply_user_id']:
data['last_reply_user'] = Users.get_by_id(
data['last_reply_user_id']).to_dict(
remove_fields_list=['password'])
data['tab'] = Tabs.get_by_id(data['tab_id']).to_dict()
data['sub_tab'] = SubTabs.get_by_id(data['sub_tab_id']).to_dict()
data['appends'] = [
i.to_dict()
for i in TopicAppends.filter_by_query(topic_id=data['id'])
]
comment_list = [
i.to_dict()
for i in Comments.filter_by_query(topic_id=data['id'])
]
comment_user_dict = {
i.id: i.to_dict()
for i in Users.query.filter(
Users.id.in_(set(
i['user_id']
for i in comment_list)), Users.available == 1)
}
for comment in comment_list:
comment['user'] = comment_user_dict[comment['user_id']]
data['comments'] = comment_list
data['comment_count'] = len(comment_list)
data['up_count'] = TopicUpDown.filter_by_query(
topic_id=data['id'], action=True).count()
data['down_count'] = TopicUpDown.filter_by_query(
topic_id=data['id'], action=False).count()
else:
data = Topics.filter_by_query()
if self.input.tab_id:
data.filter_by_query(tab_id=self.input.tab_id)
elif self.input.sub_tab_id:
data.filter_by_query(sub_tab_id=self.input.sub_tab_id)
if self.input.sk_ and self.input.sv_:
data = data.filter(
getattr(Topics,
self.input.sk_).like(f'%{self.input.sv_}%'))
if self.input.odb_ and self.input.odt_ in ['asc', 'desc']:
data = data.order_by(
getattr(getattr(Topics, self.input.odb_),
self.input.odt_)())
else:
data = data.order_by(Topics.time_create.desc())
total = None
if self.input.page and self.input.page_size:
total = data.count()
data = data.offset(
(int(self.input.page) - 1) *
int(self.input.page_size)).limit(int(self.input.page_size))
data = [_t.to_dict() for _t in data]
user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(
i['user_id'] for i in data)), Users.available == 1)
}
last_reply_user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(i['last_reply_user_id']
for i in data)), Users.available == 1)
}
tab_dict = {
i.id: i.to_dict()
for i in Tabs.query.filter(
Tabs.id.in_(set(
i['tab_id'] for i in data)), Tabs.available == 1)
}
sub_tab_dict = {
i.id: i.to_dict()
for i in SubTabs.query.filter(
SubTabs.id.in_(set(i['sub_tab_id']
for i in data)), SubTabs.available == 1)
}
comment_count_dict = dict()
for c in Comments.query.filter(
Comments.topic_id.in_(set(i['id'] for i in data)),
Comments.available == 1):
if comment_count_dict.get(c.topic_id):
comment_count_dict[c.topic_id] += 1
else:
comment_count_dict[c.topic_id] = 1
up_down_dict_list = defaultdict(list)
for up_down_obj in TopicUpDown.query.filter(
TopicUpDown.topic_id.in_(set(i['id'] for i in data)),
TopicUpDown.available == 1):
up_down_dict_list[up_down_obj.topic_id].append(
up_down_obj.action)
for _d in data:
_d['user'] = user_dict[_d['user_id']]
if _d['last_reply_user_id']:
_d['last_reply_user'] = last_reply_user_dict[
_d['last_reply_user_id']]
_d['tab'] = tab_dict[_d['tab_id']]
_d['sub_tab'] = sub_tab_dict[_d['sub_tab_id']]
_d['comment_count'] = comment_count_dict.get(_d['id'], 0)
# 获取up和down数量
if up_down_dict_list.get(_d['id']):
_d['up_count'] = len(
[i for i in up_down_dict_list[_d['id']] if i is True])
_d['down_count'] = len(
[i for i in up_down_dict_list[_d['id']] if i is False])
else:
_d['up_count'] = 0
_d['down_count'] = 0
if total is not None:
data = {'total': total, 'list': data}
return data
def get(self):
data = dict()
sub_tab_info = SubTabs.get_by_query(name=self.input.tab).to_dict()
sub_tab_info['tab'] = Tabs.get_by_id(sub_tab_info['tab_id']).to_dict()
sub_tab_info['other_tabs'] = [
i.to_dict()
for i in SubTabs.filter_by_query(tab_id=sub_tab_info['tab_id'])
]
data['sub_tab_info'] = sub_tab_info
topic_objs = Topics.filter_by_query(
sub_tab_id=sub_tab_info['id']).order_by(Topics.time_create.desc())
topic_count = topic_objs.count()
if topic_count > 100:
page = int(self.input.page) if self.input.page else 1
page_size = 50
topic_objs = topic_objs.offset(
(page - 1) * page_size).limit(page_size)
topic_list = [_t.to_dict() for _t in topic_objs]
user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(
i['user_id'] for i in topic_list)), Users.available == 1)
}
last_reply_user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(i['last_reply_user_id']
for i in topic_list)), Users.available == 1)
}
comment_count_dict = dict()
comment_objs = Comments.query.filter(
Comments.topic_id.in_(set(i['id'] for i in topic_list)),
Comments.available == 1)
for c in comment_objs:
if comment_count_dict.get(c.topic_id):
comment_count_dict[c.topic_id] += 1
else:
comment_count_dict[c.topic_id] = 1
up_down_dict_list = defaultdict(list)
for up_down_obj in TopicUpDown.query.filter(
TopicUpDown.topic_id.in_(set(i['id'] for i in topic_list)),
TopicUpDown.available == 1):
up_down_dict_list[up_down_obj.topic_id].append(up_down_obj.action)
for _d in topic_list:
_d['user'] = user_dict[_d['user_id']]
if _d['last_reply_user_id']:
_d['last_reply_user'] = last_reply_user_dict[
_d['last_reply_user_id']]
_d['comment_count'] = comment_count_dict.get(_d['id'], 0)
# 获取up和down数量
if up_down_dict_list.get(_d['id']):
_d['up_count'] = len(
[i for i in up_down_dict_list[_d['id']] if i is True])
_d['down_count'] = len(
[i for i in up_down_dict_list[_d['id']] if i is False])
else:
_d['up_count'] = 0
_d['down_count'] = 0
data['total'] = topic_count
data['list'] = topic_list
if session.get('is_login'):
if SubTabFav.filter_by_query(
user_id=session['user_info']['id'],
sub_tab_id=sub_tab_info['id']).first():
data['is_fav'] = True
else:
data['is_fav'] = False
return data
def get(self):
data = dict()
sub_tab_dict_list = defaultdict(list)
for sub_tab in SubTabs.filter_by_query().order_by(
SubTabs.sort_num.asc()):
sub_tab_dict_list[sub_tab.tab_id].append(sub_tab.to_dict())
tab_list = [
t.to_dict()
for t in Tabs.filter_by_query().order_by(Tabs.sort_num.asc())
]
for _d in tab_list:
_d['sub_tabs'] = sub_tab_dict_list.get(_d['id']) or []
data['tabs'] = tab_list
if self.input.tab:
tab_obj = Tabs.filter_by_query(name=self.input.tab).first()
if not tab_obj:
tab_obj = Tabs.get_by_query(name='tech')
else:
tab_obj = Tabs.get_by_query(name='tech')
topic_objs = Topics.filter_by_query(tab_id=tab_obj.id).order_by(
Topics.time_create.desc())
data['tab_topic_count'] = topic_objs.count()
topic_list = [_t.to_dict() for _t in topic_objs[:80]]
user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(
i['user_id'] for i in topic_list)), Users.available == 1)
}
last_reply_user_dict = {
i.id: i.to_dict(remove_fields_list=['password'])
for i in Users.query.filter(
Users.id.in_(set(i['last_reply_user_id']
for i in topic_list)), Users.available == 1)
}
tab_dict = {
i.id: i.to_dict()
for i in Tabs.query.filter(
Tabs.id.in_(set(
i['tab_id'] for i in topic_list)), Tabs.available == 1)
}
sub_tab_dict = {
i.id: i.to_dict()
for i in SubTabs.query.filter(
SubTabs.id.in_(set(
i['sub_tab_id']
for i in topic_list)), SubTabs.available == 1)
}
comment_count_dict = dict()
for c in Comments.query.filter(
Comments.topic_id.in_(set(i['id'] for i in topic_list)),
Comments.available == 1):
if comment_count_dict.get(c.topic_id):
comment_count_dict[c.topic_id] += 1
else:
comment_count_dict[c.topic_id] = 1
up_down_dict_list = defaultdict(list)
for up_down_obj in TopicUpDown.query.filter(
TopicUpDown.topic_id.in_(set(i['id'] for i in topic_list)),
TopicUpDown.available == 1):
up_down_dict_list[up_down_obj.topic_id].append(up_down_obj.action)
for _t in topic_list:
_t['user'] = user_dict[_t['user_id']]
if _t['last_reply_user_id']:
_t['last_reply_user'] = last_reply_user_dict[
_t['last_reply_user_id']]
_t['tab'] = tab_dict[_t['tab_id']]
_t['sub_tab'] = sub_tab_dict[_t['sub_tab_id']]
_t['comment_count'] = comment_count_dict.get(_t['id'], 0)
# 获取up和down数量
if up_down_dict_list.get(_t['id']):
_t['up_count'] = len(
[i for i in up_down_dict_list[_t['id']] if i is True])
_t['down_count'] = len(
[i for i in up_down_dict_list[_t['id']] if i is False])
else:
_t['up_count'] = 0
_t['down_count'] = 0
data['topics'] = topic_list
data['user_count'] = Users.filter_by_query().count()
data['topic_count'] = Topics.filter_by_query().count()
data['comment_count'] = Comments.filter_by_query().count()
today_time_start = self.strptime(
self.get_datetime_now().strftime('%Y-%m-%d'), '%Y-%m-%d')
today_topic_list = [
i.to_dict() for i in Topics.query.filter(
Topics.time_create >= today_time_start, Topics.available == 1)
]
if today_topic_list:
today_comment_dict = dict()
for _c in Comments.query.filter(
Comments.topic_id.in_(
set([i['id'] for i in today_topic_list])),
Comments.available == 1):
if today_comment_dict.get(_c.topic_id):
today_comment_dict[_c.topic_id] += 1
else:
today_comment_dict[_c.topic_id] = 1
for today_topic in today_topic_list:
today_topic['comment_count'] = today_comment_dict.get(
today_topic['id'], 0)
today_hot_topic_list = sorted(today_topic_list,
key=lambda x: x['comment_count'],
reverse=True)
if len(today_hot_topic_list) >= 10:
today_hot_topic_list = today_hot_topic_list[:10]
else:
today_hot_topic_list = []
data['hot_topics'] = today_hot_topic_list
if session.get('is_login'):
fav_sub_tab_id_list = [
i.sub_tab_id for i in SubTabFav.filter_by_query(
user_id=session['user_info']['id'])
]
sub_tab_list = [
i.to_dict() for i in SubTabs.query.filter(
SubTabs.id.in_(set(fav_sub_tab_id_list)), SubTabs.available
== 1)
]
data['fav_tabs'] = sub_tab_list
return data