the37lab-authlib 0.1.1751371611__py3-none-any.whl → 0.1.1756367559__py3-none-any.whl

the37lab_authlib/auth.py

@@ -11,6 +11,7 @@ import bcrypt
 import logging
 import os
 from functools import wraps
+from isodate import parse_duration
 
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
@@ -40,6 +41,10 @@ class AuthManager:
             user_override_env = os.getenv(f'{prefix}USER_OVERRIDE')
             if user_override_env:
                 self.user_override = user_override_env
+        else:
+            prefix = ''
+            
+        self.expiry_time = parse_duration(os.getenv(f'{prefix}JWT_TOKEN_EXPIRY_TIME', 'PT1H'))
         if self.user_override and (api_tokens or db_dsn):
             raise ValueError('Cannot set user_override together with api_tokens or db_dsn')
         if api_tokens and db_dsn:
@@ -487,12 +492,12 @@ class AuthManager:
     def _create_token(self, user):
         payload = {
             'sub': str(user['id']),
-            'exp': datetime.utcnow() + timedelta(hours=1),
+            'exp': datetime.utcnow() + self.expiry_time,
             'iat': datetime.utcnow()
         }
         logger.debug(f"Creating token with payload: {payload}")
         token = jwt.encode(payload, self.jwt_secret, algorithm='HS256')
-        logger.debug(f"Created token: {token}")
+        logger.info(f"Created token: {token}")
         return token
 
     def _create_refresh_token(self, user):

the37lab_authlib/db.py

@@ -1,15 +1,28 @@
 import psycopg2
 from psycopg2.extras import RealDictCursor
+from psycopg2 import pool
 from contextlib import contextmanager
 from .models import UUIDGenerator, IntegerGenerator
 
 class Database:
-    def __init__(self, dsn, id_type='uuid'):
+    def __init__(self, dsn, id_type='uuid', min_conn=1, max_conn=10):
         self.dsn = dsn
         self.id_generator = UUIDGenerator() if id_type == 'uuid' else IntegerGenerator()
         self.id_type = id_type
+        self.min_conn = min_conn
+        self.max_conn = max_conn
+        self._pool = None
+        self._init_pool()
         self._init_db()
 
+    def _init_pool(self):
+        self._pool = pool.ThreadedConnectionPool(
+            self.min_conn,
+            self.max_conn,
+            self.dsn,
+            cursor_factory=RealDictCursor
+        )
+
     def _init_db(self):
         with self.get_connection() as conn:
             with conn.cursor() as cur:
@@ -54,7 +67,7 @@ class Database:
 
     @contextmanager
     def get_connection(self):
-        conn = psycopg2.connect(self.dsn, cursor_factory=RealDictCursor)
+        conn = self._pool.getconn()
         try:
             yield conn
             conn.commit()
@@ -62,7 +75,7 @@ class Database:
             conn.rollback()
             raise
         finally:
-            conn.close()
+            self._pool.putconn(conn)
 
     @contextmanager
     def get_cursor(self):
@@ -71,4 +84,8 @@ class Database:
                 yield cur
 
     def get_id_generator(self):
-        return self.id_generator 
+        return self.id_generator
+
+    def close(self):
+        if self._pool:
+            self._pool.closeall() 

{the37lab_authlib-0.1.1751371611.dist-info → the37lab_authlib-0.1.1756367559.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: the37lab_authlib
-Version: 0.1.1751371611
+Version: 0.1.1756367559
 Summary: Python SDK for the Authlib
 Author-email: the37lab <info@the37lab.com>
 Classifier: Programming Language :: Python :: 3

the37lab_authlib-0.1.1756367559.dist-info/RECORD

@@ -0,0 +1,10 @@
+the37lab_authlib/__init__.py,sha256=cFVTWL-0YIMqwOMVy1P8mOt_bQODJp-L9bfp2QQ8CTo,132
+the37lab_authlib/auth.py,sha256=eAeyS9wBO7euMkbnRZH8mvaUzy06KbQ3qwckOtmjhkw,23536
+the37lab_authlib/db.py,sha256=cmnmykKvq6V5e-D0HGiRN4DjFBOGB-SL1HpFjR5uyCw,3162
+the37lab_authlib/decorators.py,sha256=L-gJUUwDUT2JXTptQ6XEey1LkI5RprbqzEfArWI7F8Y,1305
+the37lab_authlib/exceptions.py,sha256=mdplK5sKNtagPAzSGq5NGsrQ4r-k03DKJBKx6myWwZc,317
+the37lab_authlib/models.py,sha256=-PlvQlHGIsSdrH0H9Cdh_vTPlltGV8G1Z1mmGQvAg9Y,3422
+the37lab_authlib-0.1.1756367559.dist-info/METADATA,sha256=KJPNMX0OzIex4uWlzA96p3nr7gSB51DCxj3zKRNNbUI,8319
+the37lab_authlib-0.1.1756367559.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+the37lab_authlib-0.1.1756367559.dist-info/top_level.txt,sha256=6Jmxw4UeLrhfJXgRKbXWY4OhxRSaMs0dKKhNCGWWSwc,17
+the37lab_authlib-0.1.1756367559.dist-info/RECORD,,

the37lab_authlib-0.1.1751371611.dist-info/RECORD

@@ -1,10 +0,0 @@
-the37lab_authlib/__init__.py,sha256=cFVTWL-0YIMqwOMVy1P8mOt_bQODJp-L9bfp2QQ8CTo,132
-the37lab_authlib/auth.py,sha256=NKQ-k2QDB4ddw6QKO76RL5WeRpX2gNcpVr6VSHX2yrc,23358
-the37lab_authlib/db.py,sha256=fTXxnfju0lmbFGPVbXpTMeDmJMeBgURVZTndyxyRyCc,2734
-the37lab_authlib/decorators.py,sha256=L-gJUUwDUT2JXTptQ6XEey1LkI5RprbqzEfArWI7F8Y,1305
-the37lab_authlib/exceptions.py,sha256=mdplK5sKNtagPAzSGq5NGsrQ4r-k03DKJBKx6myWwZc,317
-the37lab_authlib/models.py,sha256=-PlvQlHGIsSdrH0H9Cdh_vTPlltGV8G1Z1mmGQvAg9Y,3422
-the37lab_authlib-0.1.1751371611.dist-info/METADATA,sha256=J3mleZGwizcERDRBfrFnBW-21tTRh4nzfmWR575z3Bk,8319
-the37lab_authlib-0.1.1751371611.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-the37lab_authlib-0.1.1751371611.dist-info/top_level.txt,sha256=6Jmxw4UeLrhfJXgRKbXWY4OhxRSaMs0dKKhNCGWWSwc,17
-the37lab_authlib-0.1.1751371611.dist-info/RECORD,,