class TuringChat():
def __init__(self, uiPrintMessage, sendQuit=None):
self.settings = Settings(self.printMessage)
self.turing = Backend()
self.client = None
self.server = None
self.uiPrintMessage = uiPrintMessage
self.sendQuit = sendQuit
self.connexion = None
self.server = Server(self.turing, self.printMessage,
self.writeMessages)
self.client = Client(self.turing, self.printMessage,
self.writeMessages)
self.username = self.settings.getSetting("username")
self.otherUsername = None
self.trustManager = TrustManager(self.printMessage)
self.msgBuffer = []
self.fileList = []
self.translate = Translate(self.printMessage,
self.settings.getSetting("language"))
setObject(self.translate)
def printMessage(self, text, message=False, username=None):
if username:
self.otherUsername = username
return self.printMessage(
self.translate.tr("username.other.changed") + username)
if message and not self.trustManager.connexionTrusted():
self.msgBuffer.append((text, message, username))
else:
self.uiPrintMessage(text, message, username)
if self.trustManager.connexionTrusted():
for element in self.msgBuffer:
self.uiPrintMessage(element[0], element[1], element[2])
self.msgBuffer = []
def writeMessages(self, connexion, fingerprint=None):
self.connexion = connexion
if self.connexion:
self.connexion.setTuringChat(self)
if fingerprint:
self.trustManager.setCurrentFingerprint(fingerprint)
if self.username and self.connexion:
self.connexion.send(
self.turing.createMessage("username", self.username))
def startServer(self):
self.stopConnexions()
self.server.listen()
self.server.start()
def stopServer(self):
if self.server.listening() or self.server.isStopped():
if not self.server.isStopped():
self.server.stop()
self.server.join()
self.server = Server(self.turing, self.printMessage,
self.writeMessages)
def startClient(self, addr="127.0.0.1"):
self.stopConnexions()
self.client.connect(addr)
def stopClient(self):
if self.client.connected():
self.client.close()
def stopConnexions(self):
self.stopClient()
self.stopServer()
def setUsername(self, username):
self.username = username
self.settings.setSetting("username", username)
if self.connexion:
self.connexion.send(self.turing.createMessage(
"username", username))
self.printMessage(self.translate.tr("username.changed") + username)
def connect(self, host="127.0.0.1"):
if host == "last":
host = self.settings.getSetting("lastHost")
self.settings.setSetting("lastHost", host)
self.startClient(host)
def sendMessage(self, message):
if not self.connexion:
self.printMessage(self.translate.tr("error.not.connected"))
return
if not self.trustManager.connexionTrusted():
self.printMessage(self.translate.tr("error.connexion.not.trusted"))
return
self.printMessage(self.translate.tr("user.you") + message)
self.connexion.send(self.turing.createMessage("message", message))
def quit(self):
self.stopClient()
self.stopServer()
if self.sendQuit:
self.sendQuit()
return "quit"
def setLanguage(self, language):
self.translate.setLanguage(language)
self.settings.setSetting("language", language)
self.printMessage(self.translate.tr("language.set"))
def parseCommand(self, command):
parsing.listen.setParseAction(self.startServer)
parsing.connect.setParseAction(lambda arg: self.connect(arg[1]))
parsing.quit.setParseAction(self.quit)
parsing.help.setParseAction(
lambda: self.printMessage(self.translate.tr("command.help.text")))
parsing.nick.setParseAction(lambda arg: self.setUsername(arg[1]))
parsing.trust.setParseAction(
lambda arg: self.trustManager.setTrust(arg[1]))
parsing.fingerprint.setParseAction(
lambda: self.printMessage(self.turing.getMyFingerprint()))
parsing.language.setParseAction(lambda arg: self.setLanguage(arg[1]))
parsing.file.setParseAction(lambda arg: self.saveFile(arg[2]) if arg[
1] == "download" else arg[1] == "upload" and self.sendFile(arg[2]))
try:
result = parsing.commands.parseString(command, True)
if result[1] == "quit":
return "quit"
except pyparsing.ParseException:
self.sendMessage(command)
def addFile(self, file):
self.fileList.append(file)
def saveFile(self, fileId):
fileId = int(fileId)
if fileId < len(self.fileList):
fileContent = self.fileList[fileId]
with open("file-" + str(fileId), "wb") as f:
f.write(fileContent)
self.printMessage(self.translate.tr("file.saved"))
else:
self.printMessage(self.translate.tr("error.file.notfound"))
def sendFile(self, filename):
if os.path.isfile(filename):
with open(filename, "rb") as f:
data = f.read()
self.connexion.send(self.turing.createMessage("file", data))
self.printMessage(self.translate.tr("file.sent"))
else:
self.printMessage(self.translate.tr("error.file.notfound"))
class PluginMapotempo:
"""QGIS Plugin Implementation."""
def __init__(self, iface):
"""Constructor.
:param iface: An interface instance that will be passed to this class
which provides the hook by which you can manipulate the QGIS
application at run time.
:type iface: QgsInterface
"""
self.iface = iface
self.translate = Translate()
self.running = False
# Declare instance attributes
self.actions = []
self.menu = self.translate.tr(u'&Plugin Mapotempo')
self.toolbar = self.iface.addToolBar(u'PluginMapotempo')
self.toolbar.setObjectName(u'PluginMapotempo')
self.iface.addToolBarWidget(self.toolbar)
self.dlg = PluginMapotempoDialogBase()
self.dlg_2 = CreateZoning()
self.dock = DockWidget()
#instance
self.layer_inst = PluginMapotempoLayer(
self.dlg, self.dock, self.iface, self.translate)
self.handle = PluginMapotempoHandle(
self.layer_inst, self.dlg, self.dlg_2, self.dock, self.translate)
self.layer_inst.setHandler(self.handle)
self.dock.setHandler(self.handle)
self.dlg.pushButton.clicked.connect(self.handle.handleButtonSave)
self.dlg_2.pushButton.clicked.connect(self.handle.handleButtonNewZoning)
self.dock.pushButton.clicked.connect(self.handle.handleButtonConnect)
self.dock.pushButton_2.clicked.connect(self.layer_inst.refresh)
self.dock.pushButton_4.clicked.connect(self.handle.HandleParam)
self.dock.pushButton_5.clicked.connect(self.handle.newZoningLayer)
self.dock.comboBox.activated.connect(self.handle.HandleSelect)
self.dock.pushButton_3.clicked.connect(self.handle.HandleApplyZoning)
self.iface.addDockWidget(Qt.RightDockWidgetArea, self.dock)
self.handle.keyConnection = self.handle.s.value("PluginMapotempo/key")
self.handle.hostConnection = self.handle.s.value("PluginMapotempo/host")
def add_action(
self,
icon_path,
text,
callback,
enabled_flag=True,
add_to_menu=True,
add_to_toolbar=True,
status_tip=None,
whats_this=None,
parent=None):
"""Add a toolbar icon to the toolbar.
:param icon_path: Path to the icon for this action. Can be a resource
path (e.g. ':/plugins/foo/bar.png') or a normal file system path.
:type icon_path: str
:param text: Text that should be shown in menu items for this action.
:type text: str
:param callback: Function to be called when the action is triggered.
:type callback: function
:param enabled_flag: A flag indicating if the action should be enabled
by default. Defaults to True.
:type enabled_flag: bool
:param add_to_menu: Flag indicating whether the action should also
be added to the menu. Defaults to True.
:type add_to_menu: bool
:param add_to_toolbar: Flag indicating whether the action should also
be added to the toolbar. Defaults to True.
:type add_to_toolbar: bool
:param status_tip: Optional text to show in a popup when mouse pointer
hovers over the action.
:type status_tip: str
:param parent: Parent widget for the new action. Defaults None.
:type parent: QWidget
:param whats_this: Optional text to show in the status bar when the
mouse pointer hovers over the action.
:returns: The action that was created. Note that the action is also
added to self.actions list.
:rtype: QAction
"""
icon = QIcon(icon_path)
action = QAction(icon, text, parent)
action.triggered.connect(callback)
action.setEnabled(enabled_flag)
if status_tip is not None:
action.setStatusTip(status_tip)
if whats_this is not None:
action.setWhatsThis(whats_this)
if add_to_toolbar:
self.toolbar.addAction(action)
if add_to_menu:
self.iface.addPluginToMenu(
self.menu,
action)
self.actions.append(action)
return action
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
icon_path = ':/plugins/mapotempo-qgis-plugin/icons/icon.png'
self.add_action(
icon_path,
text=self.translate.tr(u'Plugin Mapotempo'),
callback=self.run,
parent=self.iface.mainWindow())
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
for action in self.actions:
self.iface.removePluginMenu(
self.translate.tr(u'&Plugin Mapotempo'),
action)
self.iface.removeToolBarIcon(action)
# remove the toolbar
del self.toolbar
self.iface.removeDockWidget(self.dock)
def run(self):
"""Run method that performs all the real work"""
self.iface.addDockWidget(Qt.RightDockWidgetArea, self.dock)
