async def delete_account(request: Request): """ Delete account (stop monitoring) """ try: user_id = request.session["user_id"] except KeyError: LOGGER.debug( "Unauthenticated user attempted to delete account [%s]", request.client, ) return RedirectResponse("/") try: user = await User.objects.get(id=user_id) except orm.exceptions.NoMatch: LOGGER.warning( "User %s not found in the database; cannot delete [%s]", user_id, request.client, ) else: await user.delete() LOGGER.info("User %s deleted account [%s]", user_id, request.client) auth.sign_out(request) return RedirectResponse("/")
async def _update_spotify_tokens(user: User) -> bool: """ Update the user's Spotify tokens. Returns success status """ try: exchange_data: TokenExchangeData = await get_new_access_token( user.spotifyRefreshToken, GrantType.REFRESH_TOKEN) except SpotifyApiError as err: LOGGER.warning( "Exiting update loop. Could not refresh Spotify token for " "user %s: %s", user.id, err, ) err_dict = err.response_json() if err_dict.get("error_description") == "Refresh token revoked": LOGGER.warning( "Deleting user %s as their Spotify refresh token is revoked " "and we have no way to recover :(", user.id, ) await user.delete() return False await user.update( spotifyExpiresAt=calc_spotify_expiry(exchange_data.expires_in), spotifyAccessToken=exchange_data.access_token, spotifyRefreshToken=exchange_data.refresh_token or "", updatedAt=datetime.now(timezone.utc), ) return True
async def _set_user_status(user: User, user_profile_args: UserProfileArgs, status_set_last_time: bool) -> bool: """ Set the user status & update their database entry. Returns success status """ try: await set_status(user_profile_args, user.slackAccessToken) except SlackApiError as err: LOGGER.warning( "Exiting update loop. Could not set status for user %s: " "%s", user.id, err, ) return False await user.update( statusSetLastTime=status_set_last_time, updatedAt=datetime.now(timezone.utc), ) return True
async def slack_grant_callback(request: Request, code: Optional[str] = None): """ Slack grant callback (receive access token) """ if code is None: LOGGER.warning( "Spotify grant callback error: 'code' was not provided by Spotify") return RedirectResponse("/") try: exchange_data: TokenExchangeData = await get_new_access_token(code, ) except SlackApiError as err: LOGGER.warning("%s", err) return RedirectResponse("/") request.session["slack_id"] = exchange_data.user_id request.session["slack_access_token"] = exchange_data.access_token await sign_in(request) LOGGER.debug("Slack grant callback successful [%s]", request.client) return RedirectResponse("/")
async def spotify_grant_callback(request: Request, code: Optional[str] = None, error: Optional[str] = None): """ Spotify grant callback (receive access & refresh tokens) """ if code is None and error is None: error = f"Neither 'code' nor 'error' were provided from Spotify" if error is not None: LOGGER.warning("Spotify grant callback error: %s", error) return RedirectResponse("/") try: exchange_data: TokenExchangeData = await get_new_access_token( cast(str, code), GrantType.CODE # Checks above ensure code is str ) except SpotifyApiError as err: LOGGER.warning("%s", err) return RedirectResponse("/") try: me_data: MeData = await get_me(exchange_data.access_token) except SpotifyApiError as err: LOGGER.warning("%s", err) return RedirectResponse("/") request.session["spotify_id"] = me_data.id request.session["spotify_expires_at"] = ( datetime.now(timezone.utc) + timedelta(seconds=exchange_data.expires_in)).isoformat() request.session["spotify_access_token"] = exchange_data.access_token request.session["spotify_refresh_token"] = exchange_data.refresh_token await sign_in(request) LOGGER.debug("Spotify grant callback successful [%s]", request.client) return RedirectResponse("/")
async def _set_user_status(user: User, user_profile_args: UserProfileArgs, status_set_last_time: bool) -> bool: global UPDATE_THRESHOLD """ Set the user status & update their database entry. Returns success status """ try: current_profile = await get_status(user.slackAccessToken) if current_profile.profile.status_text == user_profile_args.status_text: return True LOGGER.info("Setting user status %s", user_profile_args) await asyncio.sleep(1) user_profile_data: UserProfileData = await set_status( user_profile_args, user.slackAccessToken) except SlackApiError as err: LOGGER.warning( "Exiting update loop. Could not set status for user %s: %s", user.id, err, ) UPDATE_THRESHOLD = datetime.now(timezone.utc) + timedelta(seconds=6) return False if not user_profile_data.ok: LOGGER.warning( "Exiting update loop. Could not set status for user %s: %s", user.id, user_profile_data.error, ) if user_profile_data.error in {"token_revoked"}: LOGGER.warning("Slack token revoked. Deleting user %s", user.id) await user.delete() return False await user.update( statusSetLastTime=status_set_last_time, updatedAt=datetime.now(timezone.utc), ) return True
async def _update_user(user: User) -> None: """ Update a single user """ global UPDATE_THRESHOLD # pylint:disable=global-statement update_threshold_delta = UPDATE_THRESHOLD - datetime.now(timezone.utc) if update_threshold_delta.total_seconds() > 0: LOGGER.debug( "Sleeping update thread for %s for %ss", user.id, update_threshold_delta.total_seconds(), ) await asyncio.sleep(update_threshold_delta.total_seconds()) # Handle Spotify token refreshes spotify_token_expired = user.spotifyExpiresAt <= datetime.now( timezone.utc) - timedelta(minutes=5) if spotify_token_expired and not user.spotifyRefreshToken: LOGGER.warning( "Deleting user %s as their Spotify token is expired and no " "refresh token is available :(", user.id, ) await user.delete() return if spotify_token_expired: LOGGER.debug("Refreshing Spotify token for user %s", user.id) update_ok = await _update_spotify_tokens(user) if not update_ok: return LOGGER.debug("Refreshing Spotify token for user %s COMPLETE", user.id) # Retrieve Spotify player status try: player: Optional[PlayerData] = await get_player(user.spotifyAccessToken ) except SpotifyApiError as err: if err.retry_after is None: LOGGER.warning( "Exiting update loop. Could not retrieve player data for user %s: " "%s", user.id, err, ) else: UPDATE_THRESHOLD = datetime.now( timezone.utc) + timedelta(seconds=err.retry_after) LOGGER.warning( "Exiting update loop. Spotify is throttling for %ss", err.retry_after, ) return if player is not None and player.item is not None and player.is_playing: user_profile_args = UserProfileArgs( status_text=_calc_status_text(player.item), status_emoji=get_custom_emoji(user, player.item), ) LOGGER.debug("Setting user status %s", user_profile_args) # TODO rm await _set_user_status(user, user_profile_args, True) elif user.statusSetLastTime: user_profile_args = UserProfileArgs(status_text="", status_emoji="") await _set_user_status(user, user_profile_args, False)