Python聊天室

背景

这是一篇水文，同时也是更换markdown后的第一篇，主要是为了测试markdown的情况。

服务器程序

#!/usr/bin/env python

# -*- coding:utf-8 -*-

"""

聊天室服务器程序v1.0.1

作者：挖洞的土拨鼠

联系：

GitHub：https://github.com/cisp

Blog：http://www.cnblogs.com/KevinGeorge/

"""

#引入依赖包、库文件

import sys

import time

import socket

import select

import logging

import datetime

#设置全局配置

reload(sys)

sys.setdefaultencoding("utf-8")

logging.basicConfig(filename="./talkroom.log",level=logging.INFO,filemode='a',format='%(asctime)s-%(levelname)s:%(message)s')

#定义全局变量

test_port = 38080

test_addr = "127.0.0.1"

#定义全局函数

def now():

return str(datetime.datetime.now())

#定义聊天室类

class TalkRoom(object):

"""创建聊天室类"""

def __init__(self,ipaddress,port,max):

"""

构造函数：

初始化聊天人字典结构

初始化服务器套接字

初始化轮询队列

"""

self.__register_name_dictionary = {}

self.__server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

self.__server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

self.__server.bind((ipaddress,port))

self.__server.listen(max)#设置聊天室最大限制人数

self.__readable_socket_list = [self.__server]

self.__writeable_socket_list = []

self.__error_socket_list = [self.__server]

def broadcast(self,sockfd,msg):

for sock in self.__register_name_dictionary:

if sock != sockfd:

try:

sock.send("\n"+msg)

except Exception,reason:

logging.info(reason)

sock.close()

self.broadcast(sock,"[%s %s]已经退出聊天室\n"%(self.__register_name_dictionary[sock][0],now()))

self.__register_name_dictionary.pop(sock)

def running(self):

while True:

rs,ws,es = select.select(self.__readable_socket_list,self.__writeable_socket_list,self.__error_socket_list)

for sockfd in rs:

if sockfd == self.__server:

client,addr = sockfd.accept()

self.__register_name_dictionary[client] = (None,addr)

client.send("请输入您的昵称:")

name = client.recv(1024)

self.__register_name_dictionary[client] = (name,addr)

self.broadcast(client,"[%s %s]刚刚进入聊天室\n"%(name,now()))

self.__readable_socket_list.append(client)

self.__error_socket_list.append(client)

continue

else:

try:

data = sockfd.recv(4096)

if data:

self.broadcast(sockfd,"[%s %s]"%(self.__register_name_dictionary[sockfd][0],now())+data)

except Exception,reason:

logging.info(reason)

sockfd.close()

self.broadcast(sockfd,"[%s %s]已经退出聊天室\n"%(self.__register_name_dictionary[sockfd][0],now()))

self.__register_name_dictionary.pop(sockfd)

self.__readable_socket_list.remove(sockfd)

self.__error_socket_list.remove(sockfd)

if __name__ == "__main__":

room = TalkRoom(test_addr,test_port,5)

room.running()

客户端程序

#!/usr/bin/env python

# -*- coding:utf-8 -*-

"""

聊天室客户端程序v1.0.1

作者：挖洞的土拨鼠

联系：

GitHub：https://github.com/cisp

Blog：http://www.cnblogs.com/KevinGeorge/

"""

#引入依赖包、库文件

import sys

import time

import select

import socket

import logging

import datetime

#设置全局配置

reload(sys)

sys.setdefaultencoding("utf-8")

logging.basicConfig(filename="./talkclient.log",level=logging.INFO,filemode='a',format='%(asctime)s-%(levelname)s:%(message)s')

#定义全局变量

test_port = 38080

test_addr = "127.0.0.1"

#定义全局函数

def now():

return str(datetime.datetime.now())

def inputflag() :

sys.stdout.write('[MySelf %s]'%now())

sys.stdout.flush()

#定义客户端类

class TalkClient(object):

def __init__(self,ipaddress,port):

self.__client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

self.__readable_socket_list = [sys.stdin,self.__client]

self.__writeable_socket_list = []

self.__error_socket_list = [self.__client]

try:

self.__client.connect(("127.0.0.1", 38080))

except Exception,reason:

logging.info(reason)

exit(0)

print "[MySelf %s]刚刚进入聊天室"%now()

banner = self.__client.recv(1024)

print banner

name =raw_input()

self.__client.send(name)

inputflag()

def running(self):

while True:

rs,ws,es = select.select(self.__readable_socket_list,self.__writeable_socket_list,self.__error_socket_list)

for sockfd in rs:

if sockfd == self.__client:

data = self.__client.recv(4096)

if data:

sys.stdout.write(data)

inputflag()

else:

exit(0)

else:

message = sys.stdin.readline()

self.__client.send(message)

inputflag()

#主程序入口

if __name__ == "__main__":

client = TalkClient(test_addr,test_port)

client.running()

效果如下图