the37lab-authlib 0.1.1750840415__py3-none-any.whl → 0.1.1750844514__py3-none-any.whl

the37lab_authlib/auth.py

@@ -1,5 +1,5 @@
 import inspect
-from flask import Blueprint, request, jsonify, current_app, url_for, redirect
+from flask import Blueprint, request, jsonify, current_app, url_for, redirect, g
 import jwt
 from datetime import datetime, timedelta
 from .db import Database
@@ -15,24 +15,20 @@ from functools import wraps
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
-def handle_auth_errors(f):
-    @wraps(f)
-    def decorated(*args, **kwargs):
-        try:
-            return f(*args, **kwargs)
-        except AuthError as e:
-            response = jsonify(e.to_dict())
-            response.status_code = e.status_code
-            return response
-    return decorated
-
 class AuthManager:
     def __init__(self, app=None, db_dsn=None, jwt_secret=None, oauth_config=None, id_type='integer'):
-        logger.info("INITIALIZING AUTHMANAGER {} - {} - {}".format(db_dsn, jwt_secret, not app))
         self.db = Database(db_dsn, id_type=id_type) if db_dsn else None
         self.jwt_secret = jwt_secret
-        logger.debug(f"Initializing AuthManager with JWT secret: {jwt_secret[:5]}..." if jwt_secret else "No JWT secret provided")
         self.oauth_config = oauth_config or {}
+        self.public_endpoints = {
+            'auth.login',
+            'auth.oauth_login',
+            'auth.oauth_callback',
+            'auth.refresh_token',
+            'auth.register',
+            'auth.get_roles'
+        }
+        self.bp = None
         
         if app:
             self.init_app(app)
@@ -124,16 +120,50 @@ class AuthManager:
 
             return f(*args, **kwargs)
         return decorated
+
+    def add_public_endpoint(self, endpoint):
+        """Mark an endpoint as public so it bypasses authentication."""
+        self.public_endpoints.add(endpoint)
+
+    def public_endpoint(self, f):
+        """Decorator to mark a view function as public."""
+        # Always register the bare function name so application level routes
+        # are exempt from authentication checks.
+        self.add_public_endpoint(f.__name__)
+
+        # If a blueprint is active, also register the blueprint-prefixed name
+        # used by Flask for endpoint identification.
+        if self.bp:
+            endpoint = f"{self.bp.name}.{f.__name__}"
+            self.add_public_endpoint(endpoint)
+        return f
     
     def init_app(self, app):
         app.auth_manager = self
         app.register_blueprint(self.create_blueprint())
+        @app.errorhandler(AuthError)
+        def handle_auth_error(e):
+            response = jsonify(e.to_dict())
+            response.status_code = e.status_code
+            return response
 
     def create_blueprint(self):
         bp = Blueprint('auth', __name__, url_prefix='/api/v1/users')
+        self.bp = bp
+        bp.public_endpoint = self.public_endpoint
+
+        @bp.errorhandler(AuthError)
+        def handle_auth_error(err):
+            response = jsonify(err.to_dict())
+            response.status_code = err.status_code
+            return response
+
+        @bp.before_request
+        def load_user():
+            if request.endpoint not in self.public_endpoints:
+                g.requesting_user = self._authenticate_request()
 
         @bp.route('/login', methods=['POST'])
-        @handle_auth_errors
         def login():
             data = request.get_json()
             username = data.get('username')
@@ -168,7 +198,6 @@ class AuthManager:
                 })
 
         @bp.route('/login/oauth', methods=['POST'])
-        @handle_auth_errors
         def oauth_login():
             provider = request.json.get('provider')
             if provider not in self.oauth_config:
@@ -180,7 +209,6 @@ class AuthManager:
             })
 
         @bp.route('/login/oauth2callback')
-        @handle_auth_errors
         def oauth_callback():
             code = request.args.get('code')
             provider = request.args.get('state')
@@ -197,28 +225,22 @@ class AuthManager:
             return redirect(f"{frontend_url}/oauth-callback?token={token}&refresh_token={refresh_token}")
 
         @bp.route('/login/profile')
-        @handle_auth_errors
         def profile():
-            token = request.headers.get('Authorization', '').split(' ')[-1]
-            user = self.validate_token(token)
+            user = g.requesting_user
             return jsonify(user)
 
         @bp.route('/api-tokens', methods=['GET'])
-        @handle_auth_errors
-        @self.require_auth
-        def get_tokens(requesting_user):
-            tokens = self.get_user_api_tokens(requesting_user['id'])
+        def get_tokens():
+            tokens = self.get_user_api_tokens(g.requesting_user['id'])
             return jsonify(tokens)
 
         @bp.route('/api-tokens', methods=['POST'])
-        @handle_auth_errors
-        @self.require_auth
-        def create_token(requesting_user):
+        def create_token():
             name = request.json.get('name')
             expires_in_days = request.json.get('expires_in_days')
             if not name:
                 raise AuthError('Token name is required', 400)
-            api_token = self.create_api_token(requesting_user['id'], name, expires_in_days)
+            api_token = self.create_api_token(g.requesting_user['id'], name, expires_in_days)
             return jsonify({
                 'id': api_token.id,
                 'name': api_token.name,
@@ -228,7 +250,6 @@ class AuthManager:
             })
 
         @bp.route('/token-refresh', methods=['POST'])
-        @handle_auth_errors
         def refresh_token():
             refresh_token = request.json.get('refresh_token')
             if not refresh_token:
@@ -253,20 +274,16 @@ class AuthManager:
                 raise AuthError('Invalid refresh token', 401)
 
         @bp.route('/api-tokens', methods=['POST'])
-        @handle_auth_errors
-        @self.require_auth
-        def create_api_token(requesting_user):
+        def create_api_token():
             name = request.json.get('name')
             if not name:
                 raise AuthError('Token name required', 400)
 
-            token = self.create_api_token(requesting_user['id'], name)
+            token = self.create_api_token(g.requesting_user['id'], name)
             return jsonify({'token': token.token})
 
         @bp.route('/api-tokens/validate', methods=['GET'])
-        @handle_auth_errors
-        @self.require_auth
-        def validate_api_token(requesting_user):
+        def validate_api_token():
             token = request.json.get('token')
             if not token:
                 raise AuthError('No API token provided', 401)
@@ -276,7 +293,7 @@ class AuthManager:
                 cur.execute("""
                     SELECT * FROM api_tokens 
                     WHERE user_id = %s AND id = %s
-                """, (requesting_user['id'], token))
+                """, (g.requesting_user['id'], token))
                 api_token = cur.fetchone()
 
             if not api_token:
@@ -297,9 +314,7 @@ class AuthManager:
             return jsonify({'valid': True})
 
         @bp.route('/api-tokens', methods=['DELETE'])
-        @handle_auth_errors
-        @self.require_auth
-        def delete_api_token(requesting_user):
+        def delete_api_token():
             token = request.json.get('token')
             if not token:
                 raise AuthError('Token required', 400)
@@ -310,7 +325,7 @@ class AuthManager:
                     DELETE FROM api_tokens 
                     WHERE user_id = %s AND id = %s
                     RETURNING id
-                """, (requesting_user['id'], token))
+                """, (g.requesting_user['id'], token))
                 deleted_id = cur.fetchone()
                 if not deleted_id:
                     raise ValueError('Token not found or already deleted')
@@ -318,7 +333,6 @@ class AuthManager:
             return jsonify({'deleted': True})
 
         @bp.route('/register', methods=['POST'])
-        @handle_auth_errors
         def register():
             data = request.get_json()
             
@@ -357,7 +371,6 @@ class AuthManager:
             return jsonify({'id': user.id}), 201
 
         @bp.route('/roles', methods=['GET'])
-        @handle_auth_errors
         def get_roles():
             with self.db.get_cursor() as cur:
                 cur.execute("SELECT * FROM roles")

{the37lab_authlib-0.1.1750840415.dist-info → the37lab_authlib-0.1.1750844514.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: the37lab_authlib
-Version: 0.1.1750840415
+Version: 0.1.1750844514
 Summary: Python SDK for the Authlib
 Author-email: the37lab <info@the37lab.com>
 Classifier: Programming Language :: Python :: 3
@@ -50,7 +50,7 @@ pip install -e .
 
 ```python
 from flask import Flask
-from authlib import AuthManager, require_auth
+from authlib import AuthManager
 
 app = Flask(__name__)
 
@@ -67,11 +67,24 @@ auth = AuthManager(
 )
 
 @app.route("/protected")
-@require_auth(roles=["admin"])
+@auth.require_auth(roles=["admin"])
 def protected_route():
     return "Protected content"
+
+@app.route("/public")
+@auth.public_endpoint
+def custom_public_route():
+    return "Public content"
 ```
 
+`AuthManager`'s blueprint now registers a global error handler for
+`AuthError` and authenticates requests for all of its routes by default.
+Authenticated users are made available as `flask.g.requesting_user`.
+Only the login, OAuth, token refresh, registration and role listing
+endpoints are exempt from this check. Additional routes can be marked as
+public using the `@auth.public_endpoint` decorator or
+`auth.add_public_endpoint("auth.some_endpoint")`.
+
 ## Configuration
 
 ### Required Parameters
@@ -144,7 +157,9 @@ def protected_route():
    - Get redirect URL from `/api/v1/users/login/oauth`.
    - Complete OAuth flow via `/api/v1/users/login/oauth2callback`.
 4. **Protected Routes:**
-   - Use `@require_auth()` decorator to protect Flask routes.
+   - All routes inside the provided blueprint are authenticated by default.
+     The authenticated user can be accessed via `g.requesting_user`.
+     Use `@auth.require_auth()` to protect custom routes in your application.
 
 ## User Object
 

{the37lab_authlib-0.1.1750840415.dist-info → the37lab_authlib-0.1.1750844514.dist-info}/RECORD

@@ -1,10 +1,10 @@
 the37lab_authlib/__init__.py,sha256=cFVTWL-0YIMqwOMVy1P8mOt_bQODJp-L9bfp2QQ8CTo,132
-the37lab_authlib/auth.py,sha256=dQkE6z9GZZpnl0nfqulcveho8W5lM95XUBLmtE-5JIc,20660
+the37lab_authlib/auth.py,sha256=Djqm_kwVwMO2VhFoGkHptejw66WRp787twS0GWRwwM4,21102
 the37lab_authlib/db.py,sha256=fTXxnfju0lmbFGPVbXpTMeDmJMeBgURVZTndyxyRyCc,2734
 the37lab_authlib/decorators.py,sha256=AEQfix31fHUZvhEZd4Ud8Zh2KBGjV6O_braiPL-BU7w,1219
 the37lab_authlib/exceptions.py,sha256=mdplK5sKNtagPAzSGq5NGsrQ4r-k03DKJBKx6myWwZc,317
 the37lab_authlib/models.py,sha256=-PlvQlHGIsSdrH0H9Cdh_vTPlltGV8G1Z1mmGQvAg9Y,3422
-the37lab_authlib-0.1.1750840415.dist-info/METADATA,sha256=VVfAVdbm_v4u0-7_f7V0mvcumU-iJZPzkUgRqgfP2c4,5645
-the37lab_authlib-0.1.1750840415.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-the37lab_authlib-0.1.1750840415.dist-info/top_level.txt,sha256=6Jmxw4UeLrhfJXgRKbXWY4OhxRSaMs0dKKhNCGWWSwc,17
-the37lab_authlib-0.1.1750840415.dist-info/RECORD,,
+the37lab_authlib-0.1.1750844514.dist-info/METADATA,sha256=sxPtONgXu1hIGi1UvtH9rPT642oK3jdIT1OTBU2Uu4w,6352
+the37lab_authlib-0.1.1750844514.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+the37lab_authlib-0.1.1750844514.dist-info/top_level.txt,sha256=6Jmxw4UeLrhfJXgRKbXWY4OhxRSaMs0dKKhNCGWWSwc,17
+the37lab_authlib-0.1.1750844514.dist-info/RECORD,,