sendMessageForm.py

from PySide.QtCore import *
from PySide.QtGui import *
from PySide.QtUiTools import *
import plugin.databaseConnect as database
from datetime import datetime


class sendMessageUI(QMainWindow):
    def __init__(self, id = None, bulk = None, parent = None):
        QMainWindow.__init__(self,None)
        self.setMinimumSize(626,380)
        self.setWindowTitle("Message")
        self.parent = parent
        self.id = id
        self.bulk = bulk
        self.UIinit()

    def UIinit(self):
        loader = QUiLoader()
        form = loader.load("resources/UI/sendMessage.ui",None)
        self.setCentralWidget(form)

        #QPushButton
        self.send_button = form.findChild(QPushButton,"sendButton")
        self.close_button = form.findChild(QPushButton,"closeButton")
        

        #LineEdit
        self.to_user = form.findChild(QLineEdit,"to")
        self.message = form.findChild(QTextEdit,"message")

        #Connect
        self.send_button.clicked.connect(self.sendMes)
        self.close_button.clicked.connect(self.closeWindow)

        if(self.id != None):
            self.to_user.setText(self.id)

    def closeWindow(self):
        self.close()

    ##Create message and send it to other user##
    def sendMes(self):
        db = database.databaseMessage()
        toUser = self.to_user.text()
        message = self.message.toPlainText()
        time = datetime.now()
        if(self.bulk == None):
            data = self.parent.getCurrentUser()
            fromUser = data.getID()
            if(db.sendMessage(toUser, fromUser, message, time)):
                db.disconnect()
                self.parent.showOK("Message Sent", "The message has been sent to the user!")
                self.closeWindow()
            else:
                self.parent.showERROR("UserID Not Found", "The UserID you entered does not exists.")
        else:
            data = self.parent.parent.getCurrentUser()
            fromUser = data.getID()
            val = 0
            for id in self.bulk:
                val = db.sendMessage(id, fromUser, message, time)
            if (val):
                db.disconnect()
                self.parent.parentshowOK("All Message Sent to user.", "The message has been sent to all user!")
                self.closeWindow()
            else:
                self.parent.parent.showERROR("ERROR!", "Some Messages are not delivered.")