pypomes-jwt 1.1.2__py3-none-any.whl → 1.1.4__py3-none-any.whl

pypomes_jwt/jwt_configuration.py

@@ -1,14 +1,12 @@
 from cryptography.hazmat.primitives import serialization
 from cryptography.hazmat.primitives.asymmetric import rsa
 from cryptography.hazmat.primitives.asymmetric.rsa import RSAPrivateKey, RSAPublicKey
-from enum import Enum
+from enum import Enum, StrEnum
 from pypomes_core import (
     APP_PREFIX,
     env_get_str, env_get_bytes, env_get_int
 )
-from pypomes_db import DbEngine, db_setup
 from secrets import token_bytes
-from sys import stderr
 
 
 # recommended: allow the encode and decode keys to be generated anew when app starts
@@ -52,34 +50,14 @@ class JwtConfig(Enum):
                                        def_value=86400)
 
 
-class JwtDbConfig(Enum):
+class JwtDbConfig(StrEnum):
     """
     Parameters for JWT databse connection.
     """
-    ENGINE: str = DbEngine(env_get_str(key=f"{APP_PREFIX}_JWT_DB_ENGINE"))
-    CLIENT: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_CLIENT")  # for Oracle, only
-    DRIVER: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_DRIVER")  # for SQLServer, only
-    NAME: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_NAME")
-    HOST: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_HOST")
-    PORT: int = env_get_int(key=f"{APP_PREFIX}_JWT_DB_PORT")
-    USER: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_USER")
-    PWD: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_PWD")
+    ENGINE: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_ENGINE")
     TABLE: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_TABLE")
     COL_ACCOUNT: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_COL_ACCOUNT")
     COL_ALGORITHM: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_COL_ALGORITHM")
     COL_DECODER: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_COL_DECODER")
     COL_KID: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_COL_KID")
     COL_TOKEN: str = env_get_str(key=f"{APP_PREFIX}_JWT_DB_COL_TOKEN")
-
-
-# define and validate the database engine
-# noinspection PyTypeChecker
-if not db_setup(engine=JwtDbConfig.ENGINE.value,
-                db_name=JwtDbConfig.NAME.value,
-                db_user=JwtDbConfig.USER.value,
-                db_pwd=JwtDbConfig.PWD.value,
-                db_host=JwtDbConfig.HOST.value,
-                db_port=JwtDbConfig.PORT.value,
-                db_client=JwtDbConfig.CLIENT.value,
-                db_driver=JwtDbConfig.DRIVER.value):
-    stderr.write("Invalid database parameters\n")

pypomes_jwt/jwt_pomes.py

@@ -5,7 +5,8 @@ from flask import Request, Response, request
 from logging import Logger
 from pypomes_core import exc_format
 from pypomes_db import (
-    db_connect, db_commit, db_rollback, db_select, db_delete
+    DbEngine, db_connect, db_commit,
+    db_rollback, db_select, db_delete
 )
 from typing import Any
 
@@ -177,14 +178,15 @@ def jwt_validate_token(errors: list[str] | None,
         elif token_kid and len(token_kid) > 1 and \
                 token_kid[0:1] in ["A", "R"] and token_kid[1:].isdigit():
             # token was likely issued locally
-            where_data: dict[str, Any] = {JwtDbConfig.COL_KID.value: int(token_kid[1:])}
+            where_data: dict[str, Any] = {JwtDbConfig.COL_KID: int(token_kid[1:])}
             if account_id:
-                where_data[JwtDbConfig.COL_ACCOUNT.value] = account_id
+                where_data[JwtDbConfig.COL_ACCOUNT] = account_id
             recs: list[tuple[str]] = db_select(errors=op_errors,
-                                               sel_stmt=f"SELECT {JwtDbConfig.COL_ALGORITHM.value}, "
-                                                        f"{JwtDbConfig.COL_DECODER.value} "
-                                                        f"FROM {JwtDbConfig.TABLE.value}",
+                                               sel_stmt=f"SELECT {JwtDbConfig.COL_ALGORITHM}, "
+                                                        f"{JwtDbConfig.COL_DECODER} "
+                                                        f"FROM {JwtDbConfig.TABLE}",
                                                where_data=where_data,
+                                               engine=DbEngine(JwtDbConfig.ENGINE),
                                                logger=logger)
             if recs:
                 token_alg = recs[0][0]
@@ -278,11 +280,12 @@ def jwt_revoke_token(errors: list[str] | None,
             op_errors.append("Invalid token")
         else:
             db_delete(errors=op_errors,
-                      delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE.value}",
+                      delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE}",
                       where_data={
-                          JwtDbConfig.COL_KID.value: int(token_kid[1:]),
-                          JwtDbConfig.COL_ACCOUNT.value: account_id
+                          JwtDbConfig.COL_KID: int(token_kid[1:]),
+                          JwtDbConfig.COL_ACCOUNT: account_id
                       },
+                      engine=DbEngine(JwtDbConfig.ENGINE),
                       logger=logger)
     if op_errors:
         if logger:
@@ -448,15 +451,17 @@ def jwt_refresh_tokens(errors: list[str] | None,
             # start the database transaction
             db_conn: Any = db_connect(errors=op_errors,
                                       autocommit=False,
+                                      engine=DbEngine(JwtDbConfig.ENGINE),
                                       logger=logger)
             if db_conn:
                 # delete current refresh token
                 db_delete(errors=op_errors,
-                          delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE.value}",
+                          delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE}",
                           where_data={
-                              JwtDbConfig.COL_KID.value: int(token_kid[1:]),
-                              JwtDbConfig.COL_ACCOUNT.value: account_id
+                              JwtDbConfig.COL_KID: int(token_kid[1:]),
+                              JwtDbConfig.COL_ACCOUNT: account_id
                           },
+                          engine=DbEngine(JwtDbConfig.ENGINE),
                           connection=db_conn,
                           committable=False,
                           logger=logger)

pypomes_jwt/jwt_registry.py

@@ -6,7 +6,7 @@ from datetime import datetime, timezone
 from logging import Logger
 from pypomes_core import str_random
 from pypomes_db import (
-    db_connect, db_commit, db_rollback,
+    DbEngine, db_connect, db_commit, db_rollback,
     db_select, db_insert, db_update, db_delete
 )
 from threading import Lock
@@ -139,8 +139,9 @@ class JwtRegistry:
 
         # remove from database
         db_delete(errors=None,
-                  delete_stmt=f"DELETE FROM {JwtDbConfig.value}",
-                  where_data={JwtDbConfig.COL_ACCOUNT.value: account_id},
+                  delete_stmt=f"DELETE FROM {JwtDbConfig}",
+                  where_data={JwtDbConfig.COL_ACCOUNT: account_id},
+                  engine=DbEngine(JwtDbConfig.ENGINE),
                   logger=logger)
         if logger:
             if account_data:
@@ -282,6 +283,7 @@ class JwtRegistry:
             # make sure to have a database connection
             curr_conn: Any = db_conn or db_connect(errors=errors,
                                                    autocommit=False,
+                                                   engine=DbEngine(JwtDbConfig.ENGINE),
                                                    logger=logger)
             if curr_conn:
                 # persist the candidate token (may raise an exception)
@@ -296,9 +298,10 @@ class JwtRegistry:
                                            headers={"kid": f"R{token_id}"})
                 # persist it
                 db_update(errors=errors,
-                          update_stmt=f"UPDATE {JwtDbConfig.TABLE.value}",
-                          update_data={JwtDbConfig.COL_TOKEN.value: refresh_token},
-                          where_data={JwtDbConfig.COL_KID.value: token_id},
+                          update_stmt=f"UPDATE {JwtDbConfig.TABLE}",
+                          update_data={JwtDbConfig.COL_TOKEN: refresh_token},
+                          where_data={JwtDbConfig.COL_KID: token_id},
+                          engine=DbEngine(JwtDbConfig.ENGINE),
                           connection=curr_conn,
                           committable=False,
                           logger=logger)
@@ -379,9 +382,10 @@ def _jwt_persist_token(account_id: str,
     # noinspection PyTypeChecker
     recs: list[tuple[int, str, str, str]] = \
         db_select(errors=errors,
-                  sel_stmt=f"SELECT {JwtDbConfig.COL_KID.value}, {JwtDbConfig.COL_TOKEN.value} "
-                           f"FROM {JwtDbConfig.TABLE.value}",
-                  where_data={JwtDbConfig.COL_ACCOUNT.value: account_id},
+                  sel_stmt=f"SELECT {JwtDbConfig.COL_KID}, {JwtDbConfig.COL_TOKEN} "
+                           f"FROM {JwtDbConfig.TABLE}",
+                  where_data={JwtDbConfig.COL_ACCOUNT: account_id},
+                  engine=DbEngine(JwtDbConfig.ENGINE),
                   connection=db_conn,
                   committable=False,
                   logger=logger)
@@ -422,8 +426,9 @@ def _jwt_persist_token(account_id: str,
     # remove expired tokens from persistence
     if expired:
         db_delete(errors=errors,
-                  delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE.value}",
-                  where_data={JwtDbConfig.COL_KID.value: expired},
+                  delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE}",
+                  where_data={JwtDbConfig.COL_KID: expired},
+                  engine=DbEngine(JwtDbConfig.ENGINE),
                   connection=db_conn,
                   committable=False,
                   logger=logger)
@@ -436,8 +441,9 @@ def _jwt_persist_token(account_id: str,
     if 0 < JwtConfig.ACCOUNT_LIMIT.value <= len(recs) - len(expired):
         # delete the oldest token to make way for the new one
         db_delete(errors=errors,
-                  delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE.value}",
-                  where_data={JwtDbConfig.COL_KID.value: oldest_id},
+                  delete_stmt=f"DELETE FROM {JwtDbConfig.TABLE}",
+                  where_data={JwtDbConfig.COL_KID: oldest_id},
+                  engine=DbEngine(JwtDbConfig.ENGINE),
                   connection=db_conn,
                   committable=False,
                   logger=logger)
@@ -448,13 +454,14 @@ def _jwt_persist_token(account_id: str,
                              f"'{account_id}' removed from storage")
     # persist token
     db_insert(errors=errors,
-              insert_stmt=f"INSERT INTO {JwtDbConfig.TABLE.value}",
+              insert_stmt=f"INSERT INTO {JwtDbConfig.TABLE}",
               insert_data={
-                  JwtDbConfig.COL_ACCOUNT.value: account_id,
-                  JwtDbConfig.COL_TOKEN.value: jwt_token,
-                  JwtDbConfig.COL_ALGORITHM.value: JwtConfig.DEFAULT_ALGORITHM.value,
-                  JwtDbConfig.COL_DECODER.value: urlsafe_b64encode(s=JwtConfig.DECODING_KEY.value).decode()
+                  JwtDbConfig.COL_ACCOUNT: account_id,
+                  JwtDbConfig.COL_TOKEN: jwt_token,
+                  JwtDbConfig.COL_ALGORITHM: JwtConfig.DEFAULT_ALGORITHM.value,
+                  JwtDbConfig.COL_DECODER: urlsafe_b64encode(s=JwtConfig.DECODING_KEY.value).decode()
               },
+              engine=DbEngine(JwtDbConfig.ENGINE),
               connection=db_conn,
               committable=False,
               logger=logger)
@@ -465,14 +472,15 @@ def _jwt_persist_token(account_id: str,
     # HAZARD: JWT_DB_COL_TOKEN's column type might prevent it for being used in a WHERE clause
     where_clause: str | None = None
     if existing_ids:
-        where_clause = f"{JwtDbConfig.COL_KID.value} NOT IN {existing_ids}"
+        where_clause = f"{JwtDbConfig.COL_KID} NOT IN {existing_ids}"
         where_clause = where_clause.replace("[", "(", 1).replace("]", ")", 1)
     reply: list[tuple[int]] = db_select(errors=errors,
-                                        sel_stmt=f"SELECT {JwtDbConfig.COL_KID.value} "
-                                                 f"FROM {JwtDbConfig.TABLE.value}",
+                                        sel_stmt=f"SELECT {JwtDbConfig.COL_KID} "
+                                                 f"FROM {JwtDbConfig.TABLE}",
                                         where_clause=where_clause,
-                                        where_data={JwtDbConfig.COL_ACCOUNT.value: account_id},
+                                        where_data={JwtDbConfig.COL_ACCOUNT: account_id},
                                         require_count=1,
+                                        engine=DbEngine(JwtDbConfig.ENGINE),
                                         connection=db_conn,
                                         committable=False,
                                         logger=logger)

{pypomes_jwt-1.1.2.dist-info → pypomes_jwt-1.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pypomes_jwt
-Version: 1.1.2
+Version: 1.1.4
 Summary: A collection of Python pomes, penyeach (JWT module)
 Project-URL: Homepage, https://github.com/TheWiseCoder/PyPomes-JWT
 Project-URL: Bug Tracker, https://github.com/TheWiseCoder/PyPomes-JWT/issues
@@ -12,5 +12,5 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.12
 Requires-Dist: cryptography>=44.0.2
 Requires-Dist: pyjwt>=2.10.1
-Requires-Dist: pypomes-core>=1.9.8
-Requires-Dist: pypomes-db>=2.0.9
+Requires-Dist: pypomes-core>=2.0.3
+Requires-Dist: pypomes-db>=2.1.1

pypomes_jwt-1.1.4.dist-info/RECORD

@@ -0,0 +1,8 @@
+pypomes_jwt/__init__.py,sha256=g4tjg7gt5_vwiHM_6-T6Ji4XYJ5py9RuzGmF2Z-qlXI,800
+pypomes_jwt/jwt_configuration.py,sha256=mtihd58_O00FuFXcNBKsabftG6UHu3Cj24i6cZXoskc,3096
+pypomes_jwt/jwt_pomes.py,sha256=oix-QLxno663wioj5W13zJahzzlQqTmtolkLqPFtKfI,23524
+pypomes_jwt/jwt_registry.py,sha256=8M4Ixhf3FQedqWS6icpJVe-7Z9KMd9qc9BjECDTZ_tU,23597
+pypomes_jwt-1.1.4.dist-info/METADATA,sha256=h4iHBGBfwx-3FFHkHFEOKvg58aFxfiXcTp6FGhfpRNI,632
+pypomes_jwt-1.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pypomes_jwt-1.1.4.dist-info/licenses/LICENSE,sha256=NdakochSXm_H_-DSL_x2JlRCkYikj3snYYvTwgR5d_c,1086
+pypomes_jwt-1.1.4.dist-info/RECORD,,

pypomes_jwt-1.1.2.dist-info/RECORD

@@ -1,8 +0,0 @@
-pypomes_jwt/__init__.py,sha256=g4tjg7gt5_vwiHM_6-T6Ji4XYJ5py9RuzGmF2Z-qlXI,800
-pypomes_jwt/jwt_configuration.py,sha256=GBkt8ISgcRYMh1FuTxUBWiKFACHHz-67mejSaPdiEsc,4172
-pypomes_jwt/jwt_pomes.py,sha256=e-E-T_hkk_FNmpsndee8AhcuUWR775SIxo4YBqjCFDA,23290
-pypomes_jwt/jwt_registry.py,sha256=7eWFEdzCMXjdxwR8nO_rqAeZ6SDf-WAEksDlfLPM_o4,23212
-pypomes_jwt-1.1.2.dist-info/METADATA,sha256=nIpMJ0AN-R9XZNsbEBVAu9ueTCOSQj3LxDgAOQrLKRU,632
-pypomes_jwt-1.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pypomes_jwt-1.1.2.dist-info/licenses/LICENSE,sha256=NdakochSXm_H_-DSL_x2JlRCkYikj3snYYvTwgR5d_c,1086
-pypomes_jwt-1.1.2.dist-info/RECORD,,