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
