async def after_insert(self, raw_post: Dict, values: SQLValuesToWrite,
record: DataRecord):
post_stats_do_comment(record['related_type'], record['related_id'],
record['id'])
post_number = Comment.select().where(
Comment.related_id == record['related_id'],
Comment.id <= record['id']).count()
Comment.update(post_number=post_number).where(
Comment.id == record['id']).execute()
if self.do_mentions:
# 创建提醒
loc = [record['related_type'], record['related_id']]
# record['related_id']: memoryview
loc_title = POST_TYPES.get_post_title_by_list(loc)[
record['related_id'].tobytes()]
related = [POST_TYPES.COMMENT, record['id']]
self.do_mentions(record['user_id'], loc_title, loc, related)
async def after_insert(self, values_lst: List[SQLValuesToWrite],
records: List[DataRecord]):
for record in records:
post_stats_do_comment(record['related_type'], record['related_id'],
record['id'])
post_number = Comment.select().where(
Comment.related_id == record['related_id'],
Comment.id <= record['id']).count()
Comment.update(post_number=post_number).where(
Comment.id == record['id']).execute()
if self.do_mentions:
# 创建提醒
loc = [record['related_type'], record['related_id']]
# record['related_id']: memoryview
loc_title = POST_TYPES.get_post_title_by_list(loc)[
get_bytes_from_blob(record['related_id'])]
related = [POST_TYPES.COMMENT, record['id']]
self.do_mentions(record['user_id'], loc_title, loc, related)
if config.SEARCH_ENABLE:
run_in_thread(esdb.es_update_comment, record['id'])
def fetch_notif_of_log(user_id, last_manage_log_id=b'\x00'):
from model.manage_log import ManageLog, MOP
if last_manage_log_id is None:
last_manage_log_id = b'\x00'
item_lst = ManageLog.select().where(
ManageLog.related_user_id == user_id,
ManageLog.id > last_manage_log_id,
ManageLog.operation.in_(
(MOP.POST_STATE_CHANGE, MOP.USER_PASSWORD_CHANGE,
MOP.USER_PASSWORD_RESET, MOP.USER_KEY_RESET, MOP.USER_GROUP_CHANGE, MOP.USER_CREDIT_CHANGE,
MOP.USER_REPUTE_CHANGE, MOP.USER_NICKNAME_CHANGE,
MOP.TOPIC_BOARD_MOVE, MOP.TOPIC_AWESOME_CHANGE, MOP.TOPIC_STICKY_WEIGHT_CHANGE)
)
).order_by(ManageLog.id.desc())
moves = []
def wrap(item: ManageLog):
# 总不能把MANAGE_OPERATION的内容换个号码，抄一遍写在上面。
# 因此选择过滤掉一些，其他全部归为一类。
if item.operation == MOP.USER_CREDIT_CHANGE:
if item.note == '每日签到':
# 签到得分忽略
return
elif item.operation == MOP.USER_EXP_CHANGE:
if item.note == '每日登录':
# 签到得分忽略
return
if item.operation == MOP.TOPIC_BOARD_MOVE:
moves.append([POST_TYPES.BOARD, to_bin(item.value['change'][0])])
moves.append([POST_TYPES.BOARD, to_bin(item.value['change'][1])])
return {
'type': NOTIF_TYPE.MANAGE_INFO_ABOUT_ME,
'time': item.time,
'loc_post_type': item.related_type,
'loc_post_id': item.related_id,
'loc_post_title': None,
'sender_ids': (item.user_id,),
'receiver_id': user_id,
'from_post_type': None, # 来源为managelog，但并非post
'from_post_id': item.id,
'related_type': item.related_type, # 提醒类型较为特殊
'related_id': item.related_id,
'data': {
'op': item.operation,
'role': item.role,
'value': item.value,
'title': None # 进行二次填充
}
}
ret_items = list(filter(lambda x: x, map(wrap, item_lst)))
info = POST_TYPES.get_post_title_by_list(*[[i['related_type'], i['related_id']] for i in ret_items])
info2 = POST_TYPES.get_post_title_by_list(*moves)
for i in ret_items:
t = info.get(get_bytes_from_blob(i['related_id']), None)
if t: i['data']['title'] = t
if i['related_type'] == POST_TYPES.COMMENT:
# 这里不是批量，可能要付出较大代价
c: Comment = Comment.get_by_id(i['related_id'])
if not c: continue
p = POST_TYPES.get_post(c.related_type, c.related_id)
if not p: continue
i['loc_post_type'] = c.related_type
i['loc_post_id'] = c.related_id
i['loc_post_title'] = p.get_title()
i['data']['comment'] = {
'related_type': c.related_type,
'related_id': c.related_id,
'related_title': p.get_title(),
'post_number': c.post_number,
'content': c.content
}
if i['data']['op'] == MOP.TOPIC_BOARD_MOVE:
val = i['data']['value']['change']
i['data']['move_info'] = [
info2.get(to_bin(val[0]), None),
info2.get(to_bin(val[1]), None)
]
return ret_items
