コード例 #1
0
ファイル: MainWindow.py プロジェクト: mohanrex/whatsdesk
class MainWindow(QMainWindow, mainapp_ui.Ui_MainWindow):
    def __init__(self):
        QMainWindow.__init__(self)
        self.db = Helper()
        self.credentials = self.db.get_credentials()
        self.setupUi(self)
        if self.credentials:
            self.form_widget = None
            self.frame_layout = QtWidgets.QVBoxLayout(self.message_frame)
            self.message_frame.setContentsMargins(0, 0, 0, 0)
            self.thread = YowsupThread(self.credentials['phone_number'], self.credentials['password'], debug=False)
            self.thread.interface.success_connection_signal.connect(self.on_success)
            self.thread.interface.message_received_signal.connect(self.on_success)
            self.thread.interface.presence_update_signal.connect(self.on_success)
            self.contact_table.clicked.connect(self.render_page)
            self.thread.render()
        else:
            print("You need to provide authentication credentials")

    def render_page(self, index):
        row = index.row()
        self.form_widget = Messager(index.sibling(row, 1).data(), self.thread)
        self.form_widget.message_received_signal.connect(self.on_success)
        for i in reversed(range(self.frame_layout.count())):
            self.frame_layout.itemAt(i).widget().setParent(None)
        self.frame_layout.addWidget(self.form_widget)
        self.frame_layout.setContentsMargins(0, 0, 0, 0)
        self.message_frame.setLayout(self.frame_layout)
        self.update_contact_table()

    def on_success(self):
        self.update_contact_table(True)
        pass

    def update_contact_table(self, init=False):
        if init:
            self.initialize()
        result = self.db.get_all_contact()
        self.contact_table.setRowCount(result['count'])
        for idx, record in enumerate(result['records']):
            form_widget = ContactRow(record)
            self.contact_table.setItem(idx, 1, QTableWidgetItem(str(record[0])))
            self.contact_table.setCellWidget(idx, 0, form_widget)
        return

    def initialize(self):
        self.contact_table.setColumnCount(2)
        self.contact_table.setColumnHidden(1, True)
        self.contact_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.contact_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
        self.contact_table.setSelectionBehavior(QTableView.SelectRows)
        self.contact_table.setSelectionMode(QAbstractItemView.SingleSelection)
        self.contact_table.verticalHeader().setVisible(False)
        self.contact_table.horizontalHeader().setVisible(False)
        self.contact_table.verticalHeader().setDefaultSectionSize(68)
コード例 #2
0
ファイル: MainWindow.py プロジェクト: mohanrex/whatsdesk
 def __init__(self):
     QMainWindow.__init__(self)
     self.db = Helper()
     self.credentials = self.db.get_credentials()
     self.setupUi(self)
     if self.credentials:
         self.form_widget = None
         self.frame_layout = QtWidgets.QVBoxLayout(self.message_frame)
         self.message_frame.setContentsMargins(0, 0, 0, 0)
         self.thread = YowsupThread(self.credentials['phone_number'], self.credentials['password'], debug=False)
         self.thread.interface.success_connection_signal.connect(self.on_success)
         self.thread.interface.message_received_signal.connect(self.on_success)
         self.thread.interface.presence_update_signal.connect(self.on_success)
         self.contact_table.clicked.connect(self.render_page)
         self.thread.render()
     else:
         print("You need to provide authentication credentials")