Modify Client state to use enum

During our connections, we are using states to control the Client Socket. We
defined using global variables. In case that we modify this global variable,
we need to change it in serveral parts of the code.

Using enums and declaring a new class, we can redefine the values or create new
states without changing the same code in differents python files.

1 files changed, 8 insertions, 6 deletions

diff --git a/src/ogClient.py b/src/ogClient.py
index 8c965fa..296c03c 100644
--- a/src/ogClient.py
+++ b/src/ogClient.py
@@ -5,9 +5,11 @@ import time
 import httplib

 from mimetools import Message

 from StringIO import StringIO

+from enum import Enum

 

-CONNECTING = 0

-RECEIVING = 1

+class State(Enum):

+	CONNECTING = 0

+	RECEIVING = 1

 

 class ogClient:

 	def __init__(self, ip, port):

@@ -24,7 +26,7 @@ class ogClient:
 		print "connecting"

 		self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

 		self.sock.setblocking(0)

-		self.state = CONNECTING

+		self.state = State.CONNECTING

 		self.data = ""

 		self.trailer = False

 		self.content_len = 0

@@ -45,10 +47,10 @@ class ogClient:
 		except socket.error, err:

 			if err.errno == errno.EISCONN:

 				print "connected"

-				self.state = RECEIVING

+				self.state = State.RECEIVING

 			else:

 				print "connection refused, retrying..."

-				self.state = CONNECTING

+				self.state = State.CONNECTING

 				self.sock.close()

 				self.connect()

 

@@ -60,7 +62,7 @@ class ogClient:
 			print "Error3 " + str(err)

 

 		if len(data) == 0:

-			self.state = CONNECTING

+			self.state = State.CONNECTING

 			self.sock.close()

 			self.connect()