File

mod_rest/example/prosody_oauth.py @ 5229:c24a622a7b85

mod_http_oauth2: Fix appending of query parts in error redirects Looks like this meant to check whether the redirect_uri has a ?query part, but forgot to inspect the field for this in the returned table.
author Kim Alvefur <zash@zash.se>
date Fri, 10 Mar 2023 11:54:30 +0100
parent 4952:ccce785f53e1
child 5269:0e5a37f55440
line wrap: on
line source

from oauthlib.oauth2 import LegacyApplicationClient
from requests_oauthlib import OAuth2Session


class ProsodyRestClient(LegacyApplicationClient):
    pass


class ProsodyRestSession(OAuth2Session):
    def __init__(self, base_url=None, token_url=None, rest_url=None, *args, **kwargs):
        if base_url and not token_url:
            token_url = base_url + "/oauth2/token"
        if base_url and not rest_url:
            rest_url = base_url + "/rest"
        self._prosody_rest_url = rest_url
        self._prosody_token_url = token_url

        super().__init__(client=ProsodyRestClient(*args, **kwargs))

    def fetch_token(self, *args, **kwargs):
        return super().fetch_token(token_url=self._prosody_token_url, *args, **kwargs)

    def xmpp(self, json=None, *args, **kwargs):
        return self.post(self._prosody_rest_url, json=json, *args, **kwargs)


if __name__ == "__main__":
    # Example usage

    # from prosody_oauth import ProsodyRestSession
    from getpass import getpass

    p = ProsodyRestSession(base_url=input("Base URL: "), client_id="app")
    
    p.fetch_token(username=input("XMPP Address: "), password=getpass("Password: "))

    print(p.xmpp(json={"disco": True, "to": "jabber.org"}).json())