def saveCid(self):
try:
cid = self.jsonBody['cid']
platform = self.jsonBody['platform']
logger.debug('cid: %s, type: %s', cid, type)
userid = self.user['_id']
logger.debug('userid: %s', userid)
push_cid_obj = ClassHelper('PushCid').find_one({'userid': userid})
print push_cid_obj
push_cid = MeObject('PushCid', obj=push_cid_obj)
# if push_cid is None:
# push_cid = MeObject('PushCid')
# else:
# push_cid = Me
print type(push_cid)
push_cid['userid'] = userid
push_cid['platform'] = platform
push_cid['cid'] = cid
push_cid.save()
self.write(ERR_SUCCESS.message)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def createUserInviteCode(self):
cursor = ClassHelper('User').find({})
if cursor:
i = 0
for u in cursor:
code = ClassHelper('InviteCode').find_one({'from': u['_id'], 'status': 0})
n = 0
if not code:
while (True):
try:
invite_code = MeObject('InviteCode')
invite_code['code'] = InviteCodeUtil.create_code()
invite_code['status'] = 0
invite_code['from'] = u['_id']
invite_code.save()
n = n + 1
logger.debug('create invite code success')
self.notify_new_code(u['_id'])
# logger.debug('create invite code finish %d', i)
if n >= 1:
break
except Exception, e:
logger.error(e)
else:
logger.debug('user already has a code')
i = i + 1
logger.debug('finish %d', i)
logger.debug('all finish')
def saveCid(self):
try:
cid = self.jsonBody['cid']
platform = self.jsonBody['platform']
logger.debug('cid: %s, platform: %s', cid, platform)
userid = self.user['_id']
# userid = '5a016819ca71430a9d0fe108'
logger.debug('userid: %s', userid)
push_cid_obj = ClassHelper('PushCid').find_one({'userid': userid})
print push_cid_obj
push_cid = MeObject('PushCid', obj=push_cid_obj)
print type(push_cid)
push_cid['userid'] = userid
push_cid['platform'] = platform
push_cid['cid'] = cid
push_cid['logout'] = False
push_cid.save()
self.write(ERR_SUCCESS.message)
r = ClassHelper('PushCid').db.update_many('PushCid', {
'cid': cid,
'userid': {
'$ne': userid
},
'logout': False
}, {'$set': {
'logout': True
}})
logger.debug('r:%s', r)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def post(self):
try:
xml = self.request.body
logger.debug('xml: %s', xml)
pub = Wxpay_server_pub()
pub.saveData(xml)
logger.debug('pub.data: %s', pub.data)
if pub.data['return_code'] == 'SUCCESS' and pub.data[
'result_code'] == 'SUCCESS':
print 'wx call back result is success'
order_no = pub.data.get('out_trade_no', None)
logger.debug('out_trade_no:%s', order_no)
if order_no:
flow = ClassHelper("RechargeFlow").find_one({
'orderNo': order_no,
'status': 0
})
if flow:
payUtil.orderCallback(flow['_id'], flow['user'], 1,
pub.data)
else:
order_no = pub.data.get('out_trade_no', None)
if order_no:
flow = ClassHelper("RechargeFlow").find_one({
'orderNo': order_no,
'status': 0
})
if flow:
payUtil.orderCallback(flow['_id'], flow['user'], -1,
pub.data)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
def searchUser(self):
try:
logger.debug(self.jsonBody)
obj = self.jsonBody
content = obj.get('content')
logger.debug('search content:%s', content)
scroll_id = obj.get('scrollId', None)
page_size = obj.get('pageSize', None)
if not page_size:
page_size = 10
if page_size is None or page_size <= 0 or page_size > EsClient.USER_PAGE_SIZE:
page_size = EsClient.USER_PAGE_SIZE
# logger.debug('scroll_id: %s', scroll_id)
# logger.debug('page_size: %d', page_size)
if not content and not scroll_id:
self.write(ERR_PARA.message)
else:
# print content.replace(' ', '')
# es 搜索 (去空格)
searched = EsClient.searchUser(content.replace(' ', ''),
scroll_id,
pageSize=page_size)
users = []
for item in searched['hits']['hits']:
source = item['_source']
del source['pinyin']
del source['pinyinsmall']
users.append(source)
# print users
logger.debug('find num: %d', users.__len__())
result = {}
result['scrollId'] = searched['_scroll_id']
result['users'] = users
if users.__len__() > 0:
followerHelper = ClassHelper('Followee')
for user in result['users']:
r = CountHelper.get_specific_count(user['user'])
logger.debug('r:%s', r)
logger.debug('nickName: %s,user: %s', user['nickName'],
user['user'])
user['fansCount'] = r['followers']
user['followers'] = r['followers']
# TODO 计算来自XX人贡献了xx照片
user['assigners'] = r['assigners']
user['contributeCount'] = r['assigners']
user['imageCount'] = r['medias']
user['medias'] = r['medias']
user['isUser'] = 1
del user['id']
# logger.debug('final result:' + json.dumps(result, ensure_ascii=False))
result['errCode'] = 0 # 执行成功code=0
self.write(result)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def push(self):
try:
userid = self.jsonBody['userid']
otherid = self.jsonBody['otherid']
action = self.jsonBody.get('action')
logger.debug('userid: %s, otherid: %s, action: %s', userid,
otherid, action)
if userid is None or action is None or otherid is None:
self.write(ERR_PARA.message)
return
uri = action_uri_dict.get(action, None)
if uri is None:
self.write(ERR_PARA.message)
return
uri = 'honey://' + uri
logger.debug('uri: %s', uri)
if action == 'claimed':
uri = uri + '/' + otherid
notify_content = self.getNotifyContent(action, otherid)
logger.debug('notify_content: %s', notify_content)
if notify_content is None:
self.write(ERR_PARA.message)
return
# 长链接通知
message_dict = {'t': 'notify'}
message_dict['title1'] = notify_content['title']
# message_dict['title2'] = notify_content['title2']
message_dict['title2'] = notify_content['content']
message_dict['to_id'] = userid
message_dict['uri'] = uri
message_json = json.dumps(message_dict, ensure_ascii=False)
logger.debug(message_json)
rc = RedisDb.get_connection()
publish_result = rc.publish(Constants.REDIS_CHANNEL_FOR_PUSH,
message_json)
logger.debug('public_result: %s', publish_result)
push_cid_obj = ClassHelper('PushCid').find_one({'userid': userid})
logger.debug('push_cid_obj: %s', push_cid_obj)
if push_cid_obj is None:
# 没有找到对应的push_cid
self.write(ERR_PARA.message)
return
push_cid = MeObject('PushCid', obj=push_cid_obj)
result = MyGtUtil.pushMessageToSingle(
push_cid['cid'],
notify_content['title'].decode("utf-8"),
notify_content['content'].decode("utf-8"),
data=uri)
logger.debug('result:%s', result)
# result = PushUtil.pushMessageToSingle()
self.write(result)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def unreadListByFromId(self):
try:
# test msg_id
# ObjectId("59dc380051159a123fe8a230")
# msg_id = self.get_argument('id')
# logger.debug("id: %s", msg_id)
print self.user['_id']
# to_id = "59ca0b46ca714306705996dc"
to_id = self.user['_id']
from_id = self.get_argument('from_id')
m_id = self.get_argument('m_id')
logger.debug('from_id: %s, m_id: %s', from_id, m_id)
# if msg_id:
# query = {"to_id": to_id, "status": 0,
# "_id": {"$lt": ObjectId(msg_id)}}
# else:
# query = {"to_id": to_id, "status": 0}
query = {
"to_id": to_id,
"status": 0,
"from_id": from_id,
"msg_type": 2,
"m_id": m_id
}
print query
helper = ClassHelper('Message')
cursor = helper.find(query)
unread_list = []
if cursor:
for data in cursor:
logger.debug('data:%s', data)
unread_list.append(data)
logger.debug(unread_list)
# print unread_list.__len__()
if unread_list:
for unread_msg in unread_list:
unread_msg['id'] = unread_msg['_id']
unread_msg['create_at'] = long(
unread_msg['createAt'].strftime('%s')) * 1000
del unread_msg['_id']
if unread_msg.get('acl', None) is not None:
del unread_msg['acl']
del unread_msg['createAt']
del unread_msg['updateAt']
result = {}
result['errCode'] = 0
result['unread_list'] = unread_list
r = json.dumps(result, ensure_ascii=False)
self.write(str(r)) # , cls=CJsonEncoder #格式化时间
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def get(self, action=None):
try:
if action == 'flow':
self.flow()
else:
print 'action error ', action
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def post(self, action=None):
try:
logger.debug('action: %s', action)
if action == 'login':
self.login()
else:
print "action error: " + action
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def post(self, action=None):
try:
if action == 'check':
self.check()
elif action == 'activate':
self.activate()
else:
print 'action error ', action
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def get(self, action=None):
try:
if action == 'createInviteCode':
self.createInviteCode()
elif action == 'createUserInviteCode':
self.createUserInviteCode()
elif action == 'owned':
self.owned()
else:
print 'action error ', action
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def post(self, action=None):
try:
if action == 'createWxAppOrder':
self.createWxAppOrder()
elif action == 'createAppleOrder':
self.recharge()
elif action == 'applePayComplete':
self.payComplete()
elif action == 'createAlipayAppOrder':
self.createAlipayAppOrder()
else:
print "action error: " + action
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def createInviteCode(self):
count = int(self.get_argument('count', 1))
if count > 100:
count = 100
i = 0
while (True):
try:
invite_code = MeObject('InviteCode')
invite_code['code'] = InviteCodeUtil.create_code()
invite_code['status'] = 0
invite_code.save()
i = i + 1
logger.debug('create invite code finish %d', i)
if i >= count:
break
except Exception, e:
logger.error(e)
def post(self, action=None):
try:
logger.debug('Cookie: %s', self.request.headers['Cookie'])
cookie = self.request.headers['Cookie'].split('"')[1]
RedisDb.delete(cookie)
self.clear_cookie('u')
push_cid_obj = ClassHelper('PushCid').find_one(
{'userid': self.user['_id']})
if push_cid_obj:
push_cid = MeObject('PushCid', obj=push_cid_obj)
push_cid['logout'] = True
push_cid.save()
self.write(ERR_SUCCESS.message)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_INVALID.message)
def index(self):
try:
userid = self.user['_id']
# userid = '5a1aea98ca714333dda69c5c'
cursor = ClassHelper('StatCount').find({'name': {
'$in': ['unreadMsg_' + userid, 'newFans_' + userid, 'toClaim_' + userid, 'newIncome_' + userid]}})
unread_msg = 0
fans = 0
unclaimed = 0
earnings = 0
if cursor:
for data in cursor:
if data['name'] == 'unreadMsg_' + userid:
unread_msg = data['count']
elif data['name'] == 'newFans_' + userid:
fans = data['count']
elif data['name'] == 'toClaim_' + userid:
unclaimed = data['count']
elif data['name'] == 'newIncome_' + userid:
earnings = data['count']
result = {}
result['code'] = 0
result['errCode'] = 0
result['fans'] = fans
# result['unread_msg'] = unread_msg
result['unread_msg'] = ClassHelper('Message').query_count({'to_id': userid, 'status': 0})
result['earnings'] = earnings
result['unclaimed'] = unclaimed
result['feedCount'] = int(RedisDb.get('user_unread_feed_count_%s' % userid) or 0)
faceHelper = ClassHelper( 'FaceRecommend' )
sid = RedisDb.hget( "recommendSLatestOid", userid)
if sid:
query={'_sid': {"$lte": sid},'user': userid, 'read': {'$exists': False},'backupUser': {'$exists': False}}
result['similarCount'] = faceHelper.query_count(query)
else:
result['similarCount'] = 0
recommendHelper = ClassHelper('UserRecommend')
result['possibleCount'] = recommendHelper.query_count({'user': userid})
self.write(result)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def get(self, action=None):
try:
if action == 'base':
self.base()
# elif action == 'create':
# self.create()
elif action == 'createConfig':
self.createConfig()
# elif action == 'createAlbumConfig':
# self.createAlbumConfig()
# elif action == 'createCommentSession':
# self.createCommentSession()
else:
print 'action error ', action
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def post(self, action=None):
logger.debug('action:%s', action)
try:
if action == 'createWxAppOrder':
self.createWxAppOrder()
elif action == 'createAppleAppOrder':
self.createAppleAppOrder()
elif action == 'completeAppleAppOrder':
self.completeAppleAppOrder()
elif action == 'createAlipayAppOrder':
self.createAlipayAppOrder()
elif action == 'charge': # 消费
self.charge()
else:
print "action error: " + action
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def activate(self):
logger.debug('invite code activate start ')
code = self.jsonBody.get('code')
device = self.jsonBody.get('device')
invite_code = ClassHelper('InviteCode').find_one({'code': code, 'status': 0})
logger.debug('jsonBody:%s', self.jsonBody)
if not invite_code:
logger.debug('invite code error')
result = deepcopy(ERR_PARA.message)
result['errMsg'] = '邀请码错误'
result['msg'] = 'invite code error'
self.write(result)
return
else:
active_device = MeObject('ActiveDevice')
active_device['device'] = device
active_device['code'] = code
try:
active_device.save()
invite_code_obj = MeObject('InviteCode', obj=invite_code)
invite_code_obj['status'] = 1
# invite_code_obj['to'] = self.user['_id']
invite_code_obj.save()
logger.debug('success')
self.write(ERR_SUCCESS.message)
return
except Exception, e:
logger.debug('device already active')
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
result = deepcopy(ERR_DEVICE_ALREADY_ACTIVE.message)
result['errMsg'] = '设备已激活,不能重复激活'
result['msg'] = 'device already active'
self.write(result)
return
def push(self):
try:
userid = self.jsonBody['userid']
otherid = self.jsonBody['otherid']
action = self.jsonBody.get('action')
extra = self.jsonBody.get('extra', '')
logger.debug('third push userid: %s, otherid: %s, action: %s',
userid, otherid, action)
if userid is None or action is None or otherid is None:
self.write(ERR_PARA.message)
return
uri = action_uri_dict.get(action, None)
if uri is None:
self.write(ERR_PARA.message)
return
uri = 'honey://' + uri
logger.debug('uri: %s', uri)
if action == 'claimed' or action == 'newFeed':
uri = uri + '/' + otherid
elif action == 'similarFace' or action == 'intrestedFace':
uri = uri + '?unreadCount=' + extra
notify_content = self.getNotifyContent(action, userid, otherid,
extra)
logger.debug('notify_content: %s', notify_content)
if notify_content is None:
self.write(ERR_PARA.message)
return
# 长链接通知
message_dict = {'t': 'notify'}
message_dict['title'] = notify_content['title']
# message_dict['title2'] = notify_content['title2']
message_dict['subtitle'] = notify_content['content']
message_dict['avatar'] = notify_content['avatar']
message_dict['to_id'] = userid
message_dict['uri'] = uri
message_json = json.dumps(message_dict, ensure_ascii=False)
logger.debug('publish_message:%s', message_json)
rc = RedisDb.get_connection()
publish_result = rc.publish(Constants.REDIS_CHANNEL_FOR_PUSH,
message_json)
logger.debug('publish_result: %s', publish_result)
push_cid_obj = ClassHelper('PushCid').find_one({'userid': userid})
logger.debug('push_cid_obj: %s', push_cid_obj)
if (push_cid_obj is None) or push_cid_obj['logout'] is True:
# 没有找到对应的push_cid
self.write(ERR_PARA.message)
return
claim_count = 0
message_count = 0
stat1 = ClassHelper('StatCount').find_one(
{'name': 'toClaim_' + userid})
# logger.debug('stat1:')
if stat1:
claim_count = stat1['count']
if claim_count < 0:
claim_count = 0
stat2 = ClassHelper('StatCount').find_one(
{'name': 'unreadMsg_' + userid})
if stat2:
message_count = stat2['count']
if message_count < 0:
message_count = 0
badge = claim_count + message_count
push_cid = MeObject('PushCid', obj=push_cid_obj)
result = MyGtUtil.pushMessageToSingle(
push_cid['cid'], notify_content['title'].decode("utf-8"),
notify_content['content'].decode("utf-8"), uri, badge)
logger.debug('result:%s', result)
# result = PushUtil.pushMessageToSingle()
self.write(result)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def goodsInfo(self):
try:
userid = self.user['_id']
# userid = '5a08fbfbca714330cfd35ddc'
message_id = self.get_argument('message_id', None)
message = ClassHelper('Message').get(message_id)
if not message:
self.write(ERR_PARA.message)
return
logger.debug('message:%s', message)
file_id = message['m_id']
media = ClassHelper('Media').find_one({'file': file_id})
logger.debug('media:%s', media)
face = None
if media:
face = self.getFace(media, message['from_id'],
message['to_id'])
logger.debug('face1 %s:', face)
if not face:
face['goodsId'] = None
face['mosaic'] = 2
face['price'] = 0
logger.debug('未找到face')
else:
face['goodsId'] = None
face['mosaic'] = 10
face['price'] = 0
if face.get('assign', None):
owner_id = face['assign']['user']
good = ClassHelper('Goods').find_one(
{'goods': face['_id']})
if good:
face['user'] = face['assign']['user']
face['goodsId'] = good['_id']
price = good['price'] if good else 0
face['price'] = price
if media.get('uploader',
None) and media['uploader'] == userid:
# 照片贡献者
face['mosaic'] = 0
elif owner_id == userid:
# 照片认领者
face['mosaic'] = 5
if face['mosaic'] != 0:
charge_record = ClassHelper(
'ChargeFlow').find_one({
'user': userid,
'goods': good['_id'],
'status': 1
}) if good else None
logger.debug('charge_record%s', charge_record)
if charge_record:
face['mosaic'] = 1
else:
logger.warn('goods is null')
# self.write(ERR_PARA.message)
# return
userInfo = {}
if media:
logger.debug('uploader:%s', media.get('uploader', None))
user_info_obj = ClassHelper('UserInfo').find_one(
{'user': media.get('uploader', None)})
logger.debug('user_info_obj:%s', user_info_obj)
if user_info_obj:
userInfo['_id'] = user_info_obj['_id']
userInfo['avatar'] = user_info_obj['avatar']
userInfo['nickName'] = user_info_obj['nickName']
logger.debug('userInfo:%s', userInfo)
if media and face:
r = {}
r['errCode'] = 0
r['width'] = media.get('width', 0)
r['height'] = media.get('height', 0)
r['media'] = media['_id']
r['file'] = media['file']
r['goodsId'] = face['goodsId']
r['mosaic'] = face['mosaic']
r['price'] = face['price']
r['_id'] = face['_id']
r['user'] = face.get('user', None)
r['userInfo'] = userInfo
if face.get('rect', None) is not None:
r['rect'] = face.get('rect')
r['message_id'] = message_id
self.write(r)
else:
r = {}
r['errCode'] = 0
r['width'] = 0
r['height'] = 0
r['media'] = None
r['file'] = None
r['goodsId'] = None
r['mosaic'] = 0
r['price'] = 0
r['_id'] = None
r['user'] = None
r['rect'] = None
r['userInfo'] = None
r['message_id'] = message_id
self.write(r)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def send(self):
try:
logger.debug(self.jsonBody)
obj = self.jsonBody
logger.debug('to_id:' + obj.get('to_id'))
logger.debug('c: %s', obj.get('c'))
logger.debug('c_type: %s', obj.get('c_type'))
logger.debug('m_id: %s', obj.get('m_id'))
media_id = obj.get('m_id')
from_id = self.user['_id']
logger.debug('from_id: %s', from_id)
user_query = MeQuery("UserInfo")
user_info = user_query.find_one({'user': from_id})
message_dict = {
'from_id': from_id,
'c': obj.get('c'),
'to_id': obj.get('to_id'),
'c_type': obj.get('c_type'),
'msg_type': 2,
'from_avatar': user_info['avatar'],
'from_name': user_info['nickName'],
'status': 0,
'm_id': media_id
}
message = MeObject('Message', obj=message_dict)
message['session'] = SessionUtil.create(from_id, obj.get('to_id'))
message.save()
# 格式化时间为long型
message_dict['create_at'] = long(
message['createAt'].strftime('%s')) * 1000
message_dict['t'] = 'comment'
message_dict['id'] = message['_id']
message_json = json.dumps(message_dict, ensure_ascii=False)
logger.debug(message_json)
print type(message['createAt'])
rc = RedisDb.get_connection()
rc.publish(Constants.REDIS_CHANNEL_FOR_PUSH, message_json)
# 发push
push_cid_obj = ClassHelper('PushCid').find_one(
{'userid': obj.get('to_id')})
logger.debug('push_cid_obj: %s', push_cid_obj)
if push_cid_obj and (push_cid_obj['logout'] is False):
# push_cid = MeObject('PushCid', obj=push_cid_obj)
title = user_info['nickName']
content = obj.get('c')
content = PushEmoji.getPushContent(content)
data = 'honey://comment/' + from_id + '?m_id=' + media_id
print title.encode('utf-8')
print content.encode('utf-8')
claim_count = 0
message_count = 0
stat1 = ClassHelper('StatCount').find_one(
{'name': 'toClaim_' + obj.get('to_id')})
if stat1:
claim_count = stat1['count']
if claim_count < 0:
claim_count = 0
stat2 = ClassHelper('StatCount').find_one(
{'name': 'unreadMsg_' + obj.get('to_id')})
if stat2:
message_count = stat2['count']
if message_count < 0:
message_count = 0
badge = claim_count + message_count
t = threading.Thread(target=MyGtUtil.pushMessageToSingle,
args=(
push_cid_obj['cid'],
title.encode('utf-8'),
content.encode('utf-8'),
data,
badge,
))
t.setDaemon(True)
t.start()
# logger.debug(ERR_SUCCESS.message)
r = {}
r['id'] = message['_id']
r['code'] = 0
r['errCode'] = 0
r['create_at'] = message_dict['create_at']
self.write(r)
# 计数 unreadMsg +1
logger.debug('update to_id:unreadMsg_%s unreadMsg count ',
obj.get('to_id'))
ClassHelper('StatCount').updateOne(
{'name': 'unreadMsg_' + obj.get('to_id')},
{"$inc": {
'count': 1
}},
upsert=True)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def listByFromId(self):
try:
# test msg_id
# ObjectId("59dc380051159a123fe8a230")
message_id = self.get_argument('message_id', default=None)
count = int(self.get_argument('count', default=100))
logger.debug("count: %s", count)
if count > 100:
count = 100
logger.debug("message_id: %s, count: %s", message_id, count)
# print self.user['_id']
to_id = self.user['_id']
# to_id = '5a018adeca714319e603ca09'
# to_id = '5a018adeca714319e603ca09'
# to_id = '5a0188ccca714319e603c9e8'
from_id = self.get_argument('from_id')
m_id = self.get_argument('m_id')
logger.debug('from_id: %s, m_id: %s', from_id, m_id)
helper = ClassHelper('Message')
unread_list = []
if not message_id:
query = {
'from_id': from_id,
'to_id': to_id,
'm_id': m_id,
'msg_type': 2,
'status': 0
}
logger.debug('unread query:%s', query)
cursor = helper.find(query, limit=count, sort={"_id": -1})
logger.debug('cursor when not message_id:%s', cursor)
if cursor:
for data in cursor:
logger.debug('data:%s', data)
unread_list.append(data)
logger.debug('real unread_list:%s', unread_list)
if not unread_list:
if message_id:
query = {
"$or": [{
"to_id": to_id,
"from_id": from_id
}, {
"to_id": from_id,
"from_id": to_id
}],
"msg_type":
2,
"m_id":
m_id,
"_id": {
"$lt": ObjectId(message_id)
}
}
else:
query = {
"$or": [{
"to_id": to_id,
"from_id": from_id
}, {
"to_id": from_id,
"from_id": to_id
}],
"msg_type":
2,
"m_id":
m_id
}
logger.debug('query: %s', query)
cursor = helper.find(query, limit=count, sort={"_id": -1})
if cursor:
for data in cursor:
logger.debug('data:%s', data)
unread_list.append(data)
logger.debug('unread_list:%s', unread_list)
if unread_list:
unread_list.reverse()
for unread_msg in unread_list:
unread_msg['id'] = unread_msg['_id']
unread_msg['create_at'] = long(
unread_msg['createAt'].strftime('%s')) * 1000
del unread_msg['_id']
if unread_msg.get('acl', None) is not None:
del unread_msg['acl']
del unread_msg['createAt']
del unread_msg['updateAt']
if unread_msg['from_id'] == to_id:
unread_msg['status'] = 1
result = {}
result['errCode'] = 0
result['unread_list'] = unread_list
r = json.dumps(result, ensure_ascii=False)
self.write(str(r)) # , cls=CJsonEncoder #格式化时间
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def unreadList(self):
try:
logger.debug('self.userid: %s:', self.user['_id'])
all = int(self.get_argument('all', 0))
logger.debug('all: %s', all)
if all == 1:
match = {"$match": {"to_id": self.user['_id'], "msg_type": 2}}
else:
match = {
"$match": {
"to_id": self.user['_id'],
"status": 0,
"msg_type": 2
}
}
# "59ca0b46ca714306705996dc"
message_query = MeQuery("Message")
unread_list = message_query.aggregate([
match, {
"$group": {
"_id": {
"from_id": "$from_id",
"m_id": "$m_id"
},
"count": {
"$sum": 1
},
"c": {
"$last": "$c"
},
"id": {
"$last": "$_id"
},
"create_at": {
"$last": "$createAt"
},
"c_type": {
"$last": "$c_type"
},
"status": {
"$last": "$status"
},
"from_name": {
"$last": "$from_name"
},
"msg_type": {
"$last": "$msg_type"
},
"from_id": {
"$last": "$from_id"
},
"to_id": {
"$last": "$t"
"o_id"
},
"from_avatar": {
"$last": "$from_avatar"
},
"m_id": {
"$last": "$m_id"
}
}
}, {
"$sort": {
"create_at": -1
}
}
]) # 时间倒序
logger.debug(unread_list)
# 整理数据
if unread_list:
for unread_msg in unread_list:
del unread_msg['_id']
# print type(unread_msg['id'])
# print str(unread_msg['id'])
# 转换objectId to string
unread_msg['id'] = str(unread_msg['id'])
# print type(unread_msg['create_at'])
unread_msg['create_at'] = long(
unread_msg['create_at'].strftime('%s')) * 1000
result = {}
result['errCode'] = 0
result['unread_list'] = unread_list
self.write(json.dumps(
result, ensure_ascii=False)) # , cls=CJsonEncoder #格式化时间
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def unreadList(self):
try:
logger.debug('self.userid: %s:', self.user['_id'])
all = int(self.get_argument('all', 0))
logger.debug('all: %s', all)
userid = self.user['_id']
logger.debug('userid:%s', userid)
# userid = '5a0188ccca714319e603c9e8'
# userid = '5a0d4d88ca71432378862c53'
if all == 1:
# match = {"$match": {"to_id": self.user['_id'],
# "msg_type": 2}}
# {"$or": [{"to_id": to_id, "from_id": from_id}, {"to_id": from_id, "from_id": to_id}]
match = {
"$match": {
"$or": [{
"to_id": userid
}, {
"from_id": userid
}],
"msg_type": 2
}
}
else:
match = {
"$match": {
"to_id": userid,
"status": 0,
"msg_type": 2
}
}
# "59ca0b46ca714306705996dc"
logger.debug('match:%s', match)
message_query = MeQuery("Message")
unread_list = message_query.aggregate([
match,
{
"$group": {
"_id": {
"from_id": "$session",
"m_id": "$m_id"
},
# "count": {"$sum": 1},
"c": {
"$last": "$c"
},
"id": {
"$last": "$_id"
},
"create_at": {
"$last": "$createAt"
},
"c_type": {
"$last": "$c_type"
},
"status": {
"$last": "$status"
},
"from_name": {
"$last": "$from_name"
},
"msg_type": {
"$last": "$msg_type"
},
"from_id": {
"$last": "$from_id"
},
"to_id": {
"$last": "$t"
"o_id"
},
"from_avatar": {
"$last": "$from_avatar"
},
"m_id": {
"$last": "$m_id"
}
}
},
{
"$sort": {
"create_at": -1
}
}
]) # 时间倒序
count_match = {
"$match": {
"to_id": userid,
"status": 0,
"msg_type": 2
}
}
count_list = message_query.aggregate([
count_match, {
"$group": {
"_id": {
"from_id": "$session",
"m_id": "$m_id"
},
"count": {
"$sum": 1
},
"from_id": {
"$last": "$from_id"
},
"m_id": {
"$last": "$m_id"
}
}
}
])
logger.debug('unread_list:%s', unread_list)
logger.debug('count_list:%s', count_list)
# 处理count_list
count_dict = {}
if count_list:
for count_data in count_list:
otherid = count_data['from_id']
mid = count_data['m_id']
count_dict[otherid + '_' + mid] = count_data['count']
logger.debug('count_dict:%s', count_dict)
# 整理数据
# other_userinfo = None
if unread_list:
for unread_msg in unread_list:
del unread_msg['_id']
# print type(unread_msg['id'])
# print str(unread_msg['id'])
# 转换objectId to string
unread_msg['id'] = str(unread_msg['id'])
# print type(unread_msg['create_at'])
unread_msg['create_at'] = long(
unread_msg['create_at'].strftime('%s')) * 1000
if unread_msg['from_id'] == userid:
otherid = unread_msg['to_id']
# if not other_userinfo:
other_userinfo = ClassHelper('UserInfo').find_one(
{'user': unread_msg['to_id']})
logger.debug('other_userinfo:%s', other_userinfo)
unread_msg['other_name'] = other_userinfo['nickName']
unread_msg['other_avatar'] = other_userinfo.get(
'avatar', None)
else:
otherid = unread_msg['from_id']
unread_msg['other_name'] = unread_msg['from_name']
unread_msg['other_avatar'] = unread_msg['from_avatar']
mid = unread_msg['m_id']
unread_msg['count'] = count_dict.get(
otherid + '_' + mid, 0)
logger.debug('unread_list after make data :%s', unread_list)
result = {}
result['errCode'] = 0
result['unread_list'] = unread_list
r = json.dumps(result, ensure_ascii=False)
self.write(str(r)) # , cls=CJsonEncoder #格式化时间
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def listByFromId(self):
try:
# test msg_id
# ObjectId("59dc380051159a123fe8a230")
message_id = self.get_argument('message_id', default=None)
count = int(self.get_argument('count', default=20))
logger.debug("count: %s", count)
if count > 50:
count = 50
logger.debug("message_id: %s, count: %s", message_id, count)
# print self.user['_id']
# to_id = "59ca0b46ca714306705996dc"
to_id = self.user['_id']
from_id = self.get_argument('from_id')
m_id = self.get_argument('m_id')
logger.debug('from_id: %s, m_id: %s', from_id, m_id)
if message_id:
# query = {"to_id": to_id, "status": 0,
# "_id": {"$lt": ObjectId(msg_id)}}
query = {
"$or": [{
"to_id": to_id,
"from_id": from_id
}, {
"to_id": from_id,
"from_id": to_id
}],
"status":
0,
"msg_type":
2,
"m_id":
m_id,
"_id": {
"$lt": ObjectId(message_id)
}
}
else:
query = {
"$or": [{
"to_id": to_id,
"from_id": from_id
}, {
"to_id": from_id,
"from_id": to_id
}],
"status":
0,
"msg_type":
2,
"m_id":
m_id
}
logger.debug('query: %s', query)
helper = ClassHelper('Message')
unread_list = helper.find(query, limit=count)
logger.debug(unread_list)
print unread_list.__len__()
if unread_list:
for unread_msg in unread_list:
unread_msg['id'] = unread_msg['_id']
unread_msg['create_at'] = long(
unread_msg['createAt'].strftime('%s')) * 1000
del unread_msg['_id']
if unread_msg.get('acl', None) is not None:
del unread_msg['acl']
del unread_msg['createAt']
del unread_msg['updateAt']
result = {}
result['errCode'] = 0
result['unread_list'] = unread_list
self.write(json.dumps(
result, ensure_ascii=False)) # , cls=CJsonEncoder #格式化时间
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def send(self):
try:
logger.debug(self.jsonBody)
obj = self.jsonBody
logger.debug('to_id:' + obj.get('to_id'))
logger.debug('c: %s', obj.get('c'))
logger.debug('c_type: %s', obj.get('c_type'))
logger.debug('m_id: %s', obj.get('m_id'))
media_id = obj.get('m_id')
from_id = self.user['_id']
logger.debug('from_id: %s', from_id)
user_query = MeQuery("UserInfo")
user_info = user_query.find_one({'user': from_id})
message_dict = {
'from_id': from_id,
'c': obj.get('c'),
'to_id': obj.get('to_id'),
'c_type': obj.get('c_type'),
'msg_type': 2,
'from_avatar': user_info['avatar'],
'from_name': user_info['nickName'],
'status': 0,
'm_id': media_id
}
message = MeObject('Message', obj=message_dict)
message.save()
# 格式化时间为long型
message_dict['create_at'] = long(
message['createAt'].strftime('%s')) * 1000
message_dict['t'] = 'comment'
message_dict['id'] = message['_id']
message_json = json.dumps(message_dict, ensure_ascii=False)
logger.debug(message_json)
print type(message['createAt'])
rc = RedisDb.get_connection()
rc.publish(Constants.REDIS_CHANNEL_FOR_PUSH, message_json)
# 发push
push_cid_obj = ClassHelper('PushCid').find_one(
{'userid': obj.get('to_id')})
logger.debug('push_cid_obj: %s', push_cid_obj)
if push_cid_obj:
# push_cid = MeObject('PushCid', obj=push_cid_obj)
title = user_info['nickName']
content = '新评论'
data = 'honey://comment/' + from_id + '?m_id=' + media_id
print title.encode('utf-8')
print content.encode('utf-8')
t = threading.Thread(target=MyGtUtil.pushMessageToSingle,
args=(
push_cid_obj['cid'],
title.encode('utf-8'),
content.encode('utf-8'),
data,
))
t.setDaemon(True)
t.start()
# logger.debug(ERR_SUCCESS.message)
r = {}
r['id'] = message['_id']
r['code'] = 0
r['create_at'] = message_dict['create_at']
self.write(r)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def unreadList(self):
try:
# userid = '59dc58a9ca7143258bcf6bc7'
userid = self.user['_id']
logger.debug('to_id:%s', self.user['_id'])
# "59ca0b46ca714306705996dc"
message_query = MeQuery("Message")
unread_list = message_query.aggregate([{
"$match": {
"to_id": userid,
"status": 0,
"msg_type": 0
}
}, {
"$group": {
"_id": "$from_id",
"count": {
"$sum": 1
},
"c": {
"$last": "$c"
},
"id": {
"$last": "$_id"
},
"create_at": {
"$last": "$createAt"
},
"c_type": {
"$last": "$c_type"
},
"status": {
"$last": "$status"
},
"from_name": {
"$last": "$from_name"
},
"msg_type": {
"$last": "$msg_type"
},
"from_id": {
"$last": "$from_id"
},
"to_id": {
"$last": "$to_id"
},
"from_avatar": {
"$last": "$from_avatar"
}
}
}, {
"$sort": {
"create_at": -1
}
}]) # 时间倒序
logger.debug(unread_list)
# 整理数据
if unread_list:
for unread_msg in unread_list:
del unread_msg['_id']
# print type(unread_msg['id'])
# print str(unread_msg['id'])
# 转换objectId to string
unread_msg['id'] = str(unread_msg['id'])
# print type(unread_msg['create_at'])
unread_msg['create_at'] = long(
unread_msg['create_at'].strftime('%s')) * 1000
unread_msg['other_name'] = unread_msg['from_name']
unread_msg['other_avatar'] = unread_msg['from_avatar']
result = {}
result['errCode'] = 0
result['unread_list'] = unread_list
r = json.dumps(result, ensure_ascii=False)
logger.debug('r:%s', r)
logger.debug(type(r))
self.write(str(r)) # , cls=CJsonEncoder #格式化时间
# 计数 unreadMsg 清空
# logger.debug('reset to_id:unreadMsg_%s unreadMsg count to 0', userid)
# ClassHelper('StatCount').updateOne({'name': 'unreadMsg_' + userid}, {"$set": {'count': 0}},
# upsert=True)
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
self.write(ERR_PARA.message)
def login(self):
mid = self.jsonBody['mid']
device = self.jsonBody['device']
logger.debug('mid:%s', mid)
userMigrateHelper = ClassHelper('UserMigrate')
user_migrate = userMigrateHelper.find_one({'_sid': mid, 'isused': 0})
logger.debug('user_migrate:%s', user_migrate)
if not user_migrate:
self.write(ERR_QRCODE_TO_LOGIN.message)
return
else:
logger.debug('user_migrate.createAt:%s', user_migrate['createAt'])
user = ClassHelper('User').get(user_migrate['user'])
# user = MeUser(obj=user_obj)
# user = MeQuery('User').get(user_migrate['user'])
if not user:
self.write(ERR_USER_NOTFOUND.message)
return
else:
# 长链接通知
notify_dict = {'to_id': user['_sid'], 't': 'logout'}
message_json = json.dumps(notify_dict, ensure_ascii=False)
logger.debug(message_json)
# message_json. = long(message['createAt'].strftime('%s')) * 1000
rc = RedisDb.get_connection()
r = rc.publish(Constants.REDIS_CHANNEL_FOR_PUSH, message_json)
logger.debug('redis publish r:%s', r)
# del user['password']
# del user['username']
user['device'] = device
user_info = ClassHelper('UserInfo').find_one(
{'user': user['_sid']})
del user['createAt']
del user['updateAt']
del user['password']
del user_info['createAt']
del user_info['updateAt']
result = {}
result['user'] = user
result['user_info'] = user_info
result['errCode'] = 0
result['info'] = 'success'
logger.debug('result: %s', json.dumps(result))
self.write(json.dumps(result))
self.set_secure_cookie("u", user['_id'])
new_cookie = str(self._new_cookie.get('u')).split(
'=')[1].split(';')[0].replace('"', '')
print new_cookie
print user['_id']
if new_cookie is not None:
self.set_user_cookie_record(new_cookie, user['_id'])
try:
userMigrateHelper.update(mid, {'$set': {'isused': 1}})
ClassHelper('User').update(
user['_sid'],
{'$set': {
'username': device,
'device': device
}})
self.save_login_log(user['_sid'])
except Exception, e:
logger.error(e)
msg = traceback.format_exc()
logger.error(msg)
