113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
#coding=utf-8
|
|
import socket
|
|
from pprint import pprint
|
|
from time import sleep
|
|
SOCKET_DATA_LENGTH = 1024
|
|
|
|
def kill_server_process(port_number):
|
|
"""ubija proces na porcie - przydatne podczas testowania"""
|
|
#lsof -i:12345
|
|
#kill -9 $(lsof -t -i:12345)
|
|
# sudo kill $(sudo lsof -t -i:8000)
|
|
pass
|
|
|
|
|
|
class Client:
|
|
def __init__(self, target_host="127.0.0.1", target_port=12346, recv_len=SOCKET_DATA_LENGTH):
|
|
"""Tworzy obiket socketa dla klienta"""
|
|
self.recv_len = recv_len
|
|
self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # utworzenie obiektu gniazda
|
|
self.client_socket.connect((target_host, target_port)) # polaczenie z klientem
|
|
|
|
def get_message_from_server(self):
|
|
"""Pobiera wiadomosc od serwera i ja wysiwetla, tak długo jak są jakieś dane"""
|
|
|
|
response = self.client_socket.recv(SOCKET_DATA_LENGTH)
|
|
print(response)
|
|
|
|
|
|
def send_message_to_server(self, msg):
|
|
self.client_socket.send(msg)
|
|
self.get_message_from_server()
|
|
|
|
def request_for_file_list(self):
|
|
"""Prosi serwer o liste dostepnych plikow"""
|
|
self.send_message_to_server("[1]")
|
|
|
|
|
|
def get_file_from_server(self, filename):
|
|
"""Pobiera i zapisuje plik wybrany przez użytkownika od serwera"""
|
|
self.send_message_to_server("[2]<{filename}>".format(filename=filename))
|
|
self.get_message_from_server()
|
|
|
|
|
|
def end_connection(self):
|
|
print("Zamykanie obiektu klienta")
|
|
self.send_message_to_server("[3]")
|
|
sleep(0.1)
|
|
|
|
try:
|
|
self.client_socket.close()
|
|
except BaseException:
|
|
pass
|
|
|
|
|
|
def menu():
|
|
"""Wyświetla klientowi menu z dostpenymi opcjami jakie moze żadać od serwera"""
|
|
|
|
print( """
|
|
[1] Wyświetl listę plików z serwera\n
|
|
[2] Pobierz plik; wymagana <nazwa_pliku>\n
|
|
[3] Zakończ komunikację\n
|
|
""")
|
|
|
|
|
|
def setup():
|
|
""" Pobiera od usera z inputu dane do stworzenia socketu klienta - tworzy obiekt klienta"""
|
|
server_address = str(input("Podaj adres serwera: "))
|
|
server_port = int(input("Podaj port serwera: "))
|
|
|
|
try:
|
|
client = Client(target_host=server_address, target_port=int(server_port))
|
|
client.get_message_from_server() # serwer wita
|
|
return client
|
|
except BaseException as exception:
|
|
print("Nie można utworzyć połączenia z serwerem: {0}:{1}.\n\tBłąd{2}".format(
|
|
server_address, server_port, exception
|
|
))
|
|
exit(-1)
|
|
|
|
|
|
def main():
|
|
"""
|
|
Głoówny program i pętla klienta, zakańcza się gdy ten wybierze opcje `3`
|
|
"""
|
|
client = setup() #tworzenie obiektu klietna oraz połączenia
|
|
option = None
|
|
|
|
while option != 3: #pętla dopóki klient nie wybierze opcji 3
|
|
if option not in ['1','2','3']:
|
|
menu()
|
|
|
|
if option == '1':
|
|
print(1)
|
|
client.request_for_file_list()
|
|
option = None # żeby dalej zostać w pętli
|
|
|
|
if option == '2':
|
|
print(2)
|
|
filename = str(input("Podaj nazwe pliku: "))
|
|
client.get_file_from_server(filename)
|
|
option = None # żeby dalej zostać w pętli
|
|
|
|
if option == '3':
|
|
print(3)
|
|
client.end_connection() # usuwa socket i cały obiekt klienta
|
|
exit(1)
|
|
|
|
print("option: ", option)
|
|
option = str(input("Wybierz opcje: ")) # żeby dalej w pętli pobierac opcje od klienta
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |