python基础实现tcp文件传输

准备工作，实现文件上传需要那些工具呢？

　　socket（传输）、open()（打开文件）、os（读取文件信息），当然还有辅助类sys和json，下面我们开始吧

import socket,sys
import json,os

#创建socket对象
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

#获取本地主机名
host = socket.gethostname()

#设置端口号
port = 9000

#绑定端口
server.bind((host,port))

#设置最大连接数
server.listen(5)

#建立连接
while True:
    print(\'开始监听\')
    #建立客户端连接
    clint,addr = server.accept()
    print("连接地址: %s" % str(addr))
    #接收文件属性数据,创建文件
    jsonObj = json.loads(clint.recv(1024).decode(\'utf-8\'))
    print(\'接收到数据\',jsonObj)
    if not jsonObj :
        os.mknod(jsonObj[\'name\'])
    msg = json.dumps({\'name\':\'消息\',\'信号\':\'创建成功\'})
    #回数据确认已经建立新文件
    clint.send(msg.encode(\'utf-8\'))
    #接收数据
    size = 0
    sizeValue = int(jsonObj[\'size\'])
    print(\'开始接收数据\')
    with open(jsonObj[\'name\'], \'wb\') as file:
        while size < sizeValue:
            value = sizeValue - size
            if value >1024:
                getdate = clint.recv(1024)
            else :
                getdate = clint.recv(value)
            file.write(getdate)
            size += 1024
    print(\'结束\')
    #关闭连接
    clint.close()

 

客户端

\'\'\'发送文件
Usage:
clint.py <name> [<rename>]

\'\'\'
import socket
import os,json,sys

#建立json数据，包含文件名以及大小
#{name:name,size:99999}
date = {}
if len(sys.argv) == 2:
    print(type(sys.argv[1]))
    name = sys.argv[1].split(\'\\\')[-1]
    date[\'name\'] = name
else:
    date[\'name\'] = sys.argv[2]
date[\'size\'] = os.path.getsize(sys.argv[1])
jsonString = json.dumps(date).encode(\'utf-8\')

#create connection
clint = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
# 获取本地主机名
host = socket.gethostname()
# 设置端口好
port = 9000
# 连接服务，指定主机和端口
clint.connect((host, port))
#发送文件属性信息
clint.send(jsonString)
#接收服务端发过来的确认信息
clint.recv(1024)
#发送文件数据
size = 0
with open(sys.argv[1],\'rb\') as file:
    while size < date[\'size\']:
        fileDate = file.read(1024)
        clint.send(fileDate)
        size += 1024
#关闭连接
clint.close()