redis-simplify 2.1.2__tar.gz → 2.1.4__tar.gz

{redis_simplify-2.1.2/redis_simplify.egg-info → redis_simplify-2.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: redis-simplify
-Version: 2.1.2
+Version: 2.1.4
 Summary: Convenience wrapper for Redis - connection, JSON helpers, and automatic reconnection
 Author-email: Paulo Ricardo Tebet Lyrio <tebetpaulo91@yahoo.com>
 License: MIT
@@ -383,6 +383,7 @@ DEBUG:redis_simplify.client:Get test: hello world...
 | `rate_limit_check(key, max_requests, window_seconds)`   | Check if action is allowed        |
 | `rate_limit_remaining(key, max_requests, window_seconds)` | Get remaining requests          |
 | `rate_limit_reset(key, window_seconds)`                 | Get seconds until reset           |
+| `run_with_rate_limit(operation, rate_key, max_requests, window_seconds, *args, **kwargs)` | Execute operation with automatic rate limit |
 
 ### Distributed Lock
 | Method                                      | Description                         |
@@ -433,7 +434,7 @@ DEBUG:redis_simplify.client:Get test: hello world...
 | `ping()`                                 | Verify connectivity        |
 | `pipeline()`                             | Create a Redis pipeline    |
 | `scan(cursor=0, match=None, count=None)` | Iterate keys using SCAN    |
-| `flush_all()`                            | Remove all Redis databases |
+| `flushall()`                             | Remove all Redis databases |
 | `close()`                                | Close the connection       |
 | `set_log_level(level)`                   | Change log level at runtime|
 

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/README.md

@@ -351,6 +351,7 @@ DEBUG:redis_simplify.client:Get test: hello world...
 | `rate_limit_check(key, max_requests, window_seconds)`   | Check if action is allowed        |
 | `rate_limit_remaining(key, max_requests, window_seconds)` | Get remaining requests          |
 | `rate_limit_reset(key, window_seconds)`                 | Get seconds until reset           |
+| `run_with_rate_limit(operation, rate_key, max_requests, window_seconds, *args, **kwargs)` | Execute operation with automatic rate limit |
 
 ### Distributed Lock
 | Method                                      | Description                         |
@@ -401,7 +402,7 @@ DEBUG:redis_simplify.client:Get test: hello world...
 | `ping()`                                 | Verify connectivity        |
 | `pipeline()`                             | Create a Redis pipeline    |
 | `scan(cursor=0, match=None, count=None)` | Iterate keys using SCAN    |
-| `flush_all()`                            | Remove all Redis databases |
+| `flushall()`                             | Remove all Redis databases |
 | `close()`                                | Close the connection       |
 | `set_log_level(level)`                   | Change log level at runtime|
 

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "redis-simplify"
-version = "2.1.2"
+version = "2.1.4"
 description = "Convenience wrapper for Redis - connection, JSON helpers, and automatic reconnection"
 readme = "README.md"
 authors = [

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/redis_simplify/mixins/admin.py

@@ -17,12 +17,14 @@ class AdminMixin:
             return 0, []
         
     @recorded()
-    def flush_all(self):
+    def flushall(self):
         """Limpa tudo (CUIDADO: apenas para testes!)"""
         if not self._ensure_connection():
-            return
+            return False
         try:
             self.client.flushall()
             logger.warning("Redis flushall executed!")
+            return True
         except Exception as e:
-            logger.error(f"Error on flushall: {e}")
+            logger.error(f"Error on flushall: {e}")
+            return False

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/redis_simplify/mixins/batch.py

@@ -16,6 +16,9 @@ class BatchMixin:
         
         try:
             pipe = self.pipeline()
+            if pipe is None:
+                logger.error("Failed to create pipeline")
+                return [None] * len(keys)
             for key in keys:
                 pipe.get(key)
             return pipe.execute()
@@ -33,6 +36,9 @@ class BatchMixin:
         
         try:
             pipe = self.pipeline()
+            if pipe is None:
+                logger.error("Failed to create pipeline")
+                return False
             for key, value in items:
                 if expire_seconds:
                     pipe.setex(key, expire_seconds, value)
@@ -54,6 +60,9 @@ class BatchMixin:
         
         try:
             pipe = self.pipeline()
+            if pipe is None:
+                logger.error("Failed to create pipeline")
+                return 0
             for key in keys:
                 pipe.delete(key)
             results = pipe.execute()

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/redis_simplify/mixins/connection.py

@@ -16,7 +16,7 @@ class ConnectionMixin:
                 socket_keepalive=self.socket_keepalive,
                 health_check_interval=self.health_check_interval
             )
-                # Testa conexão
+            # Testa conexão
             self.client.ping()
             logger.info(f"RedisClient connected: {self.host}:{self.port}")
         except Exception as e:

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/redis_simplify/mixins/lock.py

@@ -21,6 +21,7 @@ class RedisLock:
         self.timeout = timeout
         self.blocking_timeout = blocking_timeout
         self.token = None
+        self.acquired = False
     
     def __enter__(self):
         self.acquire()
@@ -33,6 +34,7 @@ class RedisLock:
         while True:
             self.token = str(uuid.uuid4())
             if self.client.set(self.name, self.token, self.timeout, nx=True):
+                self.acquired = True
                 return True
             
             if self.blocking_timeout is None:
@@ -47,10 +49,11 @@ class RedisLock:
         if exc_type:
             logger.error(f"Error on lock: {exc_type.__name__}: {exc_val}")
             logger.error(f"Traceback: {''.join(traceback.format_tb(exc_tb))}")
-        
-        # Libera o lock normalmente
+        if not self.acquired:
+            return False
+
         if not self.client._ensure_connection():
-            return
+            return False
         script = """
         if redis.call("get", KEYS[1]) == ARGV[1] then
             return redis.call("del", KEYS[1])
@@ -58,4 +61,10 @@ class RedisLock:
             return 0
         end
         """
-        self.client.client.eval(script, 1, self.name, self.token)
+        try:
+            self.client.client.eval(script, 1, self.name, self.token)
+        except Exception as e:
+            logger.error(f"Error releasing lock '{self.name}': {e}")
+            return False
+        
+        return None

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/redis_simplify/mixins/pubsub.py

@@ -1,5 +1,6 @@
 import logging
 import threading
+import json
 from typing import Callable
 
 from redis_simplify.mixins.decorator_metrics import recorded
@@ -27,7 +28,6 @@ class PubSubMixin:
     
     def publish_json(self, channel: str, data: dict) -> int:
         """Publica JSON em canal"""
-        import json
         return self.publish(channel, json.dumps(data))
     
     def subscribe(self, channel: str, callback: Callable, pattern: bool = False):
@@ -43,13 +43,13 @@ class PubSubMixin:
         else:
             pubsub.subscribe(channel)
         
-        # Flag para controlar o loop
-        running = True
+        # Variável para controlar o loop
+        stop_event = threading.Event()
         
         def listener():
             try:
                 for message in pubsub.listen():
-                    if not running:
+                    if stop_event.is_set():
                         break
                     if message['type'] in ('message', 'pmessage'):
                         data = message.get('data')
@@ -69,25 +69,30 @@ class PubSubMixin:
         
         # Armazena para fechar depois
         self._pubsubs.append(pubsub)
-        self._pubsub_threads.append((thread, pubsub, lambda: running))
+        self._pubsub_threads.append((thread, pubsub, stop_event))
         
         logger.info(f"Subscribed to {channel}")
         return pubsub
     
     def close_pubsubs(self):
         """Fecha todas as subscriptions ativas"""
-        for pubsub in self._pubsubs:
-            try:
-                pubsub.close()
-            except:
-                pass
-        self._pubsubs.clear()
+        for thread, pubsub, stop_event in self._pubsub_threads:
+            stop_event.set()  # Sinaliza para parar
         
         # Aguarda threads terminarem
-        for thread, pubsub, _ in self._pubsub_threads:
+        for thread, pubsub, stop_event in self._pubsub_threads:
             try:
                 if thread.is_alive():
                     thread.join(timeout=0.5)
             except:
                 pass
+        
+        # Fecha pubsubs
+        for pubsub in self._pubsubs:
+            try:
+                pubsub.close()
+            except:
+                pass
+        
+        self._pubsubs.clear()
         self._pubsub_threads.clear()

{redis_simplify-2.1.2 → redis_simplify-2.1.4/redis_simplify.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: redis-simplify
-Version: 2.1.2
+Version: 2.1.4
 Summary: Convenience wrapper for Redis - connection, JSON helpers, and automatic reconnection
 Author-email: Paulo Ricardo Tebet Lyrio <tebetpaulo91@yahoo.com>
 License: MIT
@@ -383,6 +383,7 @@ DEBUG:redis_simplify.client:Get test: hello world...
 | `rate_limit_check(key, max_requests, window_seconds)`   | Check if action is allowed        |
 | `rate_limit_remaining(key, max_requests, window_seconds)` | Get remaining requests          |
 | `rate_limit_reset(key, window_seconds)`                 | Get seconds until reset           |
+| `run_with_rate_limit(operation, rate_key, max_requests, window_seconds, *args, **kwargs)` | Execute operation with automatic rate limit |
 
 ### Distributed Lock
 | Method                                      | Description                         |
@@ -433,7 +434,7 @@ DEBUG:redis_simplify.client:Get test: hello world...
 | `ping()`                                 | Verify connectivity        |
 | `pipeline()`                             | Create a Redis pipeline    |
 | `scan(cursor=0, match=None, count=None)` | Iterate keys using SCAN    |
-| `flush_all()`                            | Remove all Redis databases |
+| `flushall()`                             | Remove all Redis databases |
 | `close()`                                | Close the connection       |
 | `set_log_level(level)`                   | Change log level at runtime|
 

{redis_simplify-2.1.2 → redis_simplify-2.1.4}/tests/test_logging.py

@@ -136,8 +136,8 @@ class TestRedisClientLogging:
         
         caplog.clear()
         
-        # flush_all gera warning
-        client.flush_all()
+        # flushall gera warning
+        client.flushall()
         
         # Deve ter pelo menos uma mensagem WARNING
         warning_messages = [r for r in caplog.records if r.levelno == logging.WARNING]