class Test:
def __init__(self):
engine = create_engine('sqlite:///:memory:', echo=False)
# Не забываем создать структуру базы данных
Base.metadata.create_all(engine)
# Создаем сессию для работы
Session = sessionmaker(bind=engine)
session = Session()
# Рекомендуется брать 1 сессию и передавать параметром куда нам надо
self.session = session
# далее создаем тестовые
client = Client('Vasya')
self.session.add(client)
self.repo = Storage(session)
# def test(self):
#
# engine = create_engine('sqlite:///:memory:', echo=False)
# # Не забываем создать структуру базы данных
# Base.metadata.create_all(engine)
# # Создаем сессию для работы
# Session = sessionmaker(bind=engine)
# session = Session()
# # Рекомендуется брать 1 сессию и передавать параметром куда нам надо
# self.session = session
# # далее создаем тестовые
#
# client = Client('Vasya')
# self.session.add(client)
#
# self.repo = Storage(session)
def test_add_client(self):
self.repo.add_client('New Ivan')
def test_get_client(self, name):
client = self.repo.get_client(name)
return client
from PyQt5 import QtWidgets
import sys
import py_server_monitor
from db.db_lib import Storage
from db.db import session
app = QtWidgets.QApplication(sys.argv)
window = QtWidgets.QMainWindow()
ui = py_server_monitor.Ui_MainWindow()
ui.setupUi(window)
repo = Storage(session)
repo.add_client('Kolya')
repo.add_client('Vasya')
repo.add_client('Shakira')
repo.add_contact('Kolya', 'Vasya')
def load_clients():
clients = repo.get_clients()
ui.listWidget.clear()
for clint in clients:
ui.listWidget.addItem(str(clint))
load_clients()
ui.pushButton.clicked.connect(load_clients)
class Server():
def __init__(self):
self.clients = []
self.jim = JimProtocol()
self.clients = []
self.who_reads = []
self.who_writes = []
self.e = []
self.repo = Storage(session)
self.from_name = ''
self.user_and_sock = {}
def action(self, msg):
if 'action' in msg and \
msg['action'] == 'msg':
print(msg)
print('msg')
# name = msg.get('from')
# message = msg.get('message')
answer_msg = '{} {} send message to all: {}'\
.format(time.ctime(), msg.get('from'), msg.get('message'))
self.from_name = msg.get('from')
return msg
elif 'action' in msg and msg['action'] == 'join':
client_name = msg.get('from')
contact_name = msg.get('name')
self.repo.add_contact(client_name, contact_name)
print('user: {} add to {}'.format(client_name, contact_name))
return None
elif 'action' in msg and msg['action'] == 'leave':
client_name = msg.get('client')
contact_name = msg.get('contact')
self.repo.del_contact(client_name, contact_name)
print('client: {} del {} from friend list'.format(
client_name, contact_name))
return None
elif 'action' in msg and msg['action'] == 'remove':
client_name = msg.get('client')
self.repo.remove_client(client_name)
print('{} is removed'.format(client_name))
return None
elif 'action' in msg and msg['action'] == 'create':
client_name = msg.get('client')
self.repo.add_client(client_name)
print('New client: {}!'.format(client_name))
return None
elif 'action' in msg and msg['action'] == 'getClients':
# clients = self.user_and_sock.keys()
# print(type(clients))
# clients = self.repo.get_clients()
# clients_array = []
# for client in clients:
# client_str = str(client)
# client_str.join()
# client_str = client_str[10:-3]
# clients_array.append(client_str)
# msg = self.jim.set_clients(clients)
str = ' '.join(self.clients)
print(str)
msg = self.jim.set_clients(str)
return msg
# return clients_array
elif 'action' in msg and msg['action'] == 'getContacts':
contact = self.repo.get_contacts(msg.get('client_name'))
print(contact)
elif 'action' in msg and msg['action'] == 'msg_to_user':
pass
else:
return None
def reads_msg(self, w_clients, all_clients):
for sock in w_clients:
try:
jmsg = self.jim.recive_msg(sock)
msg = self.action(jmsg)
print('1')
if msg is not None:
self.write_clients(msg, self.who_reads, self.clients)
except:
sock.cloce()
all_clients.remove(sock)
def write_clients(self, msgs, r_clients, all_clients):
for sock in r_clients:
try:
print('3')
msg = self.jim.create_answer_msg('msg', self.from_name, msgs)
self.jim.send_msg(sock, msgs)
print(msgs)
except:
sock.cloce()
all_clients.remove(sock)
# for msg in msgs:
# try:
# self.jim.send_msg(sock, msg)
# print(msg)
# except:
# sock.cloce()
# all_clients.remove(sock)
def presence_answer(self, presence_message):
if 'action' in presence_message and \
presence_message['action'] == 'presence':
answer = self.jim.presence_answer({'response': 200})
return answer
else:
answer = self.jim.presence_answer({
'response': 400,
'error': 'Не верный запрос'
})
return answer
def main_loop_in_server(self):
while True:
try:
sock, addr = serv_sock.accept()
presence = self.jim.recive_msg(sock)
self.repo.add_client(presence.get('user'))
self.user_and_sock[presence.get('user')] = sock
self.clients.append(presence.get('user'))
print(self.user_and_sock)
answer = self.presence_answer(presence)
self.jim.send_msg(sock, answer)
except OSError as e:
pass
else:
# self.clients.append(sock)
# presence = self.jim.recive_msg(sock)
# # #print(presence.get('user'))
# repo.add_client(presence.get('user'))
# answer = self.presence_answer(presence)
# self.jim.send_msg(sock, answer)
print('Client with ip {}'.format(addr))
finally:
# wait = 0
# self.who_writes = []
# self.who_reads = []
try:
self.who_writes, \
self.who_reads,\
self.e = \
select.select(self.user_and_sock.values(),
self.user_and_sock.values(),
self.user_and_sock.values(), 0)
self.reads_msg(self.who_writes, self.clients)
# self.write_clients(msg, self.who_reads, self.clients)
except:
pass