PythonClientServerApplications/client_gui.py at master · vmikulitskii/PythonClientServerApplications · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import datetime
import sys

from PyQt5 import QtWidgets
from PyQt5.QtWidgets import QMainWindow, QDialog
from PyQt5.QtGui import QStandardItemModel, QStandardItem, QColor
from PyQt5.QtCore import Qt, pyqtSignal, pyqtSlot
from client_storage import ClientStorage

from client_gui_main_ui import Ui_MainWindow
from client_gui_arrived_message_ui import Ui_newMesaageDialog
from client_gui_new_localcontact_ui import Ui_addNewLocalContactDialog
from client_gui_log_pass_ui import Ui_loginPasswrdDialog
from common.variables import SENT, RECEIVED


class MyWindow(QMainWindow, Ui_MainWindow):
    contact_selected = pyqtSignal(QtWidgets.QListWidgetItem)

    def view_contacts(self):
        """
        Загружаем и показываем список контактов
        :return:
        """
        contacts = self.client_db.get_contacts()
        # model = QStandardItemModel()
        # self.listContacts.setModel(model)
        for contact in contacts:
            self.listContacts.addItem(contact.name)
            # item = QStandardItem(contact.name)
            # item.setBackground(QColor('#ffff99'))
            # model.appendRow(item)

    def __init__(self, database):
        super().__init__()
        self.setupUi(self)
        self.actionExit.triggered.connect(QtWidgets.qApp.exit)
        self.client_db = database
        self.activ_contact_name = None

        self.listContacts.itemDoubleClicked.connect(self.on_change_contact)

    def on_change_contact(self, value):
        self.contact_selected.emit(value)

    def load_last_history(self, user_name):
        """
        Загружаем переписку за последние сутки с определенным контактом
        и показываем в нашей Qtableview
        :param user_name: str
        :return:
        """
        last_messages = self.client_db.get_history(user_name)
        model = QStandardItemModel()
        self.tableMessage.setModel(model)
        last_msg_status = ''
        for msg in last_messages:
            time = datetime.datetime.strftime(msg.date_time, '%H:%M')
            user_name = 'Я:' if msg.status == SENT else f'{msg.Contact.name}:'
            if last_msg_status == msg.status:
                parts = [time, msg.text]
            else:
                parts = [user_name, time, msg.text]
            for part in parts:
                item = QStandardItem(part)
                item.setEditable(False)
                if msg.status == SENT:
                    item.setTextAlignment(Qt.AlignRight)
                    item.setBackground(QColor('#c1e8ce'))
                elif msg.status == RECEIVED:
                    item.setTextAlignment(Qt.AlignLeft)
                    item.setBackground(QColor('#fceaca'))
                model.appendRow([item])
            last_msg_status = msg.status

        self.tableMessage.horizontalHeader().setDefaultSectionSize(490)
        self.tableMessage.resizeRowsToContents()

        self.tableMessage.setShowGrid(False)
        self.tableMessage.horizontalHeader().setVisible(False)
        self.tableMessage.verticalHeader().setVisible(False)
        self.tableMessage.scrollToBottom()

    @pyqtSlot(QtWidgets.QListWidgetItem)
    def get_user_message(self, contact_obj):
        self.load_last_history((contact_obj.text()))
        self.activ_contact_name = contact_obj.text()
        print(f'Активирован контакт - {self.activ_contact_name}')

    def make_connection(self, contact_list):
        contact_list.itemDoubleClicked.connect(self.get_user_message)


class ArrivedMessage(QDialog, Ui_newMesaageDialog):

    def __init__(self):
        super().__init__()
        self.setupUi(self)


class NewLocalContact(QDialog, Ui_addNewLocalContactDialog):

    def __init__(self):
        super().__init__()
        self.setupUi(self)


class LoginPass(QDialog, Ui_loginPasswrdDialog):

    def __init__(self):
        super().__init__()
        self.setupUi(self)

        self.passEdit.setEchoMode(QtWidgets.QLineEdit.EchoMode.Password)

    def edit_clear(self):
        self.loginEdit.clear()
        self.passEdit.clear()


def main():
    client_db = ClientStorage('User-1')
    # client_db.add_contact('User-1')
    app = QtWidgets.QApplication(sys.argv)
    window = MyWindow(client_db)
    window.make_connection(window.listContacts)
    window.show()
    # window.load_last_history('User-8')
    am = ArrivedMessage('Vasia')
    am.show()

    window.view_contacts()

    sys.exit(app.exec_())


if __name__ == '__main__':
    main()