chat history support (partial)

src/history.py

@@ -113,11 +113,3 @@ class History(object):
 
         def __del__(self):
             self._db.close()
-
-
-if __name__ == '__main__':
-    h = History('test')
-    if not h.friend_exists_in_db('42'):
-        h.add_friend_to_db('42')
-        h.save_messages_to_db('42', [('bugara', 1, 123412413, 1)])
-    h.delete_messages('42')

src/mainscreen.py

@@ -229,6 +229,9 @@ class MainWindow(QtGui.QMainWindow):
         self.friend_info = info
         self.profile = Profile(self.tox, self)
 
+    def closeEvent(self, *args, **kwargs):
+        self.profile.save()
+
     # -----------------------------------------------------------------------------------------------------------------
     # Functions which called when user click in menu
     # -----------------------------------------------------------------------------------------------------------------
@@ -308,6 +311,7 @@ class MainWindow(QtGui.QMainWindow):
 
     def clear_history(self, item):
         num = self.friends_list.indexFromItem(item).row()
+        self.profile.clear_history(num)
 
 # -----------------------------------------------------------------------------------------------------------------
 # Functions which called when user click somewhere else

src/menu.py

@@ -270,6 +270,8 @@ class PrivacySettings(CenteredWidget):
         settings = Settings.get_instance()
         settings['typing_notifications'] = self.typingNotifications.isChecked()
         settings['allow_auto_accept'] = self.fileautoaccept.isChecked()
+        if settings['history'] and not self.saveHistory.isChecked(): # clear history
+            Profile.get_instance().clear_history()
         settings['save_history'] = self.saveHistory.isChecked()
         settings.save()
 

src/profile.py

@@ -1,12 +1,14 @@
 from list_items import MessageItem, ContactItem
-from settings import Settings
 from PySide import QtCore, QtGui
-import os
 from tox import Tox
+import os
+from settings import Settings
 from toxcore_enums_and_consts import *
 from ctypes import *
 from util import curr_time, log, Singleton, curr_directory
 from tox_dns import tox_dns
+from history import *
+import time
 
 
 class ProfileHelper(object):
@@ -166,7 +168,7 @@ class Friend(Contact):
     Friend in list of friends. Can be hidden, properties 'has unread messages' and 'has alias' added
     """
 
-    def __init__(self, number, *args):
+    def __init__(self, message_getter, number, *args):
         """
         :param number: number of friend.
         """
@@ -175,11 +177,39 @@ class Friend(Contact):
         self._new_messages = False
         self._visible = True
         self._alias = False
+        self._message_getter = message_getter
+        self._corr = []
+        self._unsaved_messages = 0
 
     def __del__(self):
         self.set_visibility(False)
         del self._widget
 
+    # -----------------------------------------------------------------------------------------------------------------
+    # History support
+    # -----------------------------------------------------------------------------------------------------------------
+
+    def load_corr(self):
+        data = self._message_getter.get(42)
+        if data is not None and len(data):
+            data.reverse()
+        else:
+            return
+        self._corr = data + self._corr
+
+    def get_corr_for_saving(self):
+        return self._corr[-self._unsaved_messages:] if self._unsaved_messages else []
+
+    def get_corr(self):
+        return self._corr
+
+    def append_message(self, message):
+        self._corr.append(message)
+        self._unsaved_messages += 1
+
+    def clear_corr(self):
+        self._corr = []
+
     # -----------------------------------------------------------------------------------------------------------------
     # Alias support
     # -----------------------------------------------------------------------------------------------------------------
@@ -250,9 +280,12 @@ class Profile(Contact, Singleton):
         screen.online_contacts.setChecked(self.show_online)
         aliases = settings['friends_aliases']
         data = tox.self_get_friend_list()
+        self.history = History(tox.self_get_public_key())
         self._friends, self._active_friend = [], -1
         for i in data:  # creates list of friends
             tox_id = tox.friend_get_public_key(i)
+            if not self.history.friend_exists_in_db(tox_id):
+                self.history.add_friend_to_db(tox_id)
             try:
                 alias = filter(lambda x: x[0] == tox_id, aliases)[0][1]
             except:
@@ -260,7 +293,8 @@ class Profile(Contact, Singleton):
             item = self.create_friend_item()
             name = alias or tox.friend_get_name(i) or tox_id
             status_message = tox.friend_get_status_message(i)
-            friend = Friend(i, name, status_message, item, tox_id)
+            message_getter = self.history.messages_getter(tox_id)
+            friend = Friend(message_getter, i, name, status_message, item, tox_id)
             friend.set_alias(alias)
             self._friends.append(friend)
         self.set_name(tox.self_get_name().encode('utf-8'))
@@ -269,6 +303,13 @@ class Profile(Contact, Singleton):
         self._tox_id = tox.self_get_address()
         self.load_avatar()
 
+    def save(self):
+        print 'In save'
+        for friend in self._friends:
+            messages = friend.get_corr_for_saving()
+            self.history.save_messages_to_db(friend.tox_id, messages)
+        del self.history
+
     # -----------------------------------------------------------------------------------------------------------------
     # Edit current user's data
     # -----------------------------------------------------------------------------------------------------------------
@@ -346,8 +387,8 @@ class Profile(Contact, Singleton):
             if value is not None:
                 self._active_friend = value
                 self._friends[self._active_friend].set_messages(False)
-                self.screen.messages.clear()
                 self.screen.messageEdit.clear()
+            self.screen.messages.clear()
             friend = self._friends[self._active_friend]
             self.screen.account_name.setText(friend.name)
             self.screen.account_status.setText(friend.status_message)
@@ -358,7 +399,13 @@ class Profile(Contact, Singleton):
             pixmap.scaled(64, 64, QtCore.Qt.KeepAspectRatio)
             self.screen.account_avatar.setPixmap(avatar_path)
             self.screen.account_avatar.repaint()
-            # TODO: load history
+            friend.load_corr()  # TODO: call not every time and compute time
+            messages = friend.get_corr()[-42:]
+            for message in messages:
+                self.create_message_item(message[0],
+                                         curr_time(),
+                                         friend.name if message[1] else self._name,
+                                         message[3])
         except:  # no friend found. ignore
             log('Incorrect friend value: ' + str(value))
 
@@ -384,41 +431,64 @@ class Profile(Contact, Singleton):
         :param message_type: message type - plain text or action message (/me)
         :param message: text of message
         """
-        # TODO: save message to history
         if friend_num == self.get_active_number():  # add message to list
             user_name = Profile.get_instance().get_active_name()
-            item = MessageItem(message.decode('utf-8'), curr_time(), user_name, message_type, self._messages)
+            self.create_message_item(message.decode('utf-8'), curr_time(), user_name, message_type)
-            elem = QtGui.QListWidgetItem(self._messages)
+            self._friends[self._active_friend].append_message((message,
-            elem.setSizeHint(QtCore.QSize(500, item.getHeight()))
+                                                               MESSAGE_OWNER['FRIEND'],
-            self._messages.addItem(elem)
+                                                               time.time(),
-            self._messages.setItemWidget(elem, item)
+                                                               message_type))
-            self._messages.repaint()
         else:
             friend = filter(lambda x: x.number == friend_num, self._friends)[0]
             friend.set_messages(True)
+            friend.append_message((message.decode('utf-8'),
+                                   MESSAGE_OWNER['FRIEND'],
+                                   time.time(),
+                                   message_type))
 
     def send_message(self, text):
         """
         Send message to active friend
         :param text: message text
         """
-        # TODO: save message to history
         if self.is_active_online() and text:
             if text.startswith('/me '):
                 message_type = TOX_MESSAGE_TYPE['ACTION']
                 text = text[4:]
             else:
                 message_type = TOX_MESSAGE_TYPE['NORMAL']
-            self.tox.friend_send_message(self._active_friend, message_type, text.encode('utf-8'))
+            friend = self._friends[self._active_friend]
-            item = MessageItem(text, curr_time(), self._name, message_type, self._messages)
+            self.tox.friend_send_message(friend.number, message_type, text.encode('utf-8'))
+            self.create_message_item(text, curr_time(), self._name, message_type)
+            self.screen.messageEdit.clear()
+            friend.append_message((text,
+                                   MESSAGE_OWNER['ME'],
+                                   time.time(),
+                                   message_type))
+
+    # -----------------------------------------------------------------------------------------------------------------
+    # Factories for friend and message items
+    # -----------------------------------------------------------------------------------------------------------------
+
+    def create_friend_item(self):
+        """
+        Method-factory
+        :return: new widget for friend instance
+        """
+        item = ContactItem()
+        elem = QtGui.QListWidgetItem(self.screen.friends_list)
+        elem.setSizeHint(QtCore.QSize(250, 70))
+        self.screen.friends_list.addItem(elem)
+        self.screen.friends_list.setItemWidget(elem, item)
+        return item
+
+    def create_message_item(self, text, time, name, message_type):
+        item = MessageItem(text, time, name, message_type, self._messages)
         elem = QtGui.QListWidgetItem(self._messages)
         elem.setSizeHint(QtCore.QSize(500, item.getHeight()))
         self._messages.addItem(elem)
         self._messages.setItemWidget(elem, item)
-            self._messages.scrollToBottom()
         self._messages.repaint()
-            self.screen.messageEdit.clear()
-
     # -----------------------------------------------------------------------------------------------------------------
     # Work with friends (remove, set alias, clear history)
     # -----------------------------------------------------------------------------------------------------------------
@@ -450,27 +520,29 @@ class Profile(Contact, Singleton):
             settings.save()
             self.set_active()
 
+    def clear_history(self, num=None):
+        if num is not None:
+            friend = self._friends[num]
+            friend.clear_corr()
+            self.history.delete_messages(friend.tox_id)
+        else:  # clear all history
+            for friend in self._friends:
+                friend.clear_corr()
+                self.history.delete_messages(friend.tox_id)
+                self.history.delete_friend_from_db(friend.tox_id)
+        self.set_active()
+
     def friend_public_key(self, num):
         return self._friends[num].tox_id
 
-    def create_friend_item(self):
-        """
-        Method-factory
-        :return: new widget for friend instance
-        """
-        item = ContactItem()
-        elem = QtGui.QListWidgetItem(self.screen.friends_list)
-        elem.setSizeHint(QtCore.QSize(250, 70))
-        self.screen.friends_list.addItem(elem)
-        self.screen.friends_list.setItemWidget(elem, item)
-        return item
-
     def delete_friend(self, num):
         """
         Removes friend from contact list
         :param num: number of friend in list
         """
         friend = self._friends[num]
+        self.clear_history(num)
+        self.history.delete_friend_from_db(friend.tox_id)
         self.tox.friend_delete(friend.number)
         del self._friends[num]
         self.screen.friends_list.takeItem(num)
@@ -501,6 +573,7 @@ class Profile(Contact, Singleton):
             tox_id = tox_id[:TOX_PUBLIC_KEY_SIZE * 2]
             item = self.create_friend_item()
             friend = Friend(result, tox_id, '', item, tox_id)
+            self.history.add_friend_to_db(tox_id)  # add friend to db
             self._friends.append(friend)
             return True
         except Exception as ex:  # wrong data
@@ -520,6 +593,7 @@ class Profile(Contact, Singleton):
                 num = self.tox.friend_add_norequest(tox_id)  # num - friend number
                 item = self.create_friend_item()
                 friend = Friend(num, tox_id, '', item, tox_id)
+                self.history.add_friend_to_db(tox_id)  # add friend to db
                 self._friends.append(friend)
         except Exception as ex:  # something is wrong
             log('Accept friend request failed! ' + str(ex))