openstack/deb-python-kafka
Code Issues Proposed changes

Avoid AttributeErrors in _unregister_cleanup (#747)

Browse Source
This commit is contained in:
Dana Powers
2016-07-05 22:59:03 -07:00
committed by GitHub
parent 114bfa8767
commit 75ee7fd06d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
kafka/producer/kafka.py
View File

@@ -314,7 +314,7 @@ class KafkaProducer(object):
return wrapper return wrapper
def _unregister_cleanup(self): def _unregister_cleanup(self):
if getattr(self, '_cleanup'): if getattr(self, '_cleanup', None):
if hasattr(atexit, 'unregister'): if hasattr(atexit, 'unregister'):
atexit.unregister(self._cleanup) # pylint: disable=no-member atexit.unregister(self._cleanup) # pylint: disable=no-member