def get_battle_unread(battle_id): if rs.exists("battle:" + str(battle_id)) != 1: return -1 uid = loggedUserUid() if rs.hget("battle:%s" % battle_id, "uid") != str(uid): return -2 if rs.exists("battle:%s:unread" % battle_id) != 1: return [] offset = int(request.args.get("offset", 0)) count = int(request.args.get("count", 10)) rows = [] lua = """local r1 = redis.call('ZREVRANGE', KEYS[1], KEYS[2], KEYS[3]); for i = 1, table.getn(r1) do local chid = r1[i] r1[i] = {} r1[i][1] = redis.call('hget', chid, 'id') r1[i][2] = redis.call('hget', chid, 'text') r1[i][3] = redis.call('zscore', KEYS[1], chid) local uid = redis.call('hget', chid, 'rid') r1[i][4] = redis.call('get', 'users:' .. tostring(uid)) r1[i][5] = redis.call('sismember', 'users_online', uid) end return r1;""" ids = rs.eval(lua, 3, "battle:%s:unread" % battle_id, offset, offset + count - 1) for cmid in ids: cmnt = { 'id': int(cmid[0]), 'text': cmid[1], 'create_date': int(cmid[2]) } cmnt['user'] = json.loads(cmid[3]) cmnt['user']['is_online'] = cmid[4] rows.append(cmnt) return rows
def read_message(sid, mid): uid = loggedUserUid() if uid == 0: return -2 chid = "chat:message:%s:%s:%s" % (sid, uid, mid) if rs.exists(chid) != 1: return -1 if rs.hget(chid, "is_read") == "true": return 0 rs.hset(chid, "is_read", 'true') rs.zrem("chat:user:%s:unread" % uid, chid) battle_id = rs.hget(chid, "battle_id") if battle_id is not None: rs.zrem("battle:%s:unread" % battle_id, chid) rs.delete(chid) unread = get_unread._original() ntfs = rs.scard("chat:user:%s:ntfy" % uid) message = {"type": "unread", "content": {"count": len(unread)}} if ntfs > 0 and len(unread) > 0: message["content"]["message"] = unread[0] send_message_to_user(uid, json.dumps(message)) send_message_to_user( sid, json.dumps({ "type": "read_chat", "content": { "mid": mid, "rid": int(uid) } })) return 1
def getUserFollowing(user_id): if rs.exists("users:" + str(user_id)) != 1: return [] offset = request.args.get("offset", 0) count = request.args.get("count", 20) rows = rs.sort('user:'******':following', start=offset, num=count, desc=True, get='users:*') lua = """local r1 = redis.call('sort', 'user:'******':following', 'LIMIT', KEYS[3], KEYS[4], 'GET', 'users:*', 'GET', '#', 'GET', '#', 'GET', '#'); for i = 1, table.getn(r1) do if i % 4 == 1 then r1[i+1] = redis.call('sismember', 'user:'******':followers', KEYS[2]) r1[i+2] = redis.call('sismember', 'users:virtual', r1[i+2]) r1[i+3] = redis.call('sismember', 'users_online', r1[i+3]) end end return r1;""" rows = rs.eval(lua, 4, user_id, loggedUserUid(), offset, count) users = [] for i in range(0, len(rows) - 1): if i % 4 != 0 or rows[i] is None: continue u = json.loads(rows[i]) u['is_follow'] = rows[i + 1] u['is_online'] = rows[i + 3] u['virtual'] = rows[i + 2] users.append(u) return users
def get_clan_users(clan_id): if rs.exists("clan:%s" % clan_id) != 1: return [] offset = request.args.get("offset", 0) count = request.args.get("count", 20) lua = """local r1 = redis.call('sort', 'clan:'..tostring(KEYS[1])..':users', 'DESC', 'LIMIT', KEYS[3], KEYS[4], 'GET', 'users:*', 'GET', '#', 'GET', '#', 'GET', '#'); for i = 1, table.getn(r1) do if i % 4 == 1 then r1[i+1] = redis.call('sismember', 'user:'******':followers', KEYS[2]) r1[i+2] = redis.call('sismember', 'users:virtual', r1[i+2]) r1[i+3] = redis.call('sismember', 'users_online', r1[i+3]) end end return r1;""" rows = rs.eval(lua, 4, clan_id, loggedUserUid(), offset, count) users = [] for i in range(0, len(rows) - 1): print i, rows[i] if i % 4 != 0 or rows[i] is None: continue u = json.loads(rows[i]) u['is_follow'] = rows[i + 1] u['is_online'] = rows[i + 3] u['virtual'] = rows[i + 2] users.append(u) return users
def group_follow(group_id): if rs.exists("group:%s" % group_id) != 1: return -1 uid = loggedUserUid() if uid == 0: return -2 return group_add_user(group_id, uid)
def followUser(user_id): if rs.exists("users:" + str(user_id)) != 1: return -1 uid = loggedUserUid() if uid == 0: return -2 return followUserByUser(user_id, str(uid))
def get_multi_chat_history(dialog_key): if rs.exists(dialog_key) != 1: return [] offset = int(request.args.get("offset", 0)) count = int(request.args.get("count", 10)) rows = [] lua = """local r1 = redis.call('ZREVRANGE', KEYS[1], KEYS[2], KEYS[3]); for i = 1, table.getn(r1) do local chid = r1[i] r1[i] = {} r1[i][1] = redis.call('hget', chid, 'id') r1[i][2] = redis.call('hget', chid, 'text') r1[i][3] = redis.call('zscore', KEYS[1], chid) r1[i][4] = redis.call('hget', chid, 'sid') r1[i][5] = redis.call('get', 'users:'..r1[i][4]) end return r1;""" ids = rs.eval(lua, 3, dialog_key, offset, offset + count - 1) for cmid in ids: cmnt = { 'id': int(cmid[0]), 'text': cmid[1], 'create_date': int(cmid[2]), 'sid': int(cmid[3]), 'user': json.loads(cmid[4]) } rows.append(cmnt) return rows
def delete_battle(battle_id, admin=0): if rs.exists("battle:%s" % battle_id) == 1 and (admin == 1 or str( loggedUserUid()) == rs.hget("battle:%s" % battle_id, 'uid')): uid = rs.hget("battle:%s" % battle_id, 'uid') rs.zrem("user_battles:" + uid, battle_id) rs.zrem("battles_ids", battle_id) battle = json.loads(rs.hget("battle:%s" % battle_id, 'data')) rs.zrem("battles:%s" % battle['type'], battle_id) rs.zrem("privacy:%s" % battle['privacy'], battle_id) tanks = rs.smembers("battle:%s:tanks" % battle_id) for tank_id in tanks: rs.srem("tank:%s:battles" % tank_id, battle_id) users = rs.zrange('battle:%s:users' % battle_id, 0, -1) from warg.views.battle_followers import unFollowBattleByUser from warg.views.notifications import create_battle_notification, NTFY_BATTLE_KICK for user_id in users: unFollowBattleByUser(battle_id, int(user_id)) if user_id != uid: create_battle_notification(uid, user_id, battle_id, NTFY_BATTLE_KICK) rs.delete("battle:%s:tanks" % battle_id) from warg.views.jobs.battle_reminder import remove_battle_reminders remove_battle_reminders(battle_id) rs.delete("battle:%s:reminders" % battle_id) rs.delete("battle:%s" % battle_id) rs.sadd("whoosh:battles:deleted", battle_id) return 1 return 0
def battleDelUser(battle_id, user_id): if rs.exists("battle:" + str(battle_id)) != 1: return -1 uid = loggedUserUid() if rs.hget("battle:%s" % battle_id, "uid") != str(uid): return -2 create_battle_notification(uid, user_id, battle_id, NTFY_BATTLE_KICK) return unFollowBattleByUser(battle_id, user_id)
def battleAddUser(battle_id, user_id): if rs.exists("battle:" + str(battle_id)) != 1: return -1 if rs.hget("battle:%s" % battle_id, "uid") != str(loggedUserUid()): return -2 rs.sadd('battle:%s:invited' % battle_id, user_id) create_battle_notification(loggedUserUid(), user_id, battle_id, NTFY_BATTLE_INVITE) return followBattleByUser(battle_id, user_id)
def loggedUserUid(): #print "uSSID", request.cookies.get('uSSID') try: if request.cookies.get('uSSID') and rs.exists('ussid:' + request.cookies.get('uSSID')) == 1: uid = rs.hget('ussid:' + request.cookies.get('uSSID'), 'uid') return int(uid) except: pass return 0
def delete_group(group_id, admin=0): if rs.exists("group:%s" % group_id) == 1 and (admin == 1 or str(loggedUserUid()) == rs.hget("group:%s" % group_id, 'uid')): users = rs.smembers("group:%s:users" % group_id) for user_id in users: rs.srem("user:%s:groups" % user_id, group_id) rs.delete("group:%s:users" % group_id) rs.delete("group:%s" % group_id) return 1 else: return 0
def group_unfollow(group_id): if rs.exists("group:%s" % group_id) != 1: return -1 uid = loggedUserUid() if uid == 0: return -2 #user can't leave own group if rs.hget("group:%s" % group_id, "uid") == str(uid): return -3 return group_del_user(group_id, uid)
def unfollowUser(user_id): if rs.exists("users:" + str(user_id)) != 1: return -1 uid = loggedUserUid() if uid == 0: return -2 rs.srem('user:%s:following' % uid, user_id) rs.srem('user:%s:followers' % user_id, uid) rs.sdiffstore('user:%s:follownig_looks' % uid, 'user:%s:follownig_looks' % uid, 'user_looks:' + str(user_id)) return rs.scard('user:%s:followers' % user_id)
def battleRejectUser(battle_id, user_id): if rs.exists("battle:" + str(battle_id)) != 1: return -1 uid = loggedUserUid() if rs.hget("battle:%s" % battle_id, "uid") != str(uid): return -2 if uid == user_id: return -3 rs.zadd('battle:%s:users' % battle_id, user_id, 0) rs.srem('battle:%s:accepted' % battle_id, user_id) create_battle_notification(uid, user_id, battle_id, NTFY_BATTLE_REJECT) return rs.scard('battle:%s:accepted' % battle_id)
def battleAddExternalUser(battle_id, account_id): if rs.exists("battle:" + str(battle_id)) != 1: return -1 if rs.hget("battle:%s" % battle_id, "uid") != str(loggedUserUid()): return -2 access_token = request.args.get("access_token", None) if access_token is None: return json.dumps("No access token") wotuid = 'wot_user:%s' % account_id if rs.exists(wotuid) != 1: from warg.views.users import insert_wot_user, account_info data = account_info(access_token, account_id) if data['status'] == 'ok': uid = insert_wot_user(data['data'][str(account_id)], 1) else: return json.dumps("Error: " + data['error']['message']) else: uid = rs.hget(wotuid, 'uid') create_battle_notification(loggedUserUid(), int(uid), battle_id, NTFY_BATTLE_INVITE) rs.sadd('battle:%s:invited' % battle_id, uid) return followBattleByUser(battle_id, uid)
def unfollowBattle(battle_id): if rs.exists("battle:" + str(battle_id)) != 1: return -1 uid = loggedUserUid() if uid == 0: return -2 if rs.hget("battle:%s" % battle_id, "uid") == str(uid): return -3 if rs.sismember('battle:%s:invited' % battle_id, uid): create_battle_notification(uid, 0, battle_id, NTFY_INVITE_REJECT) else: create_battle_notification(uid, 0, battle_id, NTFY_BATTLE_UFLLOW) return unFollowBattleByUser(battle_id, uid)
def battleAcceptUser(battle_id, user_id, admin=False): if rs.exists("battle:" + str(battle_id)) != 1: return -1 uid = loggedUserUid() if rs.hget("battle:%s" % battle_id, "uid") != str(uid) and not admin: return -2 rs.zadd('battle:%s:users' % battle_id, user_id, 1) rs.sadd('battle:%s:accepted' % battle_id, user_id) rs.zadd('user:%s:battles' % user_id, battle_id, rs.zscore("battles_ids", battle_id)) if uid != user_id: create_battle_notification(loggedUserUid(), user_id, battle_id, NTFY_BATTLE_ACCEPT) return rs.scard('battle:%s:accepted' % battle_id)
def detail(user_id): #return request.headers.get('Cookie') u = None if (user_id is not None) and rs.exists('users:' + str(user_id)) == 1: u = json.loads(rs.get("users:" + str(user_id))) u['soc_links'] = list(rs.smembers('user_soc_links:' + str(user_id))) u['is_online'] = int(rs.sismember('users_online', user_id)) u['is_follow'] = int(rs.sismember('user:%s:followers' % user_id, loggedUserUid())) u['virtual'] = int(rs.sismember('users:virtual', user_id)) u['memo'] = rs.get('user:%s:memo:%s' % (user_id, loggedUserUid())) clan_id = int(rs.get("user:%s:clan" % user_id) or 0) if clan_id > 0: u['clan'] = json.loads(rs.hget("clan:%s" % clan_id, 'data')) #u = json.dumps(u) return u or {}
def battleFollowUser(battle_id): if rs.exists("battle:" + str(battle_id)) != 1: return -1 uid = loggedUserUid() if uid == 0: return -2 if rs.sismember('battle:%s:invited' % battle_id, uid): create_battle_notification(uid, 0, battle_id, NTFY_INVITE_ACCEPT) rs.srem('battle:%s:invited' % battle_id, uid) rs.zadd('battle:%s:users' % battle_id, uid, 1) rs.sadd('battle:%s:accepted' % battle_id, uid) rs.zadd('user:%s:battles' % uid, battle_id, rs.zscore("battles_ids", battle_id)) return 1 else: create_battle_notification(uid, 0, battle_id, NTFY_BATTLE_FOLLOW) return followBattleByUser(battle_id, uid)
def followExternalUser(account_id): uid = loggedUserUid() if uid == 0: return -2 access_token = request.args.get("access_token", None) if access_token is None: return json.dumps("No access token") wotuid = 'wot_user:%s' % account_id if rs.exists(wotuid) != 1: from warg.views.users import insert_wot_user, account_info data = account_info(access_token, account_id) if data['status'] == 'ok': user_id = insert_wot_user(data['data'][str(account_id)], 1) else: return json.dumps("Error: " + data['error']['message']) else: user_id = rs.hget(wotuid, 'uid') return followUserByUser(user_id, str(uid))
def get_dialogs(): uid = loggedUserUid() if uid == 0: return [] if rs.exists("chat:user:%s:dialogs" % uid) != 1: return [] offset = int(request.args.get("offset", 0)) count = int(request.args.get("count", 10)) rows = [] lua = """local r1 = redis.call('ZREVRANGE', KEYS[1], KEYS[2], KEYS[3]); for i = 1, table.getn(r1) do local dialog = r1[i] local chid = redis.call('ZREVRANGE', dialog, 0, 0)[1]; r1[i] = {} r1[i][1] = redis.call('hget', chid, 'id') r1[i][2] = redis.call('hget', chid, 'text') r1[i][3] = redis.call('zscore', dialog, chid) local uid = redis.call('hget', chid, 'sid') r1[i][4] = redis.call('get', 'users:' .. tostring(uid)) local uid = redis.call('hget', chid, 'rid') r1[i][5] = redis.call('get', 'users:' .. tostring(uid)) r1[i][6] = redis.call('sismember', 'users_online', uid) r1[i][7] = redis.call('hget', chid, 'is_read') end return r1;""" ids = rs.eval(lua, 3, "chat:user:%s:dialogs" % uid, offset, offset + count - 1) #ids = rs.sort('look:' + str(look_id) + ':comments', start=offset, num=count, get='#') for cmid in ids: is_read = False if cmid[6] == 'False' else json.loads(cmid[6]) cmnt = { 'id': int(cmid[0]), 'text': cmid[1], 'create_date': int(cmid[2]), 'is_read': is_read } cmnt['user'] = json.loads(cmid[3]) cmnt['user']['is_online'] = cmid[5] cmnt['companion'] = json.loads( cmid[3]) if uid != cmnt['user']['id'] else json.loads(cmid[4]) rows.append(cmnt) return rows
def register_wot(): access_token = request.args.get("access_token", None) if access_token is None: return json.dumps("No access token") wotuid = 'wot_user:'******'account_id') print "wotuid", wotuid, access_token if rs.exists(wotuid) != 1: data = account_info(access_token, request.args.get('account_id')) if data['status'] == 'ok': if data['data'][request.args.get('account_id')] is not None: insert_wot_user(data['data'][request.args.get('account_id')]) else: return json.dumps("Error: Null profile") else: return json.dumps("Error: " + data['error']['message']) if rs.hget(wotuid, 'virtual') == '1': rs.hset(wotuid, 'virtual', 0) uid = rs.hget(wotuid, 'uid') rs.srem("users:virtual", uid) return make_login_response(uid, False)
def get_user_unread(uid): if rs.exists("chat:user:%s:unread" % uid) != 1: return [] try: offset = int(request.args.get("offset", 0)) count = int(request.args.get("count", 10)) except: offset = 0 count = 10 rows = [] lua = """local r1 = redis.call('ZREVRANGE', KEYS[1], KEYS[2], KEYS[3]); for i = 1, table.getn(r1) do local chid = r1[i] r1[i] = {} r1[i][1] = redis.call('hget', chid, 'id') r1[i][2] = redis.call('hget', chid, 'text') r1[i][3] = redis.call('zscore', KEYS[1], chid) local uid = redis.call('hget', chid, 'sid') r1[i][4] = redis.call('get', 'users:' .. tostring(uid)) r1[i][5] = redis.call('hget', chid, 'type') r1[i][6] = redis.call('hget', chid, 'battle_id') r1[i][7] = redis.call('sismember', 'users_online', uid) end return r1;""" ids = rs.eval(lua, 3, "chat:user:%s:unread" % uid, offset, offset + count - 1) #ids = rs.sort('look:' + str(look_id) + ':comments', start=offset, num=count, get='#') for cmid in ids: cmnt = { 'id': int(cmid[0]), 'text': cmid[1], 'create_date': int(cmid[2]), "type": cmid[4] } if cmid[5] is not None: cmnt["battle_id"] = cmid[5] cmnt['user'] = json.loads(cmid[3]) cmnt['user']['is_online'] = cmid[6] rows.append(cmnt) return rows
def insert_wot_user(profile, virtual=0): uid = rs.incr('users_counter') wotuid = 'wot_user:'******'account_id']) user_data = {'id': uid, 'avatar': None, 'create_date': int(mktime(datetime.now().timetuple()))} for k in profile: if k != 'private': user_data[k] = profile[k] from warg.views.followers import followUserByUser if profile['private'] is not None: for fid in profile['private']['friends']: wotfid = 'wot_user:%s' % fid if rs.exists(wotfid) == 1: followUserByUser(rs.hget(wotfid, 'uid'), str(uid)) rs.set("users:" + str(uid), json.dumps(user_data)) rs.sadd("user_soc_links:" + str(uid), wotuid) rs.hmset(wotuid, {'uid': str(uid), 'profile': json.dumps(profile), "virtual": virtual}) if virtual == 1: rs.sadd("users:virtual", uid) from warg.views.full_text import storeUserInIndex storeUserInIndex(user_data) rs.sadd("user_ids", uid) return uid
def get_chat_history(sid): uid = loggedUserUid() if uid == 0: return [] dialog = "chat:dialog:%s:%s" % (min(int(uid), sid), max(int(uid), sid)) if rs.exists(dialog) != 1: return [] offset = int(request.args.get("offset", 0)) count = int(request.args.get("count", 10)) rows = [] lua = """local r1 = redis.call('ZREVRANGE', KEYS[1], KEYS[2], KEYS[3]); for i = 1, table.getn(r1) do local chid = r1[i] r1[i] = {} r1[i][1] = redis.call('hget', chid, 'id') r1[i][2] = redis.call('hget', chid, 'text') r1[i][3] = redis.call('zscore', KEYS[1], chid) r1[i][4] = redis.call('hget', chid, 'sid') r1[i][5] = redis.call('hget', chid, 'rid') r1[i][6] = redis.call('hget', chid, 'is_read') end return r1;""" ids = rs.eval(lua, 3, dialog, offset, offset + count - 1) #ids = rs.sort('look:' + str(look_id) + ':comments', start=offset, num=count, get='#') for cmid in ids: is_read = False if cmid[5] == 'False' else json.loads(cmid[5]) cmnt = { 'id': int(cmid[0]), 'text': cmid[1], 'create_date': int(cmid[2]), 'is_read': is_read, 'sid': int(cmid[3]), 'rid': int(cmid[4]) } rows.append(cmnt) return rows