/
client.py
141 lines (116 loc) · 4.88 KB
/
client.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
import sys
import Participant
import socket
import pickle
import threading
import re
import time
# Define a dictionary of available bots:
available_bots = {
'alice': Participant.Alice(),
'bob': Participant.Bob(),
'batman': Participant.Batman(),
'james': Participant.James(),
'yoda': Participant.Yoda()
}
# Force user to specify ip and port
try:
ip = sys.argv[1]
if ip == '--help' or ip == '-h':
print("To start a client that should connect to specified ip and port:\n"
"$python3 client.py <u>ip</u> <u>port</u> [ <u>bot_name</u> ]\n\n"
"> -b <u>bot_name</u>: specify the bot you want this client to be from the list of available bots\n"
">> Current available bots: Alice, Bob, Eve\n\n"
"> Example usage (connecting as a person to localhost:2410):\n"
">> $python3 client.py localhost 2410\n\n"
"> Example usage (connecting as a bot (Alice) to localhost:2410):\n"
">> $python3 client.py localhost 2410 alice")
sys.exit()
port = int(sys.argv[2])
except (IndexError, ValueError):
print("IP and port must be specified. Correct usage e.g $py client.py localhost 2410")
sys.exit()
# Define the participant:
try:
participant = \
available_bots[sys.argv[3].lower()] \
if len(sys.argv) == 4 \
else Participant.Person(input("Choose a nickname: "))
except KeyError:
print(f"Sorry, cannot summon {sys.argv[3]} :( but you could try one of these bots:\n"
f"{list(available_bots.keys())}")
sys.exit()
if isinstance(participant, Participant.Batman):
who = participant.ACTIONS['who']
identify = participant.ACTIONS['identify']
# Start the TCP connection:
client = socket.socket()
try:
client.connect((ip, port))
except ConnectionRefusedError:
print(f"Cannot establish connection with server at {ip}:{port}")
sys.exit()
# Send the server about who has connected:
client.send(pickle.dumps(participant))
# So, we want to be able to receive the broadcast message:
def receive_data():
while True:
try:
# We will have the server send back the participant and the message that they sent:
sender, message = pickle.loads(client.recv(1024))
# Then we will print the message from the sender:
if sender.name == "server":
print(f"{message}")
elif sender.name != participant.name:
print(f"{sender.name}: {message}")
# Then we check if the sender is a person or a bot
if isinstance(participant, Participant.Bot) and isinstance(sender, Participant.Person):
# If this client is a bot and the sender is a person then we will have the bot respond to the message:
response = participant.respond_to(message)
print(f"{participant.name}: {response}")
data = pickle.dumps((participant, response))
client.send(data)
if isinstance(participant, Participant.Batman) and \
(response.__contains__(who) or response.__contains__(identify)):
client.send(pickle.dumps((participant, "/update_name Botman")))
update_name('Botman')
except (ConnectionResetError, OSError, EOFError):
print("Disconnected from the server")
client.close()
break
# We also want to be able to send data. Persons are the only ones able to send messages
def send_data():
while True:
try:
# Sending empty string to check if we are still connected to the server:
client.send(pickle.dumps((participant, "")))
message = re.sub("[ ]+", " ", input().strip()) # clean up the message
# This function performs a command if the message is a command, and nothing else otherwise
if message == "/logout":
client.close()
break
perform_command(message)
client.send(pickle.dumps((participant, message)))
except (ConnectionAbortedError, EOFError, OSError):
print("Cannot establish connection with the server")
client.close()
break
time.sleep(.2)
# At last we have to start threads to run both functions simultanously:
threading.Thread(target=receive_data).start()
# Since only persons are able to send messages:
if isinstance(participant, Participant.Person):
threading.Thread(target=send_data, daemon=True).start()
# We also want certain commands to be performed:
def perform_command(message):
command = message.split(" ") # create an array of the message
if command[0] in commands:
commands[command[0]](" ".join(map(str, command[1:])))
def update_name(new_name):
old_name = participant.name
participant.name = new_name
return old_name
commands = {
'/update_name': update_name
}
sys.exit()