Python socket编程（阻塞） --基于SocketServer

SocketServer模块是Python对socket常规通信的一个经过封装的模块，使用简单，基于面向对象的设计模式，但功能有限，可用于快速开发。

Tips:

默认端口：6767

默认本地ip：127.0.0.1

代码样例：

1. 服务端：

#!/usr/bin/env python
# encoding: utf-8

from SocketServer import (TCPServer as TCP,StreamRequestHandler as SRH)
import sys
import os
from time import ctime

reload(sys)
sys.setdefaultencoding("utf-8")

conn_flag = False #连接提示标识，规定只显示一次

class myWebServer(SRH):
    def handle(self):
        global conn_flag
        if conn_flag == False:
            print "[*]Connect from: ",self.client_address
            conn_flag = True
        buf = self.rfile.readline()
        self.wfile.write("[%s] %s" % (ctime(),buf))
        print "[%s] %s" % (ctime(),buf)
def main():
    HOST = "127.0.0.1"
    PORT = 6767
    Server = TCP((HOST,PORT), myWebServer)
    print "Waiting for the Connection..."
    try:
        Server.serve_forever()
    except KeyboardInterrupt:
        sys.exit(0)
    except Exception as e:
        print "[-]Client may be offLine"
        sys.exit(-1)

if __name__ == '__main__':
    main()

2. 客户端

#!/usr/bin/env python
# encoding: utf-8
import socket
import os
import sys
import time
import traceback

reload(sys)
sys.setdefaultencoding("utf-8")

def SocketInit():
    try:
        s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    except Exception as e:
        sys.exit(-1)
    return s

def Conn(s,host,port):
    while 1:
        try:
            s.connect((host,port))
        except Exception as e:
            time.sleep(2)
        else:
            break

def main():
    host = "127.0.0.1"
    port = 6767
    try:
        while 1:
            sC = SocketInit()
            Conn(sC, host, port)
            Str_buffer = raw_input(host+":"+str(port)+">")
            Str_buffer += "\r\n"
            try:
                sC.send(Str_buffer)
            except Exception as e:
                print "[-]Can not send Data"
            else:
                recv_buffer = sC.recv(1024)
                if not recv_buffer:
                    print "[-]Server is Off"
                else:
                    print recv_buffer
    except KeyboardInterrupt:
        sC.close()
        sys.exit(0)

if __name__ == '__main__':
    main()