Add timeout to Client's properties

Lynesth · web-flow · commit babb0eb7ee32 · 2019-09-07T19:14:05.000+11:00
diff --git a/smpplib/client.py b/smpplib/client.py
@@ -68,7 +68,7 @@ def __init__(self, host, port, timeout=5, sequence_generator=None):
         self.host = host
         self.port = int(port)
         self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        self._socket.settimeout(timeout)
+        self.timeout = timeout
         if sequence_generator is None:
             sequence_generator = SimpleSequenceGenerator()
         self.sequence_generator = sequence_generator
@@ -106,6 +106,7 @@ def connect(self):
         try:
             if self._socket is None:
                 self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                self._socket.settimeout(self.timeout)
             self._socket.connect((self.host, self.port))
             self.state = consts.SMPP_CLIENT_STATE_OPEN
         except socket.error:
