file_server.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-

import socket
import sys
import os
import threading
import hashlib


def get_file_md5(file_path):
    m = hashlib.md5()

    with open(file_path, "rb") as f:
        while True:
            data = f.read(1024)
            if len(data) == 0:
                break    
            m.update(data)
    
    return m.hexdigest().upper()


def send_one_file(sock_conn, file_abs_path):
    '''
    函数功能：将一个文件发送给客户端
    参数描述：
        sock_conn 套接字对象
        file_abs_path 待发送的文件的绝对路径
    '''
    file_name = file_abs_path[len(dest_file_parent_path):]
    if file_name[0] == '\\' or file_name[0] == '/':
        file_name = file_name[1:]

    file_size = os.path.getsize(file_abs_path)
    file_md5 = get_file_md5(file_abs_path)

    file_name = file_name.encode()
    file_name += b' ' * (300 - len(file_name))
    file_size = "{:<15}".format(file_size).encode()

    file_desc_info = file_name + file_size + file_md5.encode()

    sock_conn.send(file_desc_info)
    with open(file_abs_path, "rb") as f:
        while True:
            data = f.read(1024)
            if len(data) == 0:
                break
            sock_conn.send(data)


def send_empty_dir(sock_conn, dir_abs_path):
    '''
    函数功能：将一个空文件夹发送给客户端
    参数描述：
        sock_conn 套接字对象
        dir_abs_path 待发送的空文件夹的绝对路径
    '''
    file_name = dir_abs_path[len(dest_file_parent_path):]
    print(type(file_name))
    if file_name[0] == '\\' or file_name[0] == '/':
        file_name = file_name[1:]

    file_size = -1
    file_md5 = " " * 32

    file_name = file_name.encode()
    file_name += b' ' * (300 - len(file_name))
    file_size = "{:<15}".format(file_size).encode()

    file_desc_info = file_name + file_size + file_md5.encode()
    sock_conn.send(file_desc_info)


def send_file_thread(sock_conn):
    try:
        for root, dirs, files in os.walk(dest_file_abs_path):
            if len(dirs) == 0 and len(files) == 0:
                send_empty_dir(sock_conn, root)
                continue

            for f in files:
                file_abs_path = os.path.join(root, f)
                print(file_abs_path)
                send_one_file(sock_conn, file_abs_path)
    except Exception as e:
        print(e)
    finally:
        sock_conn.close()

dest_file_abs_path = os.path.abspath(sys.argv[1])
dest_file_parent_path = os.path.dirname(dest_file_abs_path)
dest_file_name = os.path.basename(dest_file_abs_path)


sock_listen = socket.socket()
sock_listen.bind(("0.0.0.0", 9999))
sock_listen.listen(5)

while True:
    sock_conn, client_addr = sock_listen.accept()
    print(client_addr, "已连接！")
    threading.Thread(target=send_file_thread, args=(sock_conn, )).start()

sock_listen.close()