added function for list

Author: hasanimran96

client.py

@@ -35,10 +35,15 @@ def append_to_file(file_name, message):
     fd.write(message)
 
 
-def list_files(directory):
-    temp_list = os.listdir(directory)
-    for file in temp_list:
-        print(file)
+def list_files(server_sock, directory):
+    while True:
+        data = server_sock.recv(1024)
+        if not data:
+            break
+        temp_list = data
+    for item in temp_list:
+        print(item)
+
 
 
 def send_file(server_sock, file_name):
@@ -72,10 +77,15 @@ def make_socket():
     # create a socket object
     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 
-    # get local machine name
-    host = socket.gethostname()
+    # get server ip
+    print("Enter server IP")
+    msg = input()
+    host = msg
 
-    port = 9999
+    # get server port
+    print("Enter server port")
+    msg = input()
+    port = msg
 
     # connection to hostname on the port.
     s.connect((host, port))

server.py

@@ -1,5 +1,6 @@
 import socket
 import threading
+import os
 
 # set directory for file server
 root = "Root/"
@@ -53,15 +54,22 @@ def recieve_from_server(socket):
 
         msg = socket.recv(1024)
         if len(msg) == 0:
-            socket.close()
-        elif msg.decode("utf-8") == "close":
-            socket.close()
+            socket.send("error")
+        elif msg == "list local":
+            temp_list = list_local("Root")
+            socket.send(str(temp_list))
         else:
-            if msg.decode("utf-8") == "download":
-                msg = "Here is your downloaded file haha"
-                socket.send(msg.encode("utf-8"))
-            else:
-                socket.send(msg)
+            socket.send("error")
+
+
+def list_local(directory):
+    temp_list = os.listdir(directory)
+    return temp_list
+
+
+def list_global():
+    for server in servers:
+        server.send("list local")
 
 
 def send_file(client_sock, file_name):