【Python】iichats —— 命令行下的局域网聊天程序

转载请声明出处：http://www.cnblogs.com/kevince/p/3941728.html   ——By Kevince

ii系列工具第三弹，命令行下的局域网聊天程序

原理：

程序启动时向全网(255.255.255.255)BACKPORT端口广播自己的主机名以及状态（上线）。

如果接受收到的上线状态，则将其加入通信列表，同时返还一个数据包，使自己也将对面加入其通信列表。

程序退出时向全网广播自己的下线状态，如果收到该下线状态则将其从自己的通信列表中删除

为了防止在输入过程中被新输出的消息打断，可以用readlines中的get_line_buffer函数获取缓冲区内的字符并储存，清空原先行，输出新结果，并在下面输出刚刚输入的内容（可用curses改进，to be continued...）

缺陷：

使用UDP协议，未添加消息到达确认机制；

跨平台支持需要修改代码(readlines只支持linux, windows下要用pyreadline,MAC OS要用edlitline来代替)

实用性不强，功能单一，学习程序

未能实现GUI图形界面的开发（目前还木有学会……）

刚学Python没多久 且开发仓促，有Bug还请多多指教～

 

  1 #!/usr/bin/python

  2 #coding:utf8

  3 #python 2.7.6

  4 

  5 import threading

  6 import socket

  7 import time

  8 import os

  9 import sys

 10 import signal

 11 from readline import get_line_buffer

 12 BUFSIZE = 1024

 13 BACKPORT = 7789     #状态监听端口

 14 CHATPORT = 7788     #聊天信息发送窗口

 15 START = '>>'

 16 INIT = '>>'

 17 users = {}

 18 ips = {} 

 19 #起到双向字典的作用，ip和name互相映射

 20 

 21 #数据处理类（消息封装、分解）

 22 class Data(): 

 23     def gettime(self):

 24         return time.strftime('%Y-%m-%d %H:%M', time.localtime(time.time()))

 25     def getip(self):

 26         ip = os.popen("/sbin/ifconfig | grep 'inet addr' | awk '{print $2}'").read()

 27         ip = ip[ip.find(':')+1:ip.find('\n')]

 28         return ip

 29     def handlebc(self, data):

 30         data = data[5:]

 31         res = data.split('#opt:')

 32         return res

 33     def makebc(self, name, switch):

 34         data = 'name:%s#opt:%d' % (name, switch)

 35         return data

 36     def handlechat(self, data):

 37         msg = '\n' + self.gettime() + '\n' +'from '+ data + '\n'

 38         return msg

 39     def makechat(self, data, name):

 40         return name + ':' + data

 41 

 42 #后台监听类

 43 class Back(threading.Thread): 

 44     def __init__(self):

 45         threading.Thread.__init__(self)

 46         self.data = Data()

 47         self.addrb = ('255.255.255.255', BACKPORT)

 48         self.addrl = ('', BACKPORT)

 49         self.name = socket.gethostname()

 50         self.ip = self.data.getip()

 51         self.thread_stop = False

 52     def status(self, name, switch):

 53         if switch == 0:

 54             status = 'offline'

 55         elif switch == 1:

 56             status = 'online'

 57         #用来处理输入过程中被线程返回消息打乱的情况

 58         if outmutex.acquire(1):

 59             sys.stdout.write('\r'+' '*(len(get_line_buffer())+len(START))+'\r')

 60             print '[status] '+name+' '+status

 61             sys.stdout.write(START+get_line_buffer())

 62             sys.stdout.flush()

 63             outmutex.release()

 64     def broadcast(self, switch):

 65         bsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

 66         bsock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)

 67         data = self.data.makebc(self.name, switch)

 68         bsock.sendto(data, self.addrb)

 69         bsock.close()

 70     def response(self, addr, switch):

 71         rsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

 72         data = self.data.makebc(self.name, switch)

 73         rsock.sendto(data, (addr, BACKPORT))

 74         rsock.close()

 75     def check(self):

 76         if usermutex.acquire():

 77             ips.clear()

 78             users.clear()

 79             usermutex.release()

 80         self.broadcast(1)

 81     def    run(self):

 82         lsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

 83         lsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

 84         lsock.bind(self.addrl)

 85         self.broadcast(1)

 86         while not self.thread_stop:

 87             data, addr = lsock.recvfrom(BUFSIZE)

 88             datalist = self.data.handlebc(data)

 89             if usermutex.acquire(1):

 90                 if datalist[1] == '0':

 91                     if ips.has_key(addr[0]):

 92                         if anoun == 1:

 93                             self.status(datalist[0], 0)

 94                         del ips[addr[0]]

 95                         del users[datalist[0]]

 96                 elif datalist[1] == '1':

 97                     if anoun == 1 and datalist[0] != self.name:

 98                         self.status(datalist[0], 1)

 99                     users[datalist[0]] = addr[0]

100                     ips[addr[0]] = datalist[0]

101                     self.response(addr[0], 2)

102                 elif datalist[1] == '2':

103                     if anoun == 1 and datalist[0] != self.name:

104                         self.status(datalist[0], 1)

105                     users[datalist[0]] = addr[0]

106                     ips[addr[0]] = datalist[0]

107                 usermutex.release()

108         lsock.close()

109     def stop(self):

110         self.broadcast(0)

111         self.thread_stop = True

112 

113 #聊天类

114 class Listen(threading.Thread):

115     def __init__(self):

116         threading.Thread.__init__(self)

117         self.addr = ('', CHATPORT)

118         self.name = socket.getfqdn(socket.gethostname())

119         self.data = Data()

120         self.thread_stop = False

121     def ans(self, addr):#to be added 用来确认消息报的接受

122         return

123     def run(self):

124         lsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

125         lsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

126         lsock.bind(self.addr)

127         while not self.thread_stop:

128             data, addr = lsock.recvfrom(BUFSIZE)

129             msg = self.data.handlechat(data)

130             if outmutex.acquire(1):

131                 sys.stdout.write('\r'+' '*(len(get_line_buffer())+len(START))+'\r')

132                 print msg

133                 sys.stdout.write(START+get_line_buffer())

134                 sys.stdout.flush()

135                 outmutex.release()

136         lsock.close()

137     def stop(self):

138         self.thread_stop = True

139 

140 #启动入口类

141 class Start(): 

142     def __init__(self):

143         self.name = socket.getfqdn(socket.gethostname())

144         self.data = Data()

145         self.listen = Listen()

146         self.back = Back()

147         print '*******   iichats   ********'

148         print '     Written by Kevince     \n'

149         print 'This is ' + self.name

150         print self.data.gettime()+'\n'

151     #帮助信息

152     def helpinfo(self):     

153         if outmutex.acquire(1):

154             print "use ':' to use options"

155             print "\t:exit\t\t\texit iichats"

156             print "\t:list\t\t\tlist online users"

157             print "\t:quit\t\t\tquit the chat mode"

158             print "\t:chat [hostname]\tchatting to someone"

159             print "\t:set status [on|off]\tturn on/of status alarms"

160             outmutex.release()

161     def refresh(self):

162         if outmutex.acquire(1):

163             print '\n******Onlinelist******'

164             for key in users:

165                 print key

166             print '**********************\n'

167             outmutex.release()

168     def chatting(self):

169         csock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

170         if outmutex.acquire(1):

171             print "use ':help' to get help information"

172             outmutex.release()

173         name = ''

174         address = ''

175         global anoun

176         global START

177         while True:

178             arg = raw_input(START)

179             if arg[0:5] == ':quit' and START != INIT:

180                 name = ''

181                 address = ''

182                 START = INIT

183             elif arg[0] == ':' and START == INIT:

184                 if arg[1:] == 'exit':

185                     break

186                 elif arg[1:5] == 'list':

187                     self.refresh()

188                     continue

189                 elif arg[1:12] == 'set status ':

190                     if arg[12:] == 'on':

191                         anoun = 1

192                     elif arg[12:] == 'off':

193                         anoun = 0

194                     continue

195                 elif arg[1:5] == 'help':

196                     self.helpinfo()

197                     continue

198                 elif arg[1:6] == 'check':

199                     self.back.check()

200                     print 'checking the list...'

201                     time.sleep(3)

202                     if outmutex.acquire(1):

203                         outmutex.release()

204                     self.refresh()

205                 elif arg[1:6] == 'chat ':

206                     name = arg[6:]

207                     if usermutex.acquire(1):

208                         userlist = users.keys()

209                         usermutex.release()

210                     if name not in userlist:

211                         if outmutex.acquire(1):

212                             print 'this host does not exist'

213                             outmutex.release()

214                         continue

215                     address = (users.get(name), CHATPORT)

216                     if outmutex.acquire(1):

217                         print 'now chatting to ' + name+" ,use ':quit' to quit CHAT mode"

218                         START = name + INIT

219                         outmutex.release()

220                 else:

221                     if outmutex.acquire(1):

222                         print "invalid input, use ':help' to get some info"

223                         outmutex.release()

224             else:

225                 if not len(address):

226                     if outmutex.acquire(1):

227                         print "you can CHAT to someone, or use ':help'"

228                         outmutex.release()

229                     continue

230                 data = arg

231                 msg = self.data.makechat(data, self.name)

232                 csock.sendto(msg, address)

233         csock.close()

234     def start(self):

235         self.back.setDaemon(True)

236         self.back.start()

237         self.listen.setDaemon(True)

238         self.listen.start()

239         self.chatting()

240         self.back.stop()

241         self.listen.stop()

242         sys.exit()

243 

244 usermutex = threading.Lock()

245 outmutex = threading.Lock()

246 #控制status on和off的情况

247 anoun = 1

248 s = Start()

249 s.start()