def get_connection(self):
try:
conn, addr = self.socket.accept()
presence_msg_bytes = conn.recv(1024)
presence_msg = JimMessage.create_from_bytes(presence_msg_bytes)
if presence_msg.action == PRESENCE:
presence_response = JimResponse(**{RESPONSE: OK})
conn.send(bytes(presence_response))
else:
presence_response = JimResponse(**{RESPONSE: WRONG_REQUEST})
conn.send(bytes(presence_response))
except OSError as e:
pass
else:
print("Получен запрос на соединение от {}".format(str(addr)))
self.clients.append(conn)
finally:
wait = 0
read = []
write = []
try:
read, write, e = select.select(self.clients, self.clients, [], wait)
except:
pass
requests = self.read_requests(read)
self.write_responses(requests, write)
def main_loop(self):
presence_msg = JimMessage(action=PRESENCE, time=time.time())
self.socket.send(bytes(presence_msg))
presence_response_bytes = self.socket.recv(1024)
presence_response = JimResponse.create_from_bytes(
presence_response_bytes)
if presence_response.response == OK:
print('Связь с сервером установлена')
if self.mode == 'r':
while True:
message_bytes = self.socket.recv(1024)
jimmsg = JimMessage.create_from_bytes(message_bytes)
print(
'Вы получили от {} {} сообщеие'.format(
self.socket.fileno(), self.socket.getpeername()),
jimmsg.message)
elif self.mode == 'w':
while True:
message = input('Пошлите сообщение в никуда =')
msg = JimMessage(action=MSG,
time=time.time(),
encoding='utf-8',
message=message)
self.socket.send(bytes(msg))
else:
raise WrongModeError(mode)
elif presence_response.response == SERVER_ERROR:
print('Ошибка сервера')
elif presence_response.response == WRONG_REQUEST:
print('Неверный запрос на сервер')
else:
print('Неверный код ответа от сервера')
def read_requests(self, read_clients):
all_messages = []
for sock in read_clients:
try:
bytemsg = sock.recv(1024)
jimmsg = JimMessage.create_from_bytes(bytemsg)
all_messages.append(jimmsg)
except:
print('Клиент {} {} отключился'.format(sock.fileno(), sock.getpeername()))
self.clients.remove(sock)
return all_messages
def _get_connection(self):
try:
conn, addr = self.socket.accept() # Проверка подключений
# Должно прийти сообщение о присутствии
presence_msg_bytes = conn.recv(1024)
presence_msg = JimMessage.create_from_bytes(presence_msg_bytes)
if presence_msg.action == PRESENCE:
# Получаем имя пользователя
client_name = presence_msg.user[ACCOUNT_NAME]
print('К нам подключился {}'.format(client_name))
# если клиента нету в базе
if not self.repo.client_exists(client_name):
# мы его добавляем
print('Добавляем клиента')
self.repo.add_client(client_name)
print('Сохраняем')
self.repo.commit()
# добавляем историю подключения
self.repo.add_history(client_name, addr[0])
self.repo.commit()
# отправляем ответ
presence_response = JimResponse(**{RESPONSE: OK})
conn.sendall(bytes(presence_response))
else:
presence_response = JimResponse(**{RESPONSE: WRONG_REQUEST})
conn.sendall(bytes(presence_response))
except OSError as e:
pass # timeout вышел
else:
print("Получен запрос на соединение от %s" % str(addr))
# Добавляем клиента в список
self._clients.append(conn)
# Добавляем в словарь имя клиента и соответствующий ему сокет
# Мы это делаем, чтобы знать в будующем кому пересылать сообщение
self.names[client_name] = conn
finally:
# Проверить наличие событий ввода-вывода
wait = 0
try:
r, w, e = select.select(self._clients, self._clients, [], wait)
requests = self._read_requests(r) # Получаем входные сообщения
self._write_responses(
requests) # Выполним отправку входящих сообщений
except:
pass
def _read_requests(self, r_clients):
"""
Чтение сообщений, которые будут посылать клиенты
:param r_clients: клиенты которые могут отправлять сообщения
:return: список сообщений
"""
# Список входящих сообщений
messages = []
for sock in r_clients:
# Получаем входящие сообщения
bdata = sock.recv(2**20)
if bdata:
jm = JimMessage.create_from_bytes(bdata)
# Добавляем в список пару сообщение и сокет который его прислал
messages.append((jm, sock))
else:
print("Клиент отключился")
sock.close()
self._clients.remove(sock)
# Возвращаем словарь сообщений
return messages
def poll(self):
self.is_alive = True
while True:
if not self.is_alive:
break
data = self.sock.recv(2**20)
if data:
try:
# Если нам пришло сообщение
jm = JimMessage.create_from_bytes(data)
# Если это message
if MESSAGE in jm:
# Печатаем в нормальном виде
self.process_message(jm)
else:
# Добавляем сообщение в очередь т.к. это серверное сообщение
self.request_queue.put(jm)
except MandatoryKeyError:
# Если нам пришел ответ от сервера мы его добавляем в очередь для дальнейшей обработки
jr = JimResponse.create_from_bytes(data)
# При этом поток приостанавливается
self.request_queue.put(jr)
else:
break