client.py

#importing required modules
import socket
import os
import sys
import time



#IP will be of server 
sock = socket.socket()
host = '0.0.0.0'
port = 9999

sock.connect((host, port))


def choose_command(client_currentWD):
    shell_name="\n"+"<"+host+"> " + client_currentWD
    command = input(shell_name)


    if "send" in ''.join(command[:5]).lower():
        send_file_to_server(command)
    elif "receive" in ''.join(command[:9]).lower():
        receive_file_from_server(command)
    elif "sysinfo" in ''.join(command[:9]).lower():
        get_syteminfo_from_server(command)
    elif "exit" in ''.join(command[:5]).lower():
        sock.send(str.encode("exit"))
        sock.close()
        sys.exit()
    elif "list" in ''.join(command[:5]).lower():
        list_commands()

    else:
        client_currentWD=send_command(command,client_currentWD)
    choose_command(client_currentWD)


#list commands
def list_commands():
    print("-send <filepath> <filename>: send file to server")
    print("-receive <filepath> <filename>: receive file from server")
    print("-sysinfo: get system info from server")
    print("-help: list commandline options")
    print("-exit: exit the shell")
    return None


#get system info from server
def get_syteminfo_from_server(command):
    sock.send(str.encode(command))
    print("receiving system info....")
    server_response = str(sock.recv(2147483647), 'utf-8')
    print(server_response)
    return None


#send command to server
def send_command(command,client_currentWD):
    if len(str.encode(command))>0:
            #send the command
            sock.send(str.encode(command))
            #receive the response
            server_response = str(sock.recv(2147483647//2), 'utf-8')
            client_currentWD = server_response[:server_response.find("> ")+2]
            output_response = server_response[server_response.find("> ")+2:]
            print(output_response, end="")
    return client_currentWD


#send file to server
#send "source_filepath" "filename"
def send_file_to_server(command):
    sock.send(str.encode(command))
    file_info_list = command.strip(' ') # remove trailing spaces
    file_info_list = list(map(str,file_info_list.split()))
    file_path = file_info_list[1] # 0th element is the command, 1st element is the file path

    if os.path.isfile(file_path):
        
        file_size = os.path.getsize(file_path)
        sock.send(str.encode(str(file_size)))
        time.sleep(0.3)

        # Opening file and sending data.
        with open(file_path, "rb") as file:
            c = 0
            # Starting the time capture.
            start_time = time.time()

            # Running loop while c != file_size.
            print("Sending file...")
            while c <= file_size:
                data = file.read(1024)
                if not (data):
                    break
                sock.sendall(data)
                c += len(data)
            # Ending the time capture.
            end_time = time.time()


        print("File Transfer Complete. Total time to transfer: ", end_time - start_time)
    else:
        print ("File path does not exist")
    return None



#receive file from server
# receive "filepath from server" "filename"
def receive_file_from_server(command):
    sock.send(str.encode(command))
    file_info_list = command.strip(' ') # remove trailing spaces
    file_info_list = list(map(str,file_info_list.split()))
    if len(file_info_list) == 2:
        file_name="temp_file.txt"
    else:
        file_name = file_info_list[2] # 0th element is the command, 2nd element is the file name
    file_path = "./rec/" + file_name
    os.makedirs(os.path.dirname(file_path), exist_ok=True) 


    # Getting file details.
    file_size = sock.recv(100).decode()

    if file_size == "0":
        print("File does not exist on server.")
        return None
    # Opening and reading file.
    with open(file_path, "wb") as file:
        c = 0
        # Starting the time capture.
        start_time = time.time()
        print("Receiving file...")
        # Running the loop while file is recieved.
        while c < int(file_size):
            data = sock.recv(1024)
            file.write(data)
            c += len(data)

        # Ending the time capture.
        end_time = time.time()

    print("File received. Total time: ", end_time - start_time)
    return None



def main():
    print("Welcome to " + host + "'s " + "shell"+ "(enter list to list commands)")
    client_currentWD = str(sock.recv(1024), 'utf-8')
    choose_command(client_currentWD)


if __name__ == '__main__':
    main()