mirror of
https://github.com/MCV-Software/TWBlue.git
synced 2024-11-23 03:38:08 -06:00
101 lines
3.3 KiB
Python
101 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
import logging
|
|
import widgetUtils
|
|
import output
|
|
from wxUI.dialogs.mastodon import userActions as userActionsDialog
|
|
from wxUI.dialogs.mastodon import userTimeline as userTimelineDialog
|
|
from pubsub import pub
|
|
from mastodon import MastodonError, MastodonNotFoundError
|
|
#from extra.autocompletionUsers import completion
|
|
|
|
log = logging.getLogger("controller.mastodon.userActions")
|
|
|
|
class BasicUserSelector(object):
|
|
def __init__(self, session, users=[]):
|
|
super(BasicUserSelector, self).__init__()
|
|
self.session = session
|
|
self.create_dialog(users=users)
|
|
|
|
def create_dialog(self, users):
|
|
pass
|
|
|
|
def autocomplete_users(self, *args, **kwargs):
|
|
c = completion.autocompletionUsers(self.dialog, self.session.session_id)
|
|
c.show_menu("dm")
|
|
|
|
def search_user(self, user):
|
|
try:
|
|
user = self.session.api.account_lookup(user)
|
|
return user
|
|
except MastodonError:
|
|
log.exception("Error searching for user %s.".format(user))
|
|
|
|
class userActions(BasicUserSelector):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(userActions, self).__init__(*args, **kwargs)
|
|
if self.dialog.get_response() == widgetUtils.OK:
|
|
self.process_action()
|
|
|
|
def create_dialog(self, users):
|
|
self.dialog = userActionsDialog.UserActionsDialog(users)
|
|
widgetUtils.connect_event(self.dialog.autocompletion, widgetUtils.BUTTON_PRESSED, self.autocomplete_users)
|
|
|
|
def process_action(self):
|
|
action = self.dialog.get_action()
|
|
user = self.dialog.get_user()
|
|
user = self.search_user(user)
|
|
if user == None:
|
|
return
|
|
getattr(self, action)(user)
|
|
|
|
def follow(self, user):
|
|
try:
|
|
self.session.api.account_follow(user.id)
|
|
except MastodonError as err:
|
|
output.speak("Error %s" % (str(err)), True)
|
|
|
|
def unfollow(self, user):
|
|
try:
|
|
result = self.session.api.account_unfollow(user.id)
|
|
except MastodonError as err:
|
|
output.speak("Error %s" % (str(err)), True)
|
|
|
|
def mute(self, user):
|
|
try:
|
|
id = self.session.api.account_mute(user.id)
|
|
except MastodonError as err:
|
|
output.speak("Error %s" % (str(err)), True)
|
|
|
|
def unmute(self, user):
|
|
try:
|
|
id = self.session.api.account_unmute(user.id)
|
|
except MastodonError as err:
|
|
output.speak("Error %s" % (str(err)), True)
|
|
|
|
def block(self, user):
|
|
try:
|
|
id = self.session.api.account_block(user.id)
|
|
except MastodonError as err:
|
|
output.speak("Error %s" % (str(err)), True)
|
|
|
|
def unblock(self, user):
|
|
try:
|
|
id = self.session.api.account_unblock(user.id)
|
|
except MastodonError as err:
|
|
output.speak("Error %s" % (str(err)), True)
|
|
|
|
class UserTimeline(BasicUserSelector):
|
|
|
|
def create_dialog(self, users):
|
|
self.dialog = userTimelineDialog.UserTimeline(users)
|
|
widgetUtils.connect_event(self.dialog.autocompletion, widgetUtils.BUTTON_PRESSED, self.autocomplete_users)
|
|
|
|
def process_action(self):
|
|
action = self.dialog.get_action()
|
|
user = self.dialog.get_user()
|
|
user = self.search_user(user)
|
|
if user == None:
|
|
return
|
|
self.user = user
|
|
return action |