simplehacker01
diff --git a/‎auth0/v3/management/branding.py
Lines changed: 92 additions & 0 deletions b/‎auth0/v3/management/branding.py
Lines changed: 92 additions & 0 deletions
@@ -0,0 +1,92 @@
+from .rest import RestClient
+
+
+class Branding(object):
+    """Auth0 Branding endpoints
+
+    Args:
+        domain (str): Your Auth0 domain, e.g: 'username.auth0.com'
+
+        token (str): Management API v2 Token
+
+        telemetry (bool, optional): Enable or disable Telemetry
+            (defaults to True)
+
+        timeout (float or tuple, optional): Change the requests
+            connect and read timeout. Pass a tuple to specify
+            both values separately or a float to set both to it.
+            (defaults to 5.0 for both)
+
+        rest_options (RestClientOptions): Pass an instance of
+            RestClientOptions to configure additional RestClient
+            options, such as rate-limit retries.
+            (defaults to None)
+    """
+
+    def __init__(
+        self,
+        domain,
+        token,
+        telemetry=True,
+        timeout=5.0,
+        protocol="https",
+        rest_options=None,
+    ):
+        self.domain = domain
+        self.protocol = protocol
+        self.client = RestClient(
+            jwt=token, telemetry=telemetry, timeout=timeout, options=rest_options
+        )
+
+    def _url(self, *args):
+        url = "{}://{}/api/v2/branding".format(self.protocol, self.domain)
+        for p in args:
+            if p is not None:
+                url = "{}/{}".format(url, p)
+        return url
+
+    def get(self, aud=None):
+        """Retrieve branding settings. Requires "read:branding" scope.
+
+        See: https://auth0.com/docs/api/management/v2#!/Branding/get_branding
+        """
+
+        return self.client.get(self._url())
+
+    def update(self, body):
+        """Update branding settings. Requires "update:branding" scope.
+
+        Args:
+            body (dict): Attributes for the updated trigger binding.
+
+        See: https://auth0.com/docs/api/management/v2#!/Branding/patch_branding
+        """
+
+        return self.client.patch(self._url(), data=body)
+
+    def get_template_universal_login(self):
+        """Get template for New Universal Login Experience. Requires "read:branding" scope.
+
+        See: https://auth0.com/docs/api/management/v2#!/Branding/get_universal_login
+        """
+
+        return self.client.get(self._url("templates", "universal-login"))
+
+    def delete_template_universal_login(self):
+        """Delete template for New Universal Login Experience. Requires "delete:branding" scope.
+
+        See: https://auth0.com/docs/api/management/v2#!/Branding/delete_universal_login
+        """
+
+        return self.client.delete(self._url("templates", "universal-login"))
+
+    def update_template_universal_login(self, body):
+        """Update template for New Universal Login Experience. Requires "update:branding" scope.
+
+        Args:
+            body (str): Complete HTML content to assign to the template. See linked API documentation for example.
+
+        See: https://auth0.com/docs/api/management/v2#!/Branding/put_universal_login
+        """
+
+        return self.client.put(self._url("templates", "universal-login"), type="put_universal-login_body", body={"template":body})