cliente.py

import socket
import random
import hashlib
import logging
import time
import datetime
import tqdm


def main():

    filesize = 104865944
    host = '34.227.110.129'
    puerto = 55555
    print('Hola, Cliente')
    fecha = datetime.datetime.today().strftime('%Y-%m-%d_%H-%M-%S')
    logging.basicConfig(filename="./archivosC/Log" + fecha + ".log", level=logging.INFO,
                        format='%(asctime)s %(levelname)-8s %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
    print('Se terminó de configurar el log de su pronta conexión')
    hs = hashlib.sha256()
    print(hs)
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect((host, puerto))
    print("Se conecto satisfactoriamente con el host ",
          host, " en el puerto: ", puerto)
    print("Cliente comenzará a recibir información")
    logging.info("Cliente comenzará a recibir información")

    data = sock.recv(1024)
    strings = data.decode('utf8').split('|')
    filename = strings[0]
    strings2 = filename.split('-')[0].split("-")
    idcliente = strings2[0]
    idcliente = idcliente[0]
    filesize = int(strings[1])
    print("Cliente ", idcliente, " recibirá archivo ",
          str(filename), " con peso ", str(filesize))
    logging.info("Cliente " + idcliente + " recibirá archivo " +
                 str(filename) + " con peso " + str(filesize))

    input("Presione enter para enviar la confirmación de que se encuentra listo para recibir el archivo")

    sock.send(str(1).encode('utf8'))

    print("Cliente ", str(idcliente),
          " envia confirmación de estado preparado para recibir archivo")
    logging.info("Cliente " + str(idcliente) +
                 " envia confirmación de estado preparado para recibir archivo")
    dataTotal = b''
    inicio = time.time()
    cond = True
    filename = str('./archivosC/archivosRecibidos/' + filename)

    progress = tqdm.tqdm(range(
        filesize), f"Receiving {filename}", unit="B", unit_scale=True, unit_divisor=1024)

    with open(filename, "wb") as f:
        while True:
            data = sock.recv(1048576)
            dataTotal += data

            f.write(data)
            progress.update(len(data))
            if data and cond:
                cond = False
                print("Cliente ", str(idcliente),
                      " recibe el primer paquete del archivo")
                logging.info("Cliente " + str(idcliente) +
                             " recibe el primer paquete del archivo")
                inicio = time.time()
            if not data:
                print("Termino el envío con exito")
                logging.info("Termino el envio con exito")
                fin = time.time()
                break
            elif (data.__contains__(b"Hash:")):

                j = dataTotal.find(b"Hash:")
                hs.update(dataTotal[:j])
                prueba = hs.hexdigest()

                i = data.find(b"Hash:")
                recibido = data[i+5:]
                recibidoD = recibido.decode()
                logging.info("Se recibe el hash junto con el archivo ")
                print("Se recibe el siguiente hash ", recibidoD)
                print("Se crea el siguiente hash de prueba ", prueba)

                if hs.hexdigest() == recibido.decode():
                    print("El hash fue verificado. Es correcto :)")
                    logging.info("Hash verificado. Archivo correcto")
                else:
                    print(
                        "El hash no pudo ser verificado. Hay un problema con el archivo recibido :(")
                    logging.info("Hash erroneo. Archivo incorrecto")

    logging.info('Bytes recibidos: %s , paquetes recibidos: %s , tiempo utilizado: %s, tasa de transferencia: %s', len(
        dataTotal), len(dataTotal)/1048576, (fin-inicio), (len(dataTotal)/(fin-inicio)))

    sock.close()


main()