From 238fe8c494ab5be80c96441095d1c8958f95c04d Mon Sep 17 00:00:00 2001
From: NIKITA PANDEY <113332472+nikitapandeyy@users.noreply.github.com>
Date: Fri, 31 Mar 2023 19:38:13 +0530
Subject: [PATCH] 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>
---
 file_transfer/receive_file.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/file_transfer/receive_file.py b/file_transfer/receive_file.py
index 37a503036..f50ad9fe1 100644
--- a/file_transfer/receive_file.py
+++ b/file_transfer/receive_file.py
@@ -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()