Merge pull request #36 from claroty/bugfix/abuddi/add-asyncpg-as-cpython

Add asyncpg as cpython

jwthenticator/client.py

@@ -129,7 +129,7 @@ async def authenticate(self) -> None:
         :return: None or raises exception if fails.
         """
         if self._key is None:
-            raise Exception("Missing key")
+            raise TypeError("Missing key")
 
         url = urljoin(self.jwthenticator_server, "authenticate")
         request = schemas.AuthRequest(self._key, self.identifier)

jwthenticator/schemas.py

@@ -37,7 +37,7 @@ class KeyData:
     created: datetime
     expires_at: datetime
     key_hash: str
-    key: Optional[str] = field(default=None, repr=False, metadata=dict(load_only=True))
+    key: Optional[str] = field(default=None, repr=False, metadata={"load_only": True})
 
     async def is_valid(self) -> bool:
         return self.expires_at > datetime.utcnow()

jwthenticator/tokens.py

@@ -28,7 +28,7 @@ def __init__(self, public_key: str, private_key: Optional[str] = None, algorithm
         """
         # This is done to avoid an "exploit" where algorithm=none leaving the jwt tokens unsecure.
         if not algorithm or algorithm.lower() == "none":
-            raise Exception("Algorithm can't be empty!")
+            raise ValueError("Algorithm can't be empty!")
 
         self.public_key = public_key
         self.private_key = private_key
@@ -49,7 +49,7 @@ async def create_access_token(self, identifier: UUID) -> str:
         :return: The new JWT token string
         """
         if self.private_key is None:
-            raise Exception("Private key required for JWT token creation")
+            raise RuntimeError("Private key required for JWT token creation")
         utc_now = datetime.utcnow()
         payload = JWTPayloadData(
             token_id=uuid4(),
@@ -84,7 +84,7 @@ async def create_refresh_token(self, key_id: int, expires_at: Optional[datetime]
         if expires_at is None:
             expires_at = expires_at = datetime.utcnow() + timedelta(seconds=REFRESH_TOKEN_EXPIRY)
         if expires_at <= datetime.utcnow():
-            raise Exception("Refresh token can't be created in the past")
+            raise RuntimeError("Refresh token can't be created in the past")
 
         refresh_token_str = sha512(uuid4().bytes).hexdigest()
         async with self.async_session_factory() as session: