file_server.py

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

import socket
import sys
import os
import threading
import hashlib
import json
import user_reg_login


conf = json.load(open("server_conf.json"))  # 加载配置信息
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)


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):]
    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_dir(sock_conn):
    '''
    发送非空文件夹
    '''
    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)


def user_service_thread(sock_conn):
    data_len = sock_conn.recv(15).decode().rstrip()
    if len(data_len) > 0:
        data_len = int(data_len)

        recv_size = 0
        json_data = b""
        while recv_size < data_len:
            tmp = sock_conn.recv(data_len - recv_size)
            if tmp == 0:
                break
            json_data += tmp
            recv_size += len(tmp)
        
        json_data = json_data.decode()
        req = json.loads(json_data)

        if req["op"] == 1:
            # 登录校验
            rsp = {"op": 1, "error_code": 0}

            if user_reg_login.check_uname_pwd(req["args"]["uname"], req["args"]["passwd"]):
                rsp["error_code"] = 1
            
            header_data = json.dumps(rsp).encode()
            data_len = "{:<15}".format(len(header_data)).encode()
            sock_conn.send(data_len)
            sock_conn.send(header_data)

            if not rsp["error_code"]:
                send_dir(sock_conn)
            
            sock_conn.close()

        elif req["op"] == 2:
            # 用户注册
            rsp = {"op": 2, "error_code": 0}
            if not user_reg_login.user_reg(req["args"]["uname"], req["args"]["passwd"], req["args"]["phone"], req["args"]["email"]):
                # 注册失败
                rsp["error_code"] = 1

            rsp = json.dumps(rsp).encode()
            data_len = "{:<15}".format(len(rsp)).encode()
            sock_conn.send(data_len)
            sock_conn.send(rsp)            
            sock_conn.close()            

        elif req["op"] == 3:
            # 校验用户名是否存在
            rsp = {"op": 3, "error_code": 0}

            ret = user_reg_login.check_user_name(req["args"]["uname"])
            if ret == 2:
                rsp["error_code"] = 1
            
            rsp = json.dumps(rsp).encode()
            data_len = "{:<15}".format(len(rsp)).encode()
            sock_conn.send(data_len)
            sock_conn.send(rsp)            
            sock_conn.close()            

def main():
    sock_listen = socket.socket()
    sock_listen.bind((conf["app_server_ip"], conf["app_server_port"]))
    sock_listen.listen(5)

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

    sock_listen.close()


if __name__ == '__main__':
    main()