python socket 进行文件上

!/bin/python

#coding:utf-8 import SocketServer import os import datetime import MySQLdb class mysql: def init(self): self.connect = MySQLdb.connect( host = '192.168.221.203', user = 'hall', passwd = '520157', port = 3306, db = 'info', charset = 'utf8' ) def mysql_create_table_info(self): cursor = self.connect.cursor() sql = """create table if not exists info( action char(20), actin_time varchar(40), file_size varchar(20), file_name varchar(40), operation_user char(20) )""" cursor.execute(sql) cursor.close() self.connect.commit() self.connect.close() def mysql_create_table_user(self): cursor = self.connect.cursor() sql_user = """create table if not exists user( user varchar(20), passwd varchar(40) )""" cursor.execute(sql_user) data = """insert into user values( "hall", "hall"), ("hexulin", "hexulin")""" cursor.execute(data) cursor.close() self.connect.commit() self.connect.close()

class server(SocketServer.BaseRequestHandler,mysql): def handle(self): self.request self.client_address print "%s:%s is connecting....."%self.client_address recv = self.request.recv(1024) print recv self.request.send("已经建立连接") recv = self.request.recv(1024) print recv operation = recv.split(':')[1] if operation == "put": self.file_name,self.file_size = self.request.recv(1024).split('|') self.request.send("一切准备就绪") recv_size = 0 time_put = datetime.datetime.now() file_path = ['/root/',self.file_name] file_path = ''.join(file_path) f = open(file_path,'wb') while recv_size != int(self.file_size): if int(self.file_size) - recv_size > 1024: rdata = self.request.recv(1024) recv_size += len(rdata) else: rdata = self.request.recv(int(self.file_size) - recv_size) recv_size = int(self.file_size) f.write(rdata) f.close() print "文件已经保存完毕...." connect = MySQLdb.connect( host = '192.168.221.203', user = 'hall', passwd = '520157', port = 3306, db = 'info', charset = 'utf8' ) value = [operation,time_put,self.file_size,self.file_name,"hall"]

        cursor = connect.cursor()
                    data = """insert into info values(
                            %s,
                            %s,
                            %s,
                            %s,
            %s)"""
                    cursor.execute(data,value)
                    cursor.close()
                    connect.commit()
                    connect.close()

    else:
        file_path = self.request.recv(1024)
                file_name = os.path.basename(file_path)
                file_size = os.stat(file_path).st_size
                self.request.send(file_name + '|' + str(file_size))
                recv = self.request.recv(1024)
                print recv
                send_size = 0
        time_get = datetime.datetime.now()
                f = open(file_path,'rb')
                while True:
                        file_data = f.read(1024)
                        if not file_data:
                                break
                        self.request.send(file_data)
                f.close()
        print "文件已经传输完毕..."
        connect = MySQLdb.connect(
                host = '192.168.221.203',
                user = 'hall',
                passwd = '520157',
                port = 3306,
                db = 'info',
                charset = 'utf8'
                )
        value = [operation,time_get,file_size,file_name,"hall"]
        cursor = connect.cursor()
                    data = """insert into info values(
                            %s,
                            %s,
                            %s,
                            %s,
            %s)"""
                    cursor.execute(data,value)
                    cursor.close()
                    connect.commit()
                    connect.close()

class insert_mysql: def mysql_info(self): connect = MySQLdb.connect( host = '192.168.221.203', user = 'hall', passwd = '520157', port = 3306, db = 'info', charset = 'utf8' )

            cursor = connect.cursor()
            data = """insert into info values(
                   operation,
                   time),
                   self.file_size,
                   self.file_name)"""
            cursor.execute(data)
            cursor.close()
            connect.commit()
            connect.close()

if name == 'main': server = SocketServer.ThreadingTCPServer(('192.168.221.203',8888),server) server.serve_forever()

客户端 #!/bin/python #coding:utf-8 import socket import os

sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM) sock.connect(('192.168.221.203',8888)) sock.send("我要连接你进行文件传输........") recv = sock.recv(1024) print recv operation = raw_input("please input your choice Usage: put or get:") sock.send("我要进行的操作是:%s"%operation) if operation == "put": file_path = raw_input("please input your input_file_path:") file_name = os.path.basename(file_path) file_size = os.stat(file_path).st_size print file_size sock.send(file_name + '|' + str(file_size)) print sock.recv(1024) f = open(file_path,'rb') while True: file_data = f.read(1024) if not file_data: break sock.send(file_data) f.close() print "上传文件完毕....." else: file_path = raw_input("please input your get_file_path:") sock.send(file_path) file_name,file_size = sock.recv(1024).split('|') get_file_path = ['/root/',file_name] get_file_path = ''.join(get_file_path) sock.send("可以开始发送，准备就绪") recv_size = 0 f = open(get_file_path,'wb') while not int(file_size) == recv_size: if int(int(file_size) - recv_size) > 1024: rdata = sock.recv(1024) recv_size += len(rdata) else: rdata = sock.recv(int(file_size) - recv_size) recv_size = int(file_size) f.write(rdata) f.close() print "文件下载完毕....." sock.close()