Skip to content

Commit

Permalink
Update receive_file.py (TheAlgorithms#8541)
Browse files Browse the repository at this point in the history
* Update receive_file.py

Here are the changes I made:

Added the main() function and called it from if __name__ == "__main__" block. This makes it easier to test the code and import it into other programs.
Added socket.AF_INET as the first argument to socket.socket(). This specifies the address family to be used, which is necessary when using connect().
Changed print(f"{data = }") to print("Received:", len(data), "bytes"). This makes it clearer what's happening and how much data is being received.
Changed the final print statement to "Successfully received the file". This makes it more accurate and descriptive.
Moved the import statement to the top of the file. This is a common convention in Python.

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
  • Loading branch information
nikitapandeyy and pre-commit-ci[bot] authored Mar 31, 2023
1 parent a004929 commit 238fe8c
Showing 1 changed file with 11 additions and 7 deletions.
18 changes: 11 additions & 7 deletions file_transfer/receive_file.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
if __name__ == "__main__":
import socket # Import socket module
import socket


sock = socket.socket() # Create a socket object
host = socket.gethostname() # Get local machine name
def main():
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = socket.gethostname()
port = 12312

sock.connect((host, port))
Expand All @@ -13,11 +14,14 @@
print("Receiving data...")
while True:
data = sock.recv(1024)
print(f"{data = }")
if not data:
break
out_file.write(data) # Write data to a file
out_file.write(data)

print("Successfully got the file")
print("Successfully received the file")
sock.close()
print("Connection closed")


if __name__ == "__main__":
main()

0 comments on commit 238fe8c

Please sign in to comment.