diff options
author | Dana Powers <dana.powers@gmail.com> | 2016-07-05 22:59:03 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-07-05 22:59:03 -0700 |
commit | 75ee7fd06daf92557310cbeee1dc3ee778c42caa (patch) | |
tree | 0bb66b4d41f88e3e92d9608d6b6ab07d10bed15e | |
parent | 114bfa8767d89740599c4f10fe8871ef9971aa49 (diff) | |
download | kafka-python-75ee7fd06daf92557310cbeee1dc3ee778c42caa.tar.gz |
Avoid AttributeErrors in _unregister_cleanup (#747)
-rw-r--r-- | kafka/producer/kafka.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/kafka/producer/kafka.py b/kafka/producer/kafka.py index e1200f4..63cff73 100644 --- a/kafka/producer/kafka.py +++ b/kafka/producer/kafka.py @@ -314,7 +314,7 @@ class KafkaProducer(object): return wrapper def _unregister_cleanup(self): - if getattr(self, '_cleanup'): + if getattr(self, '_cleanup', None): if hasattr(atexit, 'unregister'): atexit.unregister(self._cleanup) # pylint: disable=no-member |