Added support for a sequence strategy to allow persistent sequences

Author: jakob-o

smpplib/client.py

@@ -33,6 +33,18 @@
 
 logger = logging.getLogger('smpplib.client')
 
+class SimpleSequenceStrategy(object):
+    
+    def __init__(self):
+        self._sequence = 1
+        
+    @property
+    def sequence(self):
+        return self._sequence
+    
+    def next_sequence(self):
+        self._sequence += 1
+        return self._sequence
 
 class Client(object):
     """SMPP client class"""
@@ -43,16 +55,17 @@ class Client(object):
     port = None
     vendor = None
     _socket = None
-    sequence = 1
+    sequence_strategy = None
 
-    def __init__(self, host, port, timeout=5):
+    def __init__(self, host, port, timeout=5, sequence_strategy=None):
         """Initialize"""
 
         self.host = host
         self.port = int(port)
         self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         self._socket.settimeout(timeout)
         self.receiver_mode = False
+        self.sequence_strategy = sequence_strategy or SimpleSequenceStrategy()
 
     def __del__(self):
         """Disconnect when client object is destroyed"""
@@ -66,6 +79,10 @@ def __del__(self):
                     logger.warning('%s. Ignored', e)
             self.disconnect()
 
+    @property
+    def sequence(self):
+        return self.sequence_strategy.sequence
+
     def connect(self):
         """Connect to SMSC"""
 

smpplib/pdu.py

@@ -71,8 +71,7 @@ def _set_sequence(self, sequence):
 
     def _next_seq(self):
         """Return next sequence number"""
-        self._client.sequence += 1
-        return self._client.sequence
+        return self._client.sequence_strategy.next_sequence()
 
     def is_vendor(self):
         """Return True if this is a vendor PDU, False otherwise"""