python的socket编程接收浏览器上传的文件_python网络编程socket模块实现ftp上传下载...

本实验实现ftp上传文件下载文件功能，并具有校验文件完整性，打印进度条功能,

主要练习socket,struct模块。

ftp用户文件存放在user.json文件中

user.json文件内容

{"lisi": "abcdef", "hyh": "123456"}

ftp客户端脚本ftpclient.py

#!/usr/bin/python

# --*-- coding: utf-8 --*--

import socket

import json

import time

import sys

import struct

from hashlib import md5

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

ftp_obj.connect((‘127.0.0.1‘, 8080))

def login_auth(ftp_obj):

"""用户登录三次认证,输入用户名密码跟服务器账号密码匹配返回True,否则返回False"""

count = 0

while count 

user_name = input("请输入账户: ").strip()

user_passwd = input("请输入密码: ").strip()

if user_name and user_passwd:

ftp_obj.send(user_name.encode(‘utf-8‘))

ftp_obj.send(user_passwd.encode(‘utf-8‘))

else:

count += 1

continue

#time.sleep(10)

login_res_bytes = ftp_obj.recv(1024)

# print(login_res_bytes)

# print(login_res_bytes.decode(‘gbk‘))

login_res = login_res_bytes.decode(‘gbk‘)

if login_res == ‘True‘:

return True

else:

count += 1

continue

else:

return False

def progress_bar(num, total):

"""打印进度条"""

rate = num / total

rate_num = int(rate * 100)

r = ‘\r%s%d%%‘ % (‘#‘ * rate_num, rate_num,)

print(r)

def get_md5(data):

"""校验文件内容"""

m = md5()

m.update(data.encode(‘utf-8‘))

res = m.hexdigest()

return res

def ftp_cmd(ftp_obj):

"""上传下载命令"""

while True:

cmd = input("请输入命令(dir or put filepath or get filepath or exit)>>: ").strip()

if cmd == ‘exit‘:

sys.exit(0)

#print(cmd_list)

if cmd == ‘dir‘:

ftp_obj.send(cmd.encode(‘utf-8‘))

head_struct = ftp_obj.recv(4)

head_len = struct.unpack(‘i‘, head_struct)[0]

head_bytes = ftp_obj.recv(head_len)

head_json = head_bytes.decode(‘utf-8‘)

head_dict = json.loads(head_json)

total_size = head_dict[‘total_size‘]

recv_size = 0

data = b‘‘

while recv_size 

recv_data = ftp_obj.recv(1024)

data += recv_data

recv_size += len(recv_data)

print(data.decode(‘gbk‘))

else:

cmd_list = cmd.split()

if cmd_list[0] == ‘get‘:

ftp_obj.send(cmd.encode(‘utf-8‘))

head_struct = ftp_obj.recv(4)

head_len = struct.unpack(‘i‘, head_struct)[0]

head_bytes = ftp_obj.recv(head_len)

head_json = head_bytes.decode(‘utf-8‘)

head_dict = json.loads(head_json)

total_size = head_dict[‘total_size‘]

recv_size = 0

data = b‘‘

while recv_size 

recv_data = ftp_obj.recv(1024)

data += recv_data

recv_size += len(recv_data)

progress_bar(recv_size, total_size)     #打印进度条，传入两个参数，第一个是接受的数据字节，第二个是数据头解包的数据总长度

with open(cmd_list[1], ‘w‘, encoding=‘utf-8‘) as f:

f.write(data.decode(‘utf-8‘))

check_md5 = get_md5(data.decode(‘utf-8‘))   #校验传输的内容

if head_dict[‘hashlib‘] == check_md5:

print("下载成功，文件内容完整")

else:

print("下载完成，文件内容不完整")

elif cmd_list[0] == ‘put‘:

ftp_obj.send(cmd.encode(‘utf-8‘))

with open(cmd_list[1], ‘r‘, encoding=‘utf-8‘) as f:

data = f.read()

check_put_md5 = get_md5(data)

head_dict = {‘filename‘: cmd_list[1], ‘hashlib‘: check_put_md5, ‘total_size‘: len(data)}

head_json = json.dumps(head_dict)

head_bytes = head_json.encode(‘utf-8‘)

ftp_obj.send(struct.pack(‘i‘, len(head_bytes)))

ftp_obj.send(head_bytes)

is_upload = ftp_obj.recv(10)

if is_upload.decode(‘gbk‘) == ‘True‘:

#ftp_obj.send(data.encode(‘utf-8‘))

with open(cmd_list[1], ‘r‘, encoding=‘utf-8‘) as f:

dataline = f.readlines()

data_len = 0

for i in dataline:

ftp_obj.send(i.encode(‘utf-8‘))

data_len += len(i)

time.sleep(0.1)

progress_bar(data_len, len(data))

print("上传成功")

is_check = ftp_obj.recv(10)

if is_check.decode(‘gbk‘) == ‘True‘:

print("文件上传完整")

else:

print("文件上传不完整")

else:

print("文件太大，超出磁盘限额")

continue

else:

print("命令错误，重重新输入")

continue

def main():

auth_res = login_auth(ftp_obj)

if auth_res:

ftp_cmd(ftp_obj)

else:

print("用户或密码不正确，退出程序")

sys.exit(1)

if __name__ == ‘__main__‘:

main()

ftp服务端脚本ftpserver.py

#!/usr/bin/python

# --*-- coding:utf-8

import socket

import json

import os

import subprocess

import struct

from hashlib import md5

import time

dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

hyh_size = 5    #‘hyh‘用户M空间限额5M

lisi_size = 10  #‘lisi‘用户空间限额10M

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

ftp_obj.bind((‘127.0.0.1‘, 8080))

ftp_obj.listen(5)

def size(user_home_dir):

"""计算用户家目录大小"""

# print(user_home_dir)

# print(os.listdir(user_home_dir))

file_size_bytes = 0

for i in os.listdir(user_home_dir):

file = user_home_dir + ‘\\‘ + i

file_size_bytes += os.path.getsize(file)

file_size_mbytes = file_size_bytes / (1024 * 1024)

return file_size_mbytes

def get_md5(data):

"""校验文件内容"""

m = md5()

m.update(data.encode(‘utf-8‘))

res = m.hexdigest()

return res

def ftp_load(conn, user_name):

"""接收客户端命令，并执行"""

user_home_dir = dir + ‘\\‘ + user_name

used_size = size(user_home_dir)

while True:

try:

print(‘开始接收发送数据‘)

cmd_res_bytes = conn.recv(1024)

cmd_res = cmd_res_bytes.decode(‘utf-8‘)

if cmd_res == ‘dir‘:

res = subprocess.Popen(cmd_res + ‘ ‘ + user_home_dir, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

err = res.stderr.read()

if err:

cmd_stdout = err

else:

cmd_stdout = res.stdout.read()

#print(cmd_stdout.decode(‘gbk‘))

head_dict = {‘dir‘: user_name, ‘hashlib‘: None, ‘total_size‘: len(cmd_stdout)}

head_json = json.dumps(head_dict)

head_bytes = head_json.encode(‘utf-8‘)

conn.send(struct.pack(‘i‘, len(head_bytes)))

conn.send(head_bytes)

conn.send(cmd_stdout)

else:

cmd_res_list = cmd_res.split()  #获取命令和文件列表

user_home_file = user_home_dir + ‘\\‘ + cmd_res_list[1] #获取文件

if cmd_res_list[0] == ‘get‘:

with open(user_home_file, ‘r‘, encoding=‘utf-8‘) as f:

data = f.read()

check_md5 = get_md5(data)   #校验文件内容md5

head_dict = {‘filename‘: cmd_res_list[1], ‘hashlib‘: check_md5, ‘total_size‘: len(data)}

head_json = json.dumps(head_dict)

head_bytes = head_json.encode(‘utf-8‘)

conn.send(struct.pack(‘i‘, len(head_bytes)))

conn.send(head_bytes)

with open(user_home_file, ‘r‘, encoding=‘utf-8‘) as f:

dataline = f.readlines()

for i in dataline:

time.sleep(0.1)

conn.send(i.encode(‘utf-8‘))

#conn.send(data.encode(‘utf-8‘))

else:

head_struct = conn.recv(4)

head_len = struct.unpack(‘i‘, head_struct)[0]

head_bytes = conn.recv(head_len)

head_json = head_bytes.decode(‘utf-8‘)

head_dict = json.loads(head_json)

total_size = head_dict[‘total_size‘]

if user_name == ‘hyh‘:

if hyh_size - used_size > total_size / (1024 * 1024):

conn.send(‘True‘.encode(‘utf-8‘))

recv_size = 0

data = b‘‘

while recv_size 

recv_data = conn.recv(1024)

data += recv_data

recv_size += len(recv_data)

with open(user_home_file, ‘w‘, encoding=‘utf-8‘) as f:

f.write(data.decode(‘utf-8‘))

check_put_md5 = get_md5(data.decode(‘utf-8‘))

if head_dict[‘hashlib‘] == check_put_md5:

conn.send(‘True‘.encode(‘utf-8‘))

continue

else:

conn.send(‘False‘.encode(‘utf-8‘))

continue

else:

conn.send(‘False‘.encode(‘utf-8‘))

continue

elif user_name == ‘lisi‘:

if lisi_size - used_size > total_size / (1024 * 1024):

conn.send(‘True‘.encode(‘utf-8‘))

recv_size = 0

data = b‘‘

while recv_size 

recv_data = conn.recv(1024)

data += recv_data

recv_size += len(recv_data)

with open(user_home_file, ‘w‘, encoding=‘utf-8‘) as f:

f.write(data.decode(‘utf-8‘))

continue

else:

conn.send(‘False‘.encode(‘utf-8‘))

continue

else:

continue

except Exception:

break

while True:

print(‘waitting to accept...‘)

conn, addr = ftp_obj.accept()

print(‘client: ‘, addr)

while True:

try:

user_name_bytes = conn.recv(30)

user_passwd_bytes = conn.recv(30)

user_name = user_name_bytes.decode(‘gbk‘)

user_passwd = user_passwd_bytes.decode(‘gbk‘)

with open(‘user.json‘, ‘r‘, encoding=‘utf-8‘) as f:

user_dict = json.loads(f.read())

if user_name in user_dict and user_dict[user_name] == user_passwd:

conn.send(‘True‘.encode(‘utf-8‘))

print("ftp连接成功")

ftp_load(conn, user_name)  #ftp上传下载实现

else:

conn.send(‘False‘.encode(‘utf-8‘))

except Exception:

conn.send(‘False‘.encode(‘utf-8‘))

conn.close()

ftp_obj.close()

原文：http://haoyonghui.blog.51cto.com/4278020/1941026