mirror of
https://github.com/MCV-Software/TWBlue.git
synced 2025-04-05 03:12:29 -04:00
62 lines
2.7 KiB
Python
62 lines
2.7 KiB
Python
# -*- 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()
|