3
3
import sys
4
4
import threading
5
5
6
+
6
7
class Client ():
7
8
8
9
def __init__ (self ):
9
10
self ._socket = None
10
11
self ._connected = False
11
12
self ._nickname = 'anonymous'
12
13
13
- def start (self , server_ip , server_port , nickname = None ):
14
- if nickname != None :
14
+ def start (self , server_ip , server_port , nickname = None ):
15
+ if nickname is not None :
15
16
self ._nickname = nickname
16
17
print ('Hello friend ' + self ._nickname )
17
18
dest = (server_ip , server_port )
18
19
try :
19
20
self ._socket = socket .create_connection (dest )
20
21
except :
21
22
print ('error opening socket' )
22
- return ;
23
+ return
23
24
self .__handle_messages ()
24
25
25
26
def __start_protocol (self ):
@@ -43,20 +44,19 @@ def __handle_messages(self):
43
44
socket_list = [sys .stdin , self ._socket ]
44
45
self .__prompt ()
45
46
while True :
46
- ready_to_read ,ready_to_write , in_error = select .select (socket_list , [], [])
47
+ ready_to_read , _ , _ = select .select (socket_list , [], [])
47
48
for sock in ready_to_read :
48
49
if sock == self ._socket :
49
50
# incoming message from remote server
50
51
data = sock .recv (4096 )
51
- if not data :
52
+ if not data :
52
53
print ('\n Disconnected from chat server' )
53
54
sys .exit ()
54
- else :
55
- #print data
55
+ else :
56
+ # print data
56
57
sys .stdout .write (data .decode ('utf-8' ))
57
58
self .__prompt ()
58
-
59
- else :
59
+ else :
60
60
# user entered a message
61
61
msg = sys .stdin .readline ()
62
62
msg = '[' + self ._nickname + '] ' + msg
@@ -68,6 +68,6 @@ def __prompt():
68
68
sys .stdout .write ('[Me] ' )
69
69
sys .stdout .flush ()
70
70
71
- def __gpg_encrypt (self ,msg ):
71
+ def __gpg_encrypt (self , msg ):
72
72
return msg
73
73
sys .stdout .flush ()
0 commit comments