twblue/src/extra/autocompletionUsers/manage.py

# -*- coding: utf-8 -*-
""" Management of users in the local database for autocompletion. """
import time
import widgetUtils
from tweepy.cursor import Cursor
from tweepy.errors import TweepyException
from wxUI import commonMessageDialogs
from . import storage, wx_manage

class autocompletionManage(object):
    def __init__(self, session):
        """ class constructor. Manages everything related to user autocompletion.

        :param session: Sessiom where the autocompletion management has been requested.
        :type session: sessions.base.Session.
        """
        super(autocompletionManage, self).__init__()
        self.session = session
        # Instantiate database so we can perform modifications on it.
        self.database = storage.storage(self.session.session_id)

    def show_settings(self):
        """ display user management  dialog and connect events associated to it. """
        self.dialog = wx_manage.autocompletionManageDialog()
        self.users = self.database.get_all_users()
        self.dialog.put_users(self.users)
        widgetUtils.connect_event(self.dialog.add, widgetUtils.BUTTON_PRESSED, self.add_user)
        widgetUtils.connect_event(self.dialog.remove, widgetUtils.BUTTON_PRESSED, self.remove_user)
        self.dialog.get_response()

    def update_list(self):
        """ update users list in management dialog. This function is normallhy used after we modify the database in any way, so we can reload all users in the autocompletion user management list. """
        item = self.dialog.users.get_selected()
        self.dialog.users.clear()
        self.users = self.database.get_all_users()
        self.dialog.put_users(self.users)
        self.dialog.users.select_item(item)

    def add_user(self, *args, **kwargs):
        """ Add a new Twitter username to the autocompletion database. """
        usr = self.dialog.get_user()
        if usr == False:
            return
        # check if user exists.
        # ToDo: in case we want to adapt this for other networks we'd need to refactor this check.
        try:
            data = self.session.twitter.get_user(screen_name=usr)
        except TweepyException as e:
            log.exception("Exception raised when attempting to add an user to the autocomplete database manually.")
            self.dialog.show_invalid_user_error()
            return
        self.database.set_user(data.screen_name, data.name, 0)
        self.update_list()

    def remove_user(self, *args, **kwargs):
        """ Remove focused user from the autocompletion database. """
        if commonMessageDialogs.delete_user_from_db() == widgetUtils.YES:
            item = self.dialog.users.get_selected()
            user = self.users[item]
            self.database.remove_user(user[0])
            self.update_list()