kafka-python 3.0.1__py3-none-any.whl → 3.0.2__py3-none-any.whl

kafka/net/selector.py

@@ -122,6 +122,21 @@ class Task:
             raise RuntimeError('Task exception is already set!')
         self._exc = exc
 
+    def close(self):
+        stack = self._stack
+        while stack:
+            coro, stack = stack
+            if inspect.isgenerator(coro) or inspect.iscoroutine(coro):
+                try:
+                    coro.close()
+                except ValueError:
+                    # currently-executing coroutine -- can't close it from
+                    # within itself; bail without corrupting _stack.
+                    return
+                except Exception:
+                    log.exception('Error closing coroutine for cancelled task')
+        self._stack = None
+
     @property
     def is_done(self):
         return self._stack is None
@@ -363,13 +378,29 @@ class NetworkSelector:
             result = await result
         return result
 
-    def unschedule(self, task):
+    def _remove_scheduled(self, task):
         if task.scheduled_at is not None:
-            self._scheduled.remove((task.scheduled_at, task))
+            try:
+                self._scheduled.remove((task.scheduled_at, task))
+            except ValueError:
+                pass
+            else:
+                # re-heapify to ensure heap structure is valid
+                heapq.heapify(self._scheduled)
             task.scheduled_at = None
 
+    def _retire_task(self, task):
+        if task is self._current:
+            return
+        self._pending_tasks.discard(task)
+        task.close()
+
+    def unschedule(self, task):
+        self._remove_scheduled(task)
+        self._retire_task(task)
+
     def reschedule(self, when, task):
-        self.unschedule(task)
+        self._remove_scheduled(task)
         self.call_at(when, task)
         return task
 

kafka/version.py

@@ -1 +1 @@
-__version__ = '3.0.1'
+__version__ = '3.0.2'

{kafka_python-3.0.1.dist-info → kafka_python-3.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kafka-python
-Version: 3.0.1
+Version: 3.0.2
 Summary: Pure Python client for Apache Kafka
 Author-email: Dana Powers <dana.powers@gmail.com>
 License-Expression: Apache-2.0

{kafka_python-3.0.1.dist-info → kafka_python-3.0.2.dist-info}/RECORD

@@ -6,7 +6,7 @@ kafka/errors.py,sha256=9owgtTIcttdTCD1vAF9wz5jL9G72ljpR7M-vjLKEGvY,33206
 kafka/future.py,sha256=INmMIy2-BPxJ4zlw-V_i8Tj_OG9DMbyYyUEjOQ5pn_o,5440
 kafka/structs.py,sha256=ZOFmVJRbhSXykkw9cnuv-DGYE7g2KjaUGsytSOv4xVw,3146
 kafka/util.py,sha256=hvSdRbh5-IsUbpD_BqjabjEldtNmXufZH68b_xCkNjE,5194
-kafka/version.py,sha256=Ys8n4as2Wcnr8Aq6OV-e59RKmKVeyh9Oh2nzJ2FZPFk,22
+kafka/version.py,sha256=FDUC8xOuizaRKSkyG1VOmlDwzcPJgbzFUWz58fhdUGQ,22
 kafka/admin/__init__.py,sha256=nTXg2CQo8Bydxsj80pVBzF2-jq4IhK7_gLnboeNWG60,1575
 kafka/admin/__main__.py,sha256=l5ujkHa3lBFEbKPCl7U0_pi6IIgj1IJfmB219vtcj50,69
 kafka/admin/_acls.py,sha256=QJlHguEM9t46qlABoQqTEDFQErDgG2RKwfV1NOKeKmU,13983
@@ -133,7 +133,7 @@ kafka/net/http_connect.py,sha256=j8R1yLNwhTF-nQVMEgRF88mOF5-v1Sdj7uwkBvvg5PU,513
 kafka/net/inet.py,sha256=zMo0NTCS7EY81uoNPri5ZHKcCnRVUIcCJiW6BeZWnCs,4730
 kafka/net/manager.py,sha256=nq7eKgzkazQ3sFLXQ-IOIFQ3FdyZYzfKxYxVrwYPGsc,19427
 kafka/net/metrics.py,sha256=-cdWXQtonCO7-LpovLoEcWM-_3R9sAtyQxWf_QHOAWE,7299
-kafka/net/selector.py,sha256=Vmgs69fWt3PnG5kLrDbGIX_W21M48naK2aZrvzhFYWg,24690
+kafka/net/selector.py,sha256=4_g3AtYQfmhMjzFgtFOE_TbkIIifFWW5B8zvQmyBJCI,25733
 kafka/net/socks5.py,sha256=euTwJl2avRmCJhVKyGmhZWInqgY6MNHPA1Tz2f2J1Z0,10361
 kafka/net/transport.py,sha256=hfdoQ9dMOXrMxxLJ6c4A69OGN3_FuRvDL5UuFJ1X-sA,15116
 kafka/net/wakeup_notifier.py,sha256=hmxwrrqZ8zFHSHGYaorV7vdOoKo9SF3-VoS7YGkPq_w,2926
@@ -365,9 +365,9 @@ kafka/serializer/default.py,sha256=ZKzTWlG9N4vS3QXovFLygNVrnjAcMZKS0RE4OvQ6DL4,4
 kafka/serializer/json.py,sha256=lErgU66KZGf33hofwlObYIaxLDk9gcolhy_rlcOuf8Y,469
 kafka/serializer/wrapper.py,sha256=RGXFj-PXQyL3rdMXI9ACaohRa6Kg5SHOZTRruU9jluM,485
 kafka/vendor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kafka_python-3.0.1.dist-info/licenses/LICENSE,sha256=vxnoJsqm6bKl3ZWdI1Q7Ikw_k9cOvO3vcvZNsY_1fP8,11374
-kafka_python-3.0.1.dist-info/METADATA,sha256=x94JU24iHt_fFa9M9fGvb0WymEa1jP-icNwL5bwdHfM,11590
-kafka_python-3.0.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-kafka_python-3.0.1.dist-info/entry_points.txt,sha256=LQvqZj3uM785ainO5HrXmukbjSrK-oPqrESqpvoR-As,51
-kafka_python-3.0.1.dist-info/top_level.txt,sha256=IivJz7l5WHdLNDT6RIiVAlhjQzYRwGqBBmKHZ7WjPeM,6
-kafka_python-3.0.1.dist-info/RECORD,,
+kafka_python-3.0.2.dist-info/licenses/LICENSE,sha256=vxnoJsqm6bKl3ZWdI1Q7Ikw_k9cOvO3vcvZNsY_1fP8,11374
+kafka_python-3.0.2.dist-info/METADATA,sha256=jQB9VNg9SS9r4_i52w7nofQF2k0ZWblwKCsiGc86a8k,11590
+kafka_python-3.0.2.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+kafka_python-3.0.2.dist-info/entry_points.txt,sha256=LQvqZj3uM785ainO5HrXmukbjSrK-oPqrESqpvoR-As,51
+kafka_python-3.0.2.dist-info/top_level.txt,sha256=IivJz7l5WHdLNDT6RIiVAlhjQzYRwGqBBmKHZ7WjPeM,6
+kafka_python-3.0.2.dist-info/RECORD,,