python通过socket传文件_Python通过Socket实现文件传输

客户端# -*- coding: utf-8 -*-

import socket

import argparse

import os

import hashlib

import pickle

def parse_args():

parser = argparse.ArgumentParser()

parser.add_argument("-f","--file",help="the file you want to send")

parser.add_argument("-i","--ip",help="target's ip address")

args = parser.parse_args()

filePath = args.file

ip = args.ip

print "File path : %s" % filePath

return filePath,ip

def read_file(filePath,eachCount):

f = open(filePath,'r')

size = os.path.getsize(filePath) * 1024

print "Sent file size:",size

data = ""

tmp = f.read(eachCount)

data += tmp

while len(tmp) >0 :

tmp = f.read(eachCount)

data += tmp

f.close()

# print data

md5_data = hashlib.md5(data).hexdigest()

info = {

'size':size,

'md5':md5_data,

'name':os.path.basename(filePath)

}

return data,info

if __name__ == "__main__":

filePath,ip = parse_args()

d,info = read_file(filePath,1024)

s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

if not ip :

ip = "127.0.0.1"

s.connect((ip,9999))

print s.recv(1024)

s.sendall("#info#"+pickle.dumps(info))

print s.recv(1024)

s.send("#file#"+d)

print s.recv(1024)

s.send('exit')

print s.recv(1024)

s.close()

服务端:# -*- coding: utf-8 -*-

import socket

import time

import argparse

import threading

import os

import hashlib

import pickle

def tcplink(sock,addr):

print "Now accept new connection from %s:%s..." %addr

sock.send("ok")

while True:

data = sock.recv(2048)

if data == "exit" or not data:

break

if data.startswith("#info#"):

str_info = data[6:]

info = pickle.loads(str_info)

print info

sock.send("ok")

size = info['size']

d = sock.recv(4096)

print "Rec:",d

if d.startswith("#file#"):

if len(d) >= 4096:

d = d[6:]

write_file(info['name'],d)

while size >0 :

if size >= 4096:

print size

d = sock.recv(4096)

write_file(info['name'],d)

size -= 4096

else:

print "end",size

d = sock.recv(size)

size = 0

write_file(info['name'],d)

else:

write_file(info['name'],d[6:])

sock.send("ok")

sock.send("bye")

sock.close()

print "Now disconnect socket from %s:%s..." %addr

def write_file(name,data):

f = open("recv/%s"%name,"a+")

f.write(data)

print "recv/%s"%name

f.close()

if __name__ == "__main__":

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

s.bind(("127.0.0.1",9999))

s.listen(5)

print "Waiting for connection..."

while True:

sock,addr = s.accept()

t=threading.Thread(target=tcplink(sock,addr))

Usage:python rec.py

python send.py -f '{file path}' -i '{ip address}'