def get_view_count(self):
try:
redis = RedisService().redis
result = redis.get(self.view_count_redis_key)
if not result:
return 0
return int(result.decode('utf-8'))
except KeyError:
return 0
def handle(self, *args, **options):
# connect to API
d = get_driver()
teams = d.teams.get_teams()
# connect to redis
redis = RedisService().redis
# outer vars
all_usernames = []
all_user_statuses = {}
all_response = []
for team in teams:
if team['display_name'] == 'Codefund':
continue
# pull back users on team
print(team['display_name'])
all_users = []
cont = True
per_page = 60
page = 0
while cont:
params = {
'in_team': team['id'],
'sort': 'last_activity_at',
'per_page': per_page,
'page': page
}
users = d.users.get_users(params=params)
all_users += users
cont = len(users) == per_page
page += 1
#for testing on small amount of data
#if settings.DEBUG:
# cont = False
for user in users:
pass
# get through all users
print(f"- {len(all_users)}")
users_status = d.client.post('/users/status/ids',
[ele['id'] for ele in all_users])
all_response += users_status
users_status_by_id = {ele['user_id']: ele for ele in users_status}
all_usernames += [ele['username'].lower() for ele in all_users]
# iterate through each one, and sync it to our DB
for user in all_users:
last_activity_at_1 = user['last_activity_at']
last_activity_at_2 = users_status_by_id[
user['id']]['last_activity_at']
status = users_status_by_id[user['id']]['status']
last_activity_at = max(last_activity_at_2, last_activity_at_1)
username = user['username']
timestamp = int(last_activity_at / 1000)
timestamp = timezone.datetime.utcfromtimestamp(
timestamp).replace(tzinfo=pytz.utc)
all_user_statuses[username] = (status, timestamp, user['id'])
# look for manual statuses set by /api/v0.1/chat route and clean them up if needed
for ele in all_response:
user_id = ele['user_id']
status = ele['status']
if status == 'offline':
continue
# user has been offline for 10 mins
# update mattermost, and redis
max_delta = (10 * 60)
# calc it
now = timezone.now().timestamp()
last_action_mattermost = int(ele['last_activity_at'] / 1000)
redis_response = redis.get(user_id)
last_seen_gc = int(
float(redis_response)) if redis_response else now
# do update
is_away_mm = ((now - last_action_mattermost) > max_delta)
is_away_gc = (now - last_seen_gc) > max_delta
manual = ele['manual']
update_ele = (manual and is_away_gc) or (not manual
and is_away_mm)
if update_ele or settings.DEBUG:
new_status = 'offline'
d.client.put(f'/users/{user_id}/status', {
'user_id': user_id,
'status': new_status
})
redis.set(user_id, 0)
# update all chat ids not in DB
profiles = Profile.objects.filter(handle__in=all_usernames, chat_id='')
for profile in profiles:
_, _, profile.chat_id = all_user_statuses[profile.handle]
bulk_update(profiles, update_fields=['chat_id'])
# update all chat info that is in redis
# all_user_statuses[username] = (status, timestamp, user['id'])
for username, ele in all_user_statuses.items():
key = f"chat:{ele[2]}"
redis.set(key, ele[0])
