Token add

Canguru/CanguruChat.py

@@ -66,6 +66,12 @@ class Protocol():
     		self.priority = raw_input()		
     	self.priority = str(self.priority)
 
+    def setType (self):
+        if (self.msg == 'token'):
+            self.type = str(0)
+        else:
+            self.type = str(1)
+
     #set message 
     def setMessage(self):
     	sys.stdout.write('Mensagem: ')
@@ -106,15 +112,19 @@ def client ():
         protocolMessage.setDestiny()
         protocolMessage.setPriority()
         protocolMessage.setMessage()
+        protocolMessage.setType()
         protocolMessage.setSizeof()
         protocolMessage.setCRC32()
         msg = protocolMessage.getEmpacotar()
         while protocolMessage.msg <> 'exit':
+            if (protocolMessage.type == 0):
+                print 'Vou mandar o token'
     	    udp.sendto (msg, dest)
             protocolMessage = Protocol()
             protocolMessage.setDestiny()
             protocolMessage.setPriority()
             protocolMessage.setMessage()
+            protocolMessage.setType()
             protocolMessage.setSizeof()
             protocolMessage.setCRC32()
             msg = protocolMessage.getEmpacotar()
@@ -132,6 +142,8 @@ def server ():
     while True:
         msg, cliente = udp.recvfrom(1024)
         protocolDescompactMessage.setDesempacota(msg)
+       	if (protocolDescompactMessage.type == '0'):
+            print 'RECEBI O TOKEN'
         if protocolDescompactMessage.crc == protocolDescompactMessage.getCRC32():
        		print msg
        		print '\n' + protocolDescompactMessage.origin + ' Escreveu: ' + protocolDescompactMessage.msg
@@ -142,9 +154,9 @@ def server ():
 #------------------MAIN-----------------#
 
 if len(sys.argv) > 1:
-    AUX = RECV_PORT
-    RECV_PORT = SEND_PORT
-    SEND_PORT = AUX
+    TOKEN = 1
+    print 'COMECEI COM O TOKEN'
+
 print "Ip da proxima maquina"
 HOST = raw_input()
 threadServer = threading.Thread(target=server)