Skip to content

Commit 9243242

Browse files
committed
fix progress bar reset in the end on transfering files tutorial
1 parent 7d370d2 commit 9243242

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

Diff for: general/transfer-files/client/sender.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def send_file(filename, host, port):
2626
# start sending the file
2727
progress = tqdm.tqdm(range(filesize), f"Sending {filename}", unit="B", unit_scale=True, unit_divisor=1024)
2828
with open(filename, "rb") as f:
29-
for _ in progress:
29+
while True:
3030
# read the bytes from the file
3131
bytes_read = f.read(BUFFER_SIZE)
3232
if not bytes_read:
@@ -47,7 +47,7 @@ def send_file(filename, host, port):
4747
parser = argparse.ArgumentParser(description="Simple File Sender")
4848
parser.add_argument("file", help="File name to send")
4949
parser.add_argument("host", help="The host/IP address of the receiver")
50-
parser.add_argument("-p", "--port", help="Port to use, default is 5001", default=5001)
50+
parser.add_argument("-p", "--port", help="Port to use, default is 5001", type=int, default=5001)
5151
args = parser.parse_args()
5252
filename = args.file
5353
host = args.host

Diff for: general/transfer-files/server/receiver.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
# and writing to the file stream
4242
progress = tqdm.tqdm(range(filesize), f"Receiving {filename}", unit="B", unit_scale=True, unit_divisor=1024)
4343
with open(filename, "wb") as f:
44-
for _ in progress:
44+
while True:
4545
# read 1024 bytes from the socket (receive)
4646
bytes_read = client_socket.recv(BUFFER_SIZE)
4747
if not bytes_read:

0 commit comments

Comments
 (0)