-
Notifications
You must be signed in to change notification settings - Fork 4
/
chat_client.py
111 lines (102 loc) · 4.38 KB
/
chat_client.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#coding:utf8
import cmd, socket, traceback, threading, time
import sys
def inBlack(s):
return highlight('') + "%s[30;2m%s%s[0m"%(chr(27), s, chr(27))
def inRed(s):
return highlight('') + "%s[31;2m%s%s[0m"%(chr(27), s, chr(27))
def inGreen(s):
return highlight('') + "%s[32;2m%s%s[0m"%(chr(27), s, chr(27))
def inYellow(s):
return highlight('') + "%s[33;2m%s%s[0m"%(chr(27), s, chr(27))
def inBlue(s):
return highlight('') + "%s[34;2m%s%s[0m"%(chr(27), s, chr(27))
def inPurple(s):
return highlight('') + "%s[35;2m%s%s[0m"%(chr(27), s, chr(27))
def inWhite(s):
return highlight('') + "%s[37;2m%s%s[0m"%(chr(27), s, chr(27))
def highlight(s):
return "%s[30;2m%s%s[1m"%(chr(27), s, chr(27))
class ChatClient(cmd.Cmd):
''' chat client '''
def __init__(self, host='localhost', port=10001):
cmd.Cmd.__init__(self)
self.host = host
self.port = port
self.sock = ''
self.prompt = inRed('chatClient>')
self.completekey='\n'
self.name = ''
def do_connect(self, line):#连接到server
s = socket.socket()
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.connect((self.host, self.port))
s.settimeout(0.1)#设置超时时间 0.1 秒
print inGreen('you are connected, and you can input your name, example: name Vincent')
self.sock = s
def do_name(self, line):
if not self.sock:#先得连接才能执行。
print inRed('please connect first')
line = line.strip()
self.prompt = inGreen('%s>' % line)
self.sock.send('name\t%s' % line)#发送名字
self.name = line
#print 'name', line
t = threading.Thread(target=ChatClient.continue_read, args=(self,))#启动一个子线程、
t.setDaemon(True)#守护线程在子线程没有执行完的时候程序不会被退出
t.start()
'''
个人理解：
setDaemon(True)将线程声明为守护线程，必须在start() 方法调用之前设置，如果不设置为守护线程程序会被无限挂起。
The entire Python program exits when no alive non-daemon threads are left
上面这句话的意思是：当没有存活的非守护进程时，整个python程序才会退出。
也就是说：如果主线程执行完以后,还有其他非守护线程,主线程是不会退出的
，会被无限挂起；必须将线程声明为守护线程之后，如果队列中的数据运行完了，那么整个程序想什么时候退出就退出，不用等待。
'''
def do_msg(self, line):
if not line:
print inRed('input error, msg is empty, check it and reinput')
if line:
self.sock.send('msg\t%s' % line)
def do_show(self, line):
if not self.name:
print inRed('please set your name first')
self.sock.send('show\ttmp')#发送show 和tmp
#print 'show'
def do_pm(self, line):
if not line:
print inRed('input error, msg is empty, check it and reinput')
if line:
self.sock.send('pm\t%s' % line)
@staticmethod#不用self 无法访问成员变量 且可以直接调用不用加参数
def continue_read(chatclient):#接收返回的消息并且打印出来
while 1:
try:
msg = chatclient.sock.recv(1024) #新建对象
if msg:
print inGreen(msg)
sys.stdout.write(chatclient.prompt)
sys.stdout.flush()
else:
break
except socket.timeout:
pass
except:
traceback.print_exc()
time.sleep(1)
print inRed('exit thread'), inBlue(threading.currentThread().getName())#查看当前的线程名称
def do_EOF(self, line):#退出
if self.sock:
self.sock.close()
return True
if __name__=='__main__':
info='''
help:
1.connect --- 连接到服务器
2.name [vincent] --- 用昵称登陆
3.show --- 查看当前连接人数
4.msg [message] --- 群发消息
5.pm [vincent] [message] --- 发送给某人
'''
print inPurple(info)
ChatClient().cmdloop()