client.py

import socket
from os import listdir, walk
from os.path import (
    abspath, join, dirname, getsize, 
    exists, isfile, isdir, basename
)
from buffer import Buffer
from path import convert_path

class FileClient:
    connection = None

    def __init__(self, *args, **kwargs):
        # the ip address or hostname of the server, the receiver
        self.host = kwargs.get("host", "localhost")
        # the port, if not set, the port 5001 is default
        self.port = kwargs.get("port", 5001)
        # get separator
        self.separator = kwargs.get("separator", "<SEPARATOR>")
        # send 4096 bytes each time step
        self.buffer_size = kwargs.get("buffer_size", 4096)
        # absolute path
        self.app_path = kwargs.get("app_path", dirname(abspath(__file__)))
        # files path
        self.path = kwargs.get("path", join(self.app_path, "input"))
        # initilize socket connection
        self.get_connection()

    def get_connection(self):
        if self.connection:
            self.connection.close()

        # create the client socket
        self.connection = socket.socket()
        print(f"[+] Connecting to {self.host}:{self.port}")
        # connecting to the server:

    def file_print(self, filename, filepath, filesize):
        print("\n[FILE NAME]:", filename)
        print("[FILE PATH]:", filepath)
        print("[FILE SIZE]:", filesize, "\n")

    def send_file(self, filename):
        # join file path with filename
        filepath = join(self.path, filename)

        if not exists(filepath):
            print(f"[ERROR]: file {filepath} doesn't exist")
            return

        filetype = "file" if isfile(filepath) or not isdir(filepath) else "folder"

        if filetype == "folder":
            # print(filepath)
            # input()
            if not listdir(filepath):
                self.connection.close()
                self.get_connection()
            else:
                path = basename(convert_path(filepath))
                # print(path)
                for root, subdirs, files in walk(filepath):
                    for f_ in files:
                        joined_path = join(path, f_)
                        # print(joined_path)
                        # input()
                        self.send_file(joined_path)

            return

        # get the file size
        filesize = getsize(filepath)

        self.connection.connect((self.host, self.port))
        print("[+] Connected.")

        sbuf = Buffer(self.connection, self.buffer_size)
        sbuf.put_utf8(f"{filepath}{self.separator}{filesize}{self.separator}{filetype}")

        # pretty print at terminal
        self.file_print(filename, filepath, filesize)

        # start sending the file
        # import tqdm
        # progress = tqdm.tqdm(
        #     range(filesize), f"Sending {filepath}", unit="B", 
        #     unit_scale=True, unit_divisor=1024)

        with open(filepath, "rb") as file_:
            sbuf.put_bytes(file_)
            self.connection.close()
            self.get_connection()

    def bulk_send_file(self):
        for filename in listdir(self.path):
            self.send_file(filename)

if __name__ == "__main__":
    from local import (
        client_host, port, buffer_size, separator,
        app_path, client_path
    )

    # if host not set, the default is localhost
    fc = FileClient(
        host=client_host, port=port, buffer_size=buffer_size,
        separator=separator, app_path=app_path, path=client_path
    )
    # fc = FileClient()
    # fc.send_file("abc.txt")
    fc.bulk_send_file() # send all files from file path to server