import socket
socket.setdefaulttimeout(1200)

# 测收发、发收文件是否一致
def send_file(path, ip, port):
    if "-" in ip:
        ip = ip.split('-')[0]
    print(f"Sending file to {ip}:{port}")
    with open(path, "rb") as fi:
        while True:
            try:
                s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                s.connect((ip, port))
                s.sendfile(fi)
                break
            except ConnectionRefusedError:
                pass
            finally:
                s.close()
def recv_file(path, port):
    with open(path, "wb") as fi:
        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
            s.bind(('', port))
            s.listen(1)
            conn, ip = s.accept()
            print(f"Receiving file from {ip}:{port}")
            with conn:
                while True:
                    data = conn.recv(4096)
                    if not data:
                        break
                    fi.write(data)