oauth2-cli-auth 1.1.0__tar.gz → 1.2.0__tar.gz

{oauth2_cli_auth-1.1.0 → oauth2_cli_auth-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oauth2-cli-auth
-Version: 1.1.0
+Version: 1.2.0
 Summary: Authenticate against OAuth2 Provider in Python CLIs
 Home-page: https://github.com/timo-reymann/python-oauth2-cli-auth
 License: MIT

oauth2_cli_auth-1.2.0/oauth2_cli_auth/_timeout.py

@@ -0,0 +1,23 @@
+import signal
+
+
+class TimeoutException(Exception):
+    pass
+
+
+def _timeout_handler(signum, frame):
+    raise TimeoutException("Operation timed out")
+
+
+def _method_with_timeout(your_method, timeout_seconds=5, *args, **kwargs):
+    signal.signal(signal.SIGALRM, _timeout_handler)
+    signal.alarm(timeout_seconds)
+
+    try:
+        result = your_method(*args, **kwargs)
+    except TimeoutException as te:
+       raise te
+    finally:
+        signal.alarm(0)  # Reset the alarm
+
+    return result

oauth2_cli_auth-1.2.0/oauth2_cli_auth/_urllib_util.py

@@ -0,0 +1,25 @@
+import json
+import time
+import urllib
+
+
+def _urlopen_with_backoff(url, max_retries=3, base_delay=1, timeout=15):
+    retries = 0
+
+    while retries < max_retries:
+        try:
+            response = urllib.request.urlopen(url, timeout=timeout)
+            return response
+        except Exception:
+            retries += 1
+            delay = base_delay * (2 ** retries)
+            time.sleep(delay)
+
+    raise Exception(f"Failed to open URL after {max_retries} retries")
+
+
+def _load_json(url_or_request: str | urllib.request.Request) -> dict:
+    with _urlopen_with_backoff(url_or_request) as response:
+        response_data = response.read().decode('utf-8')
+        json_response = json.loads(response_data)
+    return json_response

{oauth2_cli_auth-1.1.0 → oauth2_cli_auth-1.2.0}/oauth2_cli_auth/code_grant.py

@@ -3,7 +3,7 @@ from dataclasses import dataclass
 import urllib.request
 import urllib.parse
 import base64
-import json
+from oauth2_cli_auth._urllib_util import _load_json
 
 
 @dataclass
@@ -31,13 +31,6 @@ class OAuth2ClientInfo:
         )
 
 
-def _load_json(url_or_request: str | urllib.request.Request) -> dict:
-    with urllib.request.urlopen(url_or_request) as response:
-        response_data = response.read().decode('utf-8')
-        json_response = json.loads(response_data)
-    return json_response
-
-
 def load_oidc_config(odic_well_known_endpoint: str) -> dict:
     config = _load_json(odic_well_known_endpoint)
     return config

oauth2_cli_auth-1.2.0/oauth2_cli_auth/conftest.py

@@ -0,0 +1,12 @@
+import urllib
+from unittest.mock import patch
+
+import pytest
+
+
+@pytest.fixture
+def create_urlopen_mock():
+    def mock(buffer):
+        return patch.object(urllib.request, 'urlopen', return_value=buffer)
+
+    return mock

{oauth2_cli_auth-1.1.0 → oauth2_cli_auth-1.2.0}/oauth2_cli_auth/http_server.py

@@ -2,6 +2,7 @@ from http.server import BaseHTTPRequestHandler, HTTPServer
 from string import Template
 from typing import Optional
 from urllib.parse import parse_qs, urlparse
+from oauth2_cli_auth._timeout import _method_with_timeout, TimeoutException
 
 
 class CallbackPageTemplate:
@@ -150,10 +151,13 @@ class CallbackPageTemplate:
 class OAuthRedirectHandler(BaseHTTPRequestHandler):
     callback_template = CallbackPageTemplate()
 
+    def log_message(self, format, *args):
+        # silence the log messages
+        pass
+
     def do_GET(self):
         params = parse_qs(urlparse(self.path).query)
 
-
         has_error = "code" not in params or len(params['code']) != 1 or params['code'][0].strip() == ""
 
         if has_error:
@@ -197,9 +201,22 @@ class OAuthCallbackHttpServer(HTTPServer):
     def callback_url(self):
         return f"http://localhost:{self.server_port}"
 
-    def wait_for_code(self, attempts: int = 3) -> Optional[int]:
+    def wait_for_code(self, attempts: int = 3, timeout_per_attempt=10) -> Optional[int]:
+        """
+        Wait for the server to open the callback page containing the code query parameter.
+
+        It tries for #attempts with a timeout of #timeout_per_attempts for each attempt.
+        This prevents the CLI from getting stuck by unsolved callback URls
+
+        :param attempts: Amount of attempts
+        :param timeout_per_attempt: Timeout for each attempt to be successful
+        :return: Code from callback page or None if the callback page is not called successfully
+        """
         for i in range(0, attempts):
-            self.handle_request()
+            try:
+                _method_with_timeout(self.handle_request, timeout_seconds=timeout_per_attempt)
+            except TimeoutException:
+                continue
             if self.get_code() is not None:
                 return self.get_code()
 

{oauth2_cli_auth-1.1.0 → oauth2_cli_auth-1.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "oauth2-cli-auth"
-version = "1.1.0"
+version = "1.2.0"
 description = "Authenticate against OAuth2 Provider in Python CLIs"
 authors = ["Timo Reymann <mail@timo-reymann.de>"]
 license = "MIT"

oauth2_cli_auth-1.1.0/oauth2_cli_auth/success.html

@@ -1,118 +0,0 @@
-<html lang="en">
-
-<head>
-    <title>bcs</title>
-    <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
-    <meta name="charset" content="utf-8">
-    <style>
-        * {
-            margin: 0;
-            padding: 0;
-        }
-
-        body {
-            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol";
-        }
-
-        html, body {
-            display: flex;
-            align-items: center;
-            justify-content: center;
-            height: 100%;
-        }
-
-        .message {
-            text-align: center;
-            padding: 40px;
-        }
-
-        .animation-ctn {
-            text-align: center;
-            margin-top: 5em;
-        }
-
-        @keyframes checkmark {
-            0% {
-                stroke-dashoffset: 100px
-            }
-
-            100% {
-                stroke-dashoffset: 0px
-            }
-        }
-
-        @keyframes checkmark-circle {
-            0% {
-                stroke-dashoffset: 480px
-            }
-
-            100% {
-                stroke-dashoffset: 960px
-            }
-        }
-
-        @keyframes colored-circle {
-            0% {
-                opacity: 0
-            }
-
-            100% {
-                opacity: 100
-            }
-        }
-
-
-        .icon svg polyline {
-            -webkit-animation: checkmark 0.25s ease-in-out 0.7s backwards;
-            animation: checkmark 0.25s ease-in-out 0.7s backwards
-        }
-
-        .icon svg circle {
-            -webkit-animation: checkmark-circle 0.6s ease-in-out backwards;
-            animation: checkmark-circle 0.6s ease-in-out backwards;
-        }
-
-        .icon svg circle#colored {
-            -webkit-animation: colored-circle 0.6s ease-in-out 0.7s backwards;
-            animation: colored-circle 0.6s ease-in-out 0.7s backwards;
-        }
-    </style>
-</head>
-
-<body>
-<div class="message">
-    <div class="animation-ctn">
-        <div class="icon">
-            <svg xmlns="http://www.w3.org/2000/svg" width="154px" height="154px">
-                <g fill="none" stroke="#22AE73" stroke-width="2">
-                    <circle cx="77" cy="77" r="72"
-                            style="stroke-dasharray:480px, 480px; stroke-dashoffset: 960px;"></circle>
-                    <circle id="colored" fill="#22AE73" cx="77" cy="77" r="72"
-                            style="stroke-dasharray:480px, 480px; stroke-dashoffset: 960px;"></circle>
-                    <polyline class="st0" stroke="#fff" stroke-width="10" points="43.5,77.8 63.7,97.9 112.2,49.4 "
-                              style="stroke-dasharray:100px, 100px; stroke-dashoffset: 200px;"/>
-                </g>
-            </svg>
-        </div>
-    </div>
-
-    <div class="animation-ctn">
-        <div class="icon">
-            <svg xmlns="http://www.w3.org/2000/svg" width="154px" height="154px">
-                <g fill="none" stroke="#F44812" stroke-width="2">
-                    <circle cx="77" cy="77" r="72"
-                            style="stroke-dasharray:480px, 480px; stroke-dashoffset: 960px;"></circle>
-                    <circle id="colored" fill="#F44812" cx="77" cy="77" r="72"
-                            style="stroke-dasharray:480px, 480px; stroke-dashoffset: 960px;"></circle>
-                    <polyline class="st0" stroke="#fff" stroke-width="10" points="43.5,77.8  112.2,77.8 "
-                              style="stroke-dasharray:100px, 100px; stroke-dashoffset: 200px;"/>
-                </g>
-            </svg>
-        </div>
-    </div>
-
-    <h1>$title</h1>
-    <p>$message</p>
-</div>
-</body>
-</html>