def chat_presence(request): """Sets user presence on mattermost.""" if not request.user.is_authenticated: return JsonResponse({'status': 'OK'}) profile = request.user.profile if not profile.chat_id: return JsonResponse({'status': 'OK'}) # setup driver driver = get_driver() # determine current status/ should we set the user as online in mm? current_status = driver.client.post('/users/status/ids', [profile.chat_id]) manual = current_status[0]['manual'] current_status = current_status[0]['status'] set_status = current_status == 'offline' or manual or settings.DEBUG # if so, make it so if set_status: new_status = 'online' if current_status in ['away', 'dnd']: new_status = current_status driver.client.put(f'/users/{profile.chat_id}/status', { 'user_id': profile.chat_id, 'status': new_status }) # set a marker of when this user was last seen.. # so that get_user_prsence can clean it up later redis = RedisService().redis redis.set(profile.chat_id, timezone.now().timestamp()) redis.set(f"chat:{profile.chat_id}", new_status) return JsonResponse({'status': 'OK'})
def create_grant_category_size_cache(): print('create_grant_category_size_cache') redis = RedisService().redis for category in GrantCategory.objects.all(): key = f"grant_category_{category.category}" val = Grant.objects.filter(categories__category__contains=category.category).count() redis.set(key, val)
def create_grant_category_size_cache(): print('create_grant_category_size_cache') grant_types = GrantType.objects.all() redis = RedisService().redis for g_type in grant_types: for category in GrantCategory.objects.all(): key = f"grant_category_{g_type.name}_{category.category}" val = Grant.objects.filter(active=True, hidden=False, grant_type=g_type, categories__category__contains=category.category).count() redis.set(key, val)
def set_view_count(self, amount): try: redis = RedisService().redis result = redis.set(self.view_count_redis_key, amount) 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])