python-TCP模拟ftp文件传输

 #!/usr/bin/python
#coding=utf-8
#server
from socket import*
import sys,os
def command():
    l=[ "Welcome!\n","command\n", "list\n","upload\n","download\n"]
    return l
def list():
    l=[]
    for filename in os.listdir('.'):
        l.append(filename)
    return l
HOST="192.168.1.33"
PORT=5126
ADDR=(HOST,PORT)
sockob=socket(AF_INET,SOCK_STREAM)
sockob.bind(ADDR)
sockob.listen(5)

a=[]
while 1:
    sock,addr=sockob.accept()

    while 1:
        data=sock.recv(1024) 
        str=data.split(" ")

        if data =="command":
            a=command()
            for i in a:
                sock.send(i)
        elif data =="list":
            a=list()
            for i in a:
                sock.send(i)
        elif str[0]=="upload":
            f=open(str[1],'w')
            s=sock.recv(1024)
            print s
            for i in s: #range(len(str)):
                f.write(i)
            print "upload is ok!"
        elif str[0]=="download":
            f=open(str[1],'r')
            s=f.readlines()
            for i in s:
                sock.send(i)
            print "download is ok!"
sockob.close()
sock.close()

#!/usr/bin/python
#coding=utf-8
#client
from socket import *
import sys

HOST='192.168.1.33'
PORT=5126
ADDR=(HOST,PORT)
sockob=socket(AF_INET,SOCK_STREAM)
sockob.connect(ADDR)

while 1:
    data=raw_input(">",)
    str=data.split(' ')
    if not str:
        break
    elif str[0]=="upload":
        f=open(str[1],'r')
        sockob.send(data)
        s0=f.readlines()
        print s0
        for i in s0:
            sockob.send(i)
    elif str[0]=="download":
        f=open(str[1],'w')
        sockob.send(data)
        s1=sockob.recv(1024)
        f.write(s1)
    elif data=="command" or data=="list":
        sockob.send(data)
        data=sockob.recv(1024)
        print data


sockob.close()