client4.py

来自「python web programming 部分」· Python 代码 · 共 29 行

import socket
import sys
import netstring

sockType = socket.SOCK_STREAM
if len(sys.argv) > 1:
    if sys.argv[1] == "-u":
        sockType = socket.SOCK_DGRAM
    elif sys.argv[1] != "-t":
        sys.exit("Unknown option: "+sys.argv[1])        

while 1:
    arg = raw_input("Send: ")
    if not arg:
        break
    s = socket.socket(socket.AF_INET, sockType)
    if sockType == socket.SOCK_STREAM:
        addr = ('127.0.0.1', 8081)
        s.connect(addr)
        netstring.writens(s, arg)
        ret = netstring.readns(s)
    else:
        s.sendto(netstring.encode(arg), ('127.0.0.1', 8081))
        ret, addr = s.recvfrom(1024)
        ret = netstring.decode(ret)
    print "Address:", addr, "Returned:", ret
    s.close()