abs-auth-rbac-core 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of abs-auth-rbac-core might be problematic. Click here for more details.

@@ -1,4 +1,4 @@
1
- from fastapi import Depends
1
+ from fastapi import Depends, Request
2
2
  from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
3
3
  import logging
4
4
  from typing import Callable, Any
@@ -21,13 +21,12 @@ def auth_middleware(
21
21
  This middleware is used for authentication of the user.
22
22
  Args:
23
23
  db_session: Callable[...,Any]: Session of the SQLAlchemy database engine
24
- Users: User table fo teh system
25
24
  jwt_secret_key: Secret key of the JWT for jwt functions
26
25
  jwt_algorithm: Algorithm used for JWT
27
26
 
28
27
  Returns:
29
28
  """
30
- def get_auth(token: HTTPAuthorizationCredentials = Depends(security)):
29
+ async def get_auth(request: Request, token: HTTPAuthorizationCredentials = Depends(security)):
31
30
  jwt_functions = JWTFunctions(secret_key=jwt_secret_key,algorithm=jwt_algorithm)
32
31
  try:
33
32
  if not token or not token.credentials:
@@ -42,7 +41,9 @@ def auth_middleware(
42
41
  logger.error(f"Authentication failed: User with id {uuid} not found")
43
42
  raise UnauthorizedError(detail="Authentication failed")
44
43
 
45
- return user # Attach user for use in route
44
+ # Attach user to request state
45
+ request.state.user = user
46
+ return user # Still return user for backward compatibility
46
47
 
47
48
  except Exception as e:
48
49
  logger.error(f"Authentication error: {str(e)}", exc_info=True)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: abs-auth-rbac-core
3
- Version: 0.1.1
3
+ Version: 0.1.3
4
4
  Summary: RBAC and Auth core utilities including JWT token management.
5
5
  License: MIT
6
6
  Author: AutoBridgeSystems
@@ -2,7 +2,7 @@ abs_auth_rbac_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
2
2
  abs_auth_rbac_core/auth/__init__.py,sha256=Pvetd98VD9jsok1F9e82yS9kLZMFqTpbN51hkxZQHzw,67
3
3
  abs_auth_rbac_core/auth/auth_functions.py,sha256=fhfKRhtpE_J9MHu2jSsIA-cy77A3wCAQKbGGZeh4pe4,947
4
4
  abs_auth_rbac_core/auth/jwt_functions.py,sha256=9vhjWrxXdE8fVQ4FGrPj9y6PoSEsaeFohPhgI-3hToI,4111
5
- abs_auth_rbac_core/auth/middleware.py,sha256=1J0yE2aEgfMzTgkSz1Tz_8xs0Nl095k5S1wbygShkl0,1806
5
+ abs_auth_rbac_core/auth/middleware.py,sha256=c1HY2UHNAflo7cqSPE-_zYP27T-CGhaACO3cOWED8Lc,1896
6
6
  abs_auth_rbac_core/models/__init__.py,sha256=HHnXZ1wIq6Us3MP9YywjHVeCViXD_zyMWl8mBglqFBc,231
7
7
  abs_auth_rbac_core/models/base_model.py,sha256=AaWObslm8sTetv4H1Ia_gPpi_75uF5z1o7Et9WAvstU,612
8
8
  abs_auth_rbac_core/models/gov_casbin_rule.py,sha256=uWkzOgtG0_8CXR4YLBz_JINA8apcB1D4aWiCgSkuTCs,789
@@ -14,12 +14,11 @@ abs_auth_rbac_core/models/seeder/permission_seeder.py,sha256=j-aUy8uLHnUWpMmw1Dq
14
14
  abs_auth_rbac_core/models/user.py,sha256=t_ardJOsfBiyedPg6Z4WywLiZRTPmAGYkMealZcEqJc,833
15
15
  abs_auth_rbac_core/models/user_role.py,sha256=20pqmtJPzlUrI9ulHGouk8XlFgrGG7I6ikctb8sMUGs,706
16
16
  abs_auth_rbac_core/rbac/__init__.py,sha256=oYjtpmfrkEbwWCBAWuRoU1fM4fCpBxkF_lwQrelK1As,79
17
- abs_auth_rbac_core/rbac/database.py,sha256=4lxs3gDIMq4pnor8oHN51UImp9Ad-6_HiIdPiLI-LGc,1475
18
17
  abs_auth_rbac_core/rbac/decorator.py,sha256=y1TJqVQLWJeMqj7PxYqlAQZ5sOj1hZv9AvChGspl8A0,1794
19
18
  abs_auth_rbac_core/rbac/policy.conf,sha256=wghhhKxgZH0rPhh1QFrIpq9nevJT3s7OxxvXiU3zzuI,305
20
19
  abs_auth_rbac_core/rbac/service.py,sha256=C0LTqnyecePrLSh2bAAhO9xcwZ_UAfi4CREczgcQNWc,25206
21
20
  abs_auth_rbac_core/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
21
  abs_auth_rbac_core/util/permission_constants.py,sha256=JFavEKkAJ5mtltttCwRTTtpRlu_mJiVPB_MwsP-bIAg,65337
23
- abs_auth_rbac_core-0.1.1.dist-info/METADATA,sha256=-rvnH3oBo96pOgFsJ-KJkQZz_2lO5mRAvBQNFU_dSPw,6586
24
- abs_auth_rbac_core-0.1.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
25
- abs_auth_rbac_core-0.1.1.dist-info/RECORD,,
22
+ abs_auth_rbac_core-0.1.3.dist-info/METADATA,sha256=TI0c8jITxxaCb6GFkrgPbYVmTPjr3T-97BKRavFOghw,6586
23
+ abs_auth_rbac_core-0.1.3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
24
+ abs_auth_rbac_core-0.1.3.dist-info/RECORD,,
@@ -1,52 +0,0 @@
1
- from contextlib import AbstractContextManager, contextmanager
2
- from typing import Any, Generator
3
- from loguru import logger
4
-
5
- from sqlalchemy import create_engine, orm
6
- from sqlalchemy.orm import Session
7
- from abs_repository_core.models import BaseModel
8
-
9
-
10
- class Database:
11
- def __init__(self, db_url: str) -> None:
12
- """
13
- Initialize the database engine and session factory
14
- """
15
- self._engine = create_engine(
16
- db_url,
17
- echo=False,
18
- echo_pool=False,
19
- pool_pre_ping=True,
20
- pool_recycle=3600,
21
- query_cache_size=0,
22
- )
23
- self._session_factory = orm.scoped_session(
24
- orm.sessionmaker(
25
- autocommit=False,
26
- autoflush=False,
27
- bind=self._engine,
28
- ),
29
- )
30
-
31
- def create_database(self) -> None:
32
- """
33
- Create all the tables in the database
34
- """
35
- BaseModel.metadata.create_all(self._engine)
36
-
37
- @contextmanager
38
- def session(self) -> Generator[Any, Any, AbstractContextManager[Session]]:
39
- """
40
- Provides a database session for the request
41
- """
42
- session: Session = self._session_factory()
43
- try:
44
- yield session
45
- except Exception as e:
46
- session.rollback()
47
- import traceback
48
-
49
- logger.error(f"Exception: {e}\n{traceback.format_exc()}")
50
- raise e
51
- finally:
52
- session.close()