python学习笔记——捌 ScoketServer

服务端：

import socketserverimport json,osimport hashlibclass MyTCPHandler(socketserver.BaseRequestHandler):

def put(self,*args):        server_md5=hashlib.md5()        ‘‘‘接收客户端文件‘‘‘        cmd_dic = args[0]        filename = cmd_dic["filename"]        filesize = cmd_dic["size"]        if os.path.isfile(filename):            f = open(filename + ".new","wb")        else:            f = open(filename , "wb")

self.request.send(b"200 ok")        received_size = 0        while received_size < filesize:            data = self.request.recv(1024)            server_md5.update(data)            f.write(data)            received_size += len(data)

else:            print("file [%s] has uploaded..." % filename)

self.request.send(server_md5.hexdigest().encode())

def get(self,*args):        md5_server=hashlib.md5()        #接收客户端的信息        cmd_dic=args[0]        #得到文件名称        filename=cmd_dic["filename"]        #判断是不是个文件        if os.path.isfile(filename):            #得到文件大小            file_size=os.stat(filename).st_size            #j将文件大小发送给客户端            self.request.send(str(file_size).encode())            #对应客户端的胡写，防止粘包            self.request.recv(1024)            #开始发送文件了            f=open(filename,"rb")            for line in f :                self.request.send(line)                md5_server.update(line)            #将MD5后的值 发送给客户端            print(md5_server.hexdigest())            self.request.send((md5_server.hexdigest()).encode())        else:            print("服务端没有此文件")

def handle(self):

while True:            try:                self.data = self.request.recv(1024).strip()                print("{} wrote:".format(self.client_address[0]))                print(self.data)                cmd_dic = json.loads(self.data.decode())                action = cmd_dic["action"]                if hasattr(self,action):                    func = getattr(self,action)                    func(cmd_dic)

except ConnectionResetError as e:                print("err",e)                break

if __name__ == "__main__":    HOST, PORT = "localhost", 9997    # Create the server, binding to localhost on port 9999    server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)    server.serve_forever()

客户端

_author__ = "Alex Li"import socketimport jsonimport osimport hashlib#用于打印进度条import sys,time

# client.connect((‘192.168.16.200‘,9999))

class FtpClient(object):    def __init__(self):        self.client = socket.socket()

def help(self):        msg = ‘‘‘        ls        pwd        cd ../..        get filename        put filename        ‘‘‘        print(msg)

def connect(self, ip, port):        self.client.connect((ip, port))    #利用反射，通过输入的字符串 来调用不同的方法    def interactive(self):        md5_client=hashlib.md5()        # self.authenticate()

#执行操作的代码        while True:            cmd = input(">>").strip()            if len(cmd) == 0: continue            cmd_str = cmd.split()[0]            if hasattr(self, "cmd_%s" % cmd_str):                func = getattr(self, "cmd_%s" % cmd_str)                func(cmd)            else:                self.help()

def cmd_put(self, *args):        cmd_split = args[0].split()        if len(cmd_split) > 1:            filename = cmd_split[1]            if os.path.isfile(filename):                filesize = os.stat(filename).st_size                msg_dic = {                    "action": "put",                    "filename": filename,                    "size": filesize,                    "overridden": True                }                self.client.send(json.dumps(msg_dic).encode("utf-8"))                print("send", json.dumps(msg_dic).encode("utf-8"))                # 防止粘包，等服务器确认                server_response = self.client.recv(1024)                f = open(filename, "rb")                #用于MD5校验                client_md5=hashlib.md5()                #用于计算发送了多少数据                clien_size=0                b=0                for line in f:                    self.client.send(line)                    client_md5.update(line)                    clien_size +=len(line)

# 计算接收的数据 占 总文件的百分比                    a = int(clien_size / filesize * 100)                    if b != a and a % 2 == 0:                        # print("%d" % a + "%")                        sys.stdout.write("#")                        sys.stdout.flush()                    else:                        pass                    b = a                else:                    print("100%")                    f.close()                 #客户端发送数据的md5值                client_md5=client_md5.hexdigest()

else:                print(filename, "is not exist")

server_ma5=self.client.recv(1024)            server_ma5=server_ma5.decode()            #判断 两边的md5值是否是一致的            if server_ma5==client_md5:                print("文件成功上传至服务器")            else:                print("上传出错")

def cmd_get(self, *args):        # 用md5对比文件是否一致        md5_clent = hashlib.md5()        ‘‘‘此方法用于下载操作‘‘‘        # 把输入的命令 按空格分隔 切分成列表        cmd_split = args[0].split()        # 如果命令的长度大于0        if len(cmd_split) > 1:            # 得到文件名            filename = cmd_split[1]            # 服务端那边解析的是一个字典，所以我也写个字典

msg_dic = {                "action": "get",                "filename": filename            }            print(msg_dic)            # 将动作和文件名发给服务端            self.client.send(json.dumps(msg_dic).encode("utf-8"))            print("发送完毕")

# 获得服务端文件的大小            size = self.client.recv(1024).decode()            print(type(size))            print(size)            server_size = int(size)            print("获得服务器文件的大小是%s" % server_size)            # 胡写一个，防止粘包            self.client.send(b"jiehsou wenjian")            # 准备接受 服务端发来的数据            # 已经获取文件的大小            clien_size = 0            if os.path.isfile(filename):                f = open(filename + "new", "wb")            else:                f = open(filename, "wb")            #防止重复出现相同的百分比 而设置的变量            b=0            while clien_size < server_size:                if server_size - clien_size > 1024:                    data_size = 1024                else:                    data_size = server_size - clien_size

data=self.client.recv(data_size)                md5_clent.update(data)                f.write(data)                #计算接收过来的数据长度                clien_size += len(data)

#计算接收的数据 占 总文件的百分比                a=int(clien_size/server_size*100)                if b!=a  and a%2==0:                    # print("%d" % a + "%")                    sys.stdout.write("#")                    sys.stdout.flush()                else:                    pass                b=a

# 接收server过来的md5值        server_md51 = self.client.recv(1024)        server_md52 = server_md51.decode()        clien_md5= md5_clent.hexdigest()        # print("md5_server1--%s" % server_md51)        # print("md5_server2--%s" % server_md52)        #        # print("md5_client--%s" % clien_md5)        # print(server_md51==clien_md5)        # print(server_md52==clien_md5)        if server_md52==clien_md5 :            print("100%")        else:            print("文件可能被篡改，请重新下载")

ftp = FtpClient()ftp.connect("localhost", 9997)ftp.interactive()