转载请声明出处：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还请多多指教～
#!/usr/bin/python
#coding:utf8
#python 2.7.6 import threading
import socket
import time
import os
import sys
import signal
from readline import get_line_buffer
BUFSIZE = 1024
BACKPORT = 7789 #状态监听端口
CHATPORT = 7788 #聊天信息发送窗口
START = '>>'
INIT = '>>'
users = {}
ips = {}
#起到双向字典的作用，ip和name互相映射 #数据处理类（消息封装、分解）
class Data():
def gettime(self):
return time.strftime('%Y-%m-%d %H:%M', time.localtime(time.time()))
def getip(self):
ip = os.popen("/sbin/ifconfig | grep 'inet addr' | awk '{print $2}'").read()
ip = ip[ip.find(':')+1:ip.find('\n')]
return ip
def handlebc(self, data):
data = data[5:]
res = data.split('#opt:')
return res
def makebc(self, name, switch):
data = 'name:%s#opt:%d' % (name, switch)
return data
def handlechat(self, data):
msg = '\n' + self.gettime() + '\n' +'from '+ data + '\n'
return msg
def makechat(self, data, name):
return name + ':' + data #后台监听类
class Back(threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
self.data = Data()
self.addrb = ('255.255.255.255', BACKPORT)
self.addrl = ('', BACKPORT)
self.name = socket.gethostname()
self.ip = self.data.getip()
self.thread_stop = False
def status(self, name, switch):
if switch == 0:
status = 'offline'
elif switch == 1:
status = 'online'
#用来处理输入过程中被线程返回消息打乱的情况
if outmutex.acquire(1):
sys.stdout.write('\r'+' '*(len(get_line_buffer())+len(START))+'\r')
print '[status] '+name+' '+status
sys.stdout.write(START+get_line_buffer())
sys.stdout.flush()
outmutex.release()
def broadcast(self, switch):
bsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
bsock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
data = self.data.makebc(self.name, switch)
bsock.sendto(data, self.addrb)
bsock.close()
def response(self, addr, switch):
rsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
data = self.data.makebc(self.name, switch)
rsock.sendto(data, (addr, BACKPORT))
rsock.close()
def check(self):
if usermutex.acquire():
ips.clear()
users.clear()
usermutex.release()
self.broadcast(1)
def run(self):
lsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
lsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
lsock.bind(self.addrl)
self.broadcast(1)
while not self.thread_stop:
data, addr = lsock.recvfrom(BUFSIZE)
datalist = self.data.handlebc(data)
if usermutex.acquire(1):
if datalist[1] == '':
if ips.has_key(addr[0]):
if anoun == 1:
self.status(datalist[0], 0)
del ips[addr[0]]
del users[datalist[0]]
elif datalist[1] == '':
if anoun == 1 and datalist[0] != self.name:
self.status(datalist[0], 1)
users[datalist[0]] = addr[0]
ips[addr[0]] = datalist[0]
self.response(addr[0], 2)
elif datalist[1] == '':
if anoun == 1 and datalist[0] != self.name:
self.status(datalist[0], 1)
users[datalist[0]] = addr[0]
ips[addr[0]] = datalist[0]
usermutex.release()
lsock.close()
def stop(self):
self.broadcast(0)
self.thread_stop = True #聊天类
class Listen(threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
self.addr = ('', CHATPORT)
self.name = socket.getfqdn(socket.gethostname())
self.data = Data()
self.thread_stop = False
def ans(self, addr):#to be added 用来确认消息报的接受
return
def run(self):
lsock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
lsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
lsock.bind(self.addr)
while not self.thread_stop:
data, addr = lsock.recvfrom(BUFSIZE)
msg = self.data.handlechat(data)
if outmutex.acquire(1):
sys.stdout.write('\r'+' '*(len(get_line_buffer())+len(START))+'\r')
print msg
sys.stdout.write(START+get_line_buffer())
sys.stdout.flush()
outmutex.release()
lsock.close()
def stop(self):
self.thread_stop = True #启动入口类
class Start():
def __init__(self):
self.name = socket.getfqdn(socket.gethostname())
self.data = Data()
self.listen = Listen()
self.back = Back()
print '******* iichats ********'
print ' Written by Kevince \n'
print 'This is ' + self.name
print self.data.gettime()+'\n'
#帮助信息
def helpinfo(self):
if outmutex.acquire(1):
print "use ':' to use options"
print "\t:exit\t\t\texit iichats"
print "\t:list\t\t\tlist online users"
print "\t:quit\t\t\tquit the chat mode"
print "\t:chat [hostname]\tchatting to someone"
print "\t:set status [on|off]\tturn on/of status alarms"
outmutex.release()
def refresh(self):
if outmutex.acquire(1):
print '\n******Onlinelist******'
for key in users:
print key
print '**********************\n'
outmutex.release()
def chatting(self):
csock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
if outmutex.acquire(1):
print "use ':help' to get help information"
outmutex.release()
name = ''
address = ''
global anoun
global START
while True:
arg = raw_input(START)
if arg[0:5] == ':quit' and START != INIT:
name = ''
address = ''
START = INIT
elif arg[0] == ':' and START == INIT:
if arg[1:] == 'exit':
break
elif arg[1:5] == 'list':
self.refresh()
continue
elif arg[1:12] == 'set status ':
if arg[12:] == 'on':
anoun = 1
elif arg[12:] == 'off':
anoun = 0
continue
elif arg[1:5] == 'help':
self.helpinfo()
continue
elif arg[1:6] == 'check':
self.back.check()
print 'checking the list...'
time.sleep(3)
if outmutex.acquire(1):
outmutex.release()
self.refresh()
elif arg[1:6] == 'chat ':
name = arg[6:]
if usermutex.acquire(1):
userlist = users.keys()
usermutex.release()
if name not in userlist:
if outmutex.acquire(1):
print 'this host does not exist'
outmutex.release()
continue
address = (users.get(name), CHATPORT)
if outmutex.acquire(1):
print 'now chatting to ' + name+" ,use ':quit' to quit CHAT mode"
START = name + INIT
outmutex.release()
else:
if outmutex.acquire(1):
print "invalid input, use ':help' to get some info"
outmutex.release()
else:
if not len(address):
if outmutex.acquire(1):
print "you can CHAT to someone, or use ':help'"
outmutex.release()
continue
data = arg
msg = self.data.makechat(data, self.name)
csock.sendto(msg, address)
csock.close()
def start(self):
self.back.setDaemon(True)
self.back.start()
self.listen.setDaemon(True)
self.listen.start()
self.chatting()
self.back.stop()
self.listen.stop()
sys.exit() usermutex = threading.Lock()
outmutex = threading.Lock()
#控制status on和off的情况
anoun = 1
s = Start()
s.start()