diff options
Diffstat (limited to 'src/ogClient.py')
| -rw-r--r-- | src/ogClient.py | 18 |
1 files changed, 8 insertions, 10 deletions
diff --git a/src/ogClient.py b/src/ogClient.py index fe4e591..1457575 100644 --- a/src/ogClient.py +++ b/src/ogClient.py @@ -47,6 +47,12 @@ class ogClient: def get_state(self): return self.state + def cleanup(self): + self.data = "" + self.content_len = 0 + self.header_len = 0 + self.trailer = False + def connect(self): print('connecting...') self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) @@ -57,10 +63,7 @@ class ogClient: self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_KEEPCNT, 4) self.state = State.CONNECTING - self.data = "" - self.trailer = False - self.content_len = 0 - self.header_len = 0 + self.cleanup() try: self.sock.connect((self.ip, self.port)) @@ -121,12 +124,7 @@ class ogClient: if self.trailer and (len(self.data) >= self.content_len + self.header_len): request.parser(self.data) self.ogrest.process_request(request, self) - - # Cleanup state information from request - self.data = "" - self.content_len = 0 - self.header_len = 0 - self.trailer = False + self.cleanup() def disconnect(self): self.state = State.FORCE_DISCONNECTED |