Update receive_file.py (#8541)

* 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>
This commit is contained in:
NIKITA PANDEY 2023-03-31 19:38:13 +05:30 committed by GitHub
parent a00492911a
commit 238fe8c494
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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))
@ -13,11 +14,14 @@ if __name__ == "__main__":
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()