fix AttributeError when executing close

When self.device is already None an AttributeError will occur.
This is relevant to #189.
This commit is contained in:
Patrick Kanzler 2017-01-30 16:12:37 +01:00
parent 1038844567
commit e92f00cdf3
No known key found for this signature in database
GPG Key ID: F07F07153306FCEF
1 changed files with 6 additions and 4 deletions

View File

@ -211,8 +211,9 @@ class Network(Escpos):
def close(self):
""" Close TCP connection """
self.device.shutdown(socket.SHUT_RDWR)
self.device.close()
if self.device is not None:
self.device.shutdown(socket.SHUT_RDWR)
self.device.close()
class File(Escpos):
@ -263,8 +264,9 @@ class File(Escpos):
def close(self):
""" Close system file """
self.device.flush()
self.device.close()
if self.device is not None:
self.device.flush()
self.device.close()
class Dummy(Escpos):