def connect_to_server(self):
self.server_handler = PC("198.211.120.146").ident(self.get_json_dict())
self.photo_json_list = json.loads(self.server_handler.photo_data)
self.btn_accept.show()
self.btn_reject.show()
self.btn_connect.hide()
self.update_photo()
self.btn_accept.setDisabled(False)
self.btn_reject.setDisabled(False)
self.btn_connect.setDisabled(True)
self.thread = MessageThread(self.chatbox, self.server_handler)
self.thread.message.connect(self.append_messages, Qt.QueuedConnection)
self.handletoggle()
class CryptoPear(QWidget):
def __init__(self):
super(CryptoPear, self).__init__()
self.initUI()
def listdir(self):
path = os.getcwd() + '/test_images/'
files = os.listdir(path)
return map(lambda f: path + f, files)
def random_path(self):
return self.paths[random.randint(0, len(self.paths) - 1)]
def initUI(self):
self.paths = self.listdir()
self.photo_json_list = []
self.accept_reject_dict = {}
self.default_image = self.paths.pop(0)
pixmap = QPixmap(self.default_image).scaledToHeight(200)
self.lbl = QLabel(self)
self.lbl.setPixmap(pixmap)
self.name_label = QLabel(self)
self.btn_connect = QPushButton("Connect")
self.btn_connect.clicked.connect(self.connect_to_server)
self.btn_accept = QPushButton("Yes")
self.btn_reject = QPushButton("No")
self.btn_accept.clicked.connect(self.accept_participent)
self.btn_reject.clicked.connect(self.reject_participent)
self.btn_accept.setDisabled(True)
self.btn_reject.setDisabled(True)
self.btn_accept.hide()
self.btn_reject.hide()
hbox = QHBoxLayout()
hbox.addStretch(1)
hbox.addWidget(self.btn_connect)
hbox.addWidget(self.btn_accept)
hbox.addWidget(self.btn_reject)
self.chatbox = QTextEdit(self)
self.chatbox.setMinimumHeight(400)
self.chatbox.setReadOnly(True)
self.text_entry = QTextEdit(self)
self.text_entry.installEventFilter(self)
self.btn_submit = QPushButton("Submit")
self.btn_submit.clicked.connect(self.submit_message)
self.btn_submit.setDisabled(True)
text_hbox = QHBoxLayout()
text_hbox.addWidget(self.text_entry)
text_hbox.addWidget(self.btn_submit)
vbox = QVBoxLayout()
vbox.addStretch(1)
vbox.addWidget(self.lbl)
vbox.addWidget(self.name_label)
vbox.addLayout(hbox)
vbox.addWidget(self.chatbox)
vbox.addLayout(text_hbox)
self.setLayout(vbox)
#starting x, starting y, width, height
self.setGeometry(0, 0, 500, 700)
self.setWindowTitle('CryptoPear')
self.show()
def eventFilter(self, obj, event):
if obj == self.text_entry:
if isinstance(event, QKeyEvent):
if event.key() == Qt.Key_Return:
if event.type() == QEvent.KeyPress:
self.submit_message()
return True
return False
def get_camera(self):
try:
cam = Camera()
img = cam.getImage()
img.save("mZOMGGUYS.png")
except:
pass
return voting.encode_image(self.random_path())
def get_user(self):
text, ok = QInputDialog.getText(self, 'Input Dialog',
'Enter your name:')
if ok:
return text
else:
return None
def get_json_dict(self):
data = { 'name':self.get_user(), 'photo':self.get_camera()}
return data
def connect_to_server(self):
self.server_handler = PC("198.211.120.146").ident(self.get_json_dict())
self.photo_json_list = json.loads(self.server_handler.photo_data)
self.btn_accept.show()
self.btn_reject.show()
self.btn_connect.hide()
self.update_photo()
self.btn_accept.setDisabled(False)
self.btn_reject.setDisabled(False)
self.btn_connect.setDisabled(True)
self.thread = MessageThread(self.chatbox, self.server_handler)
self.thread.message.connect(self.append_messages, Qt.QueuedConnection)
self.handletoggle()
def append_messages(self, messages):
if messages:
self.chatbox.setTextColor('red')
self.chatbox.setReadOnly(False)
self.chatbox.clear()
self.chatbox.append(messages)
self.chatbox.setReadOnly(True)
def handletoggle(self):
if self.thread.isRunning():
self.thread.exiting=True
else:
self.thread.exiting=False
self.thread.start()
def submit_message(self):
message = self.text_entry.toPlainText().encode('ascii')
if message:
self.server_handler.encrypted_send(message+'\n')
self.text_entry.clear()
def set_picture(self, file_path):
image = QImage(file_path)
if image.isNull():
QMessageBox.information(self, "Image Viewer",
"Cannot load file")
return
self.lbl.setPixmap(QPixmap.fromImage(image).scaledToHeight(200))
self.lbl.adjustSize()
def update_photo(self):
self.user_data = self.photo_json_list.pop(0)
voting.decode_image(self.user_data['photo'], self.user_data['name'])
path = os.getcwd() + '/test_images/' + self.user_data['name'] + '.jpeg'
print path
self.set_picture(path)
self.name_label.setText('Is this ' + self.user_data['name'] + '?')
def accept_participent(self):
self.accept_reject_dict[self.user_data['id']] = True
if self.photo_json_list:
self.update_photo()
else:
self.server_handler.vote_result(json.dumps(self.accept_reject_dict))
self.set_picture(self.default_image)
self.btn_accept.setDisabled(True)
self.btn_reject.setDisabled(True)
self.btn_submit.setDisabled(False)
self.btn_accept.hide()
self.btn_reject.hide()
self.btn_connect.hide()
self.name_label.setText("")
def reject_participent(self):
self.accept_reject_dict[self.user_data['id']] = False
if self.photo_json_list:
self.update_photo()
else:
print self.user_data
self.server_handler.vote_result(json.dumps(self.accept_reject_dict))
self.set_picture(self.default_image)
self.btn_accept.setDisabled(True)
self.btn_reject.setDisabled(True)
self.btn_submit.setDisabled(False)
self.btn_accept.hide()
self.btn_reject.hide()
self.btn_connect.hide()
self.name_label.setText("")