minhtrannhat/MyFTP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: refactored list command

Browse Source
This commit is contained in:
minhtrannhat
2023-11-30 15:35:41 -05:00
parent a50e6c75c8
commit e079dd8a5d
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/myftp/client.py
View File

@@ -51,11 +51,7 @@ class UDPClient:
# list files available on the server
elif command == "list":
client_socket.send(command.encode())
encoded_message, server_address = client_socket.recvfrom(4096)
file_list = pickle.loads(encoded_message)
print(f"Received file list from {server_address}: {file_list}")
client_socket.close()
self.get_files_list_from_server(client_socket)
continue
# help
@@ -104,6 +100,10 @@ class UDPClient:
print(
f"myftp> - {self.mode} - ConnectionRefusedError happened. Please restart the client program, make sure the server is running and/or put a different server name and server port."
)
except Exception as error:
print(
f"myftp> - {self.mode} - {error} happened."
)
finally:
client_socket.close() # type: ignore
@@ -140,6 +140,16 @@ class UDPClient:
# Close the socket
client_socket.close()
# get list of files currently on the server
def get_files_list_from_server(self, client_socket: socket) -> list[str]:
client_socket.send("list".encode())
encoded_message, server_address = client_socket.recvfrom(4096)
file_list = pickle.loads(encoded_message)
print(f"Received file list from {server_address}: {file_list}")
client_socket.close()
return file_list
def get_address_input() -> Address:
while True: