From d43738a2ec0bd9d8fd742576fd09c2ceaff96f1a Mon Sep 17 00:00:00 2001 From: Manuel Cortez Date: Mon, 7 Nov 2022 09:54:09 -0600 Subject: [PATCH] Added base mastodon controller --- src/controller/mainController.py | 10 ++++- src/controller/mastodon/__init__.py | 1 + src/controller/mastodon/handler.py | 61 +++++++++++++++++++++++++++++ 3 files changed, 70 insertions(+), 2 deletions(-) create mode 100644 src/controller/mastodon/__init__.py create mode 100644 src/controller/mastodon/handler.py diff --git a/src/controller/mainController.py b/src/controller/mainController.py index 14413fa5..081da2bb 100644 --- a/src/controller/mainController.py +++ b/src/controller/mainController.py @@ -38,6 +38,7 @@ from mysc.repeating_timer import RepeatingTimer from mysc import restart from mysc import localization from controller.twitter import handler as TwitterHandler +from controller.mastodon import handler as MastodonHandler log = logging.getLogger("mainController") @@ -216,6 +217,8 @@ class Controller(object): if handler == None: if type == "twitter": handler = TwitterHandler.Handler() + elif type == "mastodon": + handler = MastodonHandler.Handler() self.handlers[type]=handler return handler @@ -257,8 +260,11 @@ class Controller(object): if sessions.sessions[i].is_logged == False: self.create_ignored_session_buffer(sessions.sessions[i]) continue - if sessions.sessions[i].type == "twitter": - handler = self.get_handler(type="twitter") + # Valid types currently are twitter and mastodon (Work in progress) + # More can be added later. + valid_session_types = ["twitter", "mastodon"] + if sessions.sessions[i].type in valid_session_types: + handler = self.get_handler(type=sessions.sessions[i].type) handler.create_buffers(sessions.sessions[i], controller=self) log.debug("Setting updates to buffers every %d seconds..." % (60*config.app["app-settings"]["update_period"],)) self.update_buffers_function = RepeatingTimer(60*config.app["app-settings"]["update_period"], self.update_buffers) diff --git a/src/controller/mastodon/__init__.py b/src/controller/mastodon/__init__.py new file mode 100644 index 00000000..40a96afc --- /dev/null +++ b/src/controller/mastodon/__init__.py @@ -0,0 +1 @@ +# -*- coding: utf-8 -*- diff --git a/src/controller/mastodon/handler.py b/src/controller/mastodon/handler.py new file mode 100644 index 00000000..6f250f7a --- /dev/null +++ b/src/controller/mastodon/handler.py @@ -0,0 +1,61 @@ +# -*- coding: utf-8 -*- +from pubsub import pub +from sessions.twitter import utils + +class Handler(object): + + def __init__(self): + super(Handler, self).__init__() + + def create_buffers(self, session, createAccounts=True, controller=None): + session.get_user_info() + if createAccounts == True: + pub.sendMessage("core.create_account", name=session.db["user_name"], session_id=session.session_id) + root_position =controller.view.search(session.db["user_name"], session.db["user_name"]) +# for i in session.settings['general']['buffer_order']: +# if i == 'home': +# pub.sendMessage("createBuffer", buffer_type="BaseBuffer", session_type=session.type, buffer_title=_("Home"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="home_timeline", name="home_timeline", sessionObject=session, account=session.db["user_name"], sound="tweet_received.ogg", include_ext_alt_text=True, tweet_mode="extended")) +# elif i == 'mentions': +# pub.sendMessage("createBuffer", buffer_type="BaseBuffer", session_type=session.type, buffer_title=_("Mentions"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="mentions_timeline", name="mentions", sessionObject=session, account=session.db["user_name"], sound="mention_received.ogg", include_ext_alt_text=True, tweet_mode="extended")) +# elif i == 'dm': +# pub.sendMessage("createBuffer", buffer_type="DirectMessagesBuffer", session_type=session.type, buffer_title=_("Direct messages"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_direct_messages", name="direct_messages", sessionObject=session, account=session.db["user_name"], bufferType="dmPanel", compose_func="compose_direct_message", sound="dm_received.ogg")) +# elif i == 'sent_dm': +# pub.sendMessage("createBuffer", buffer_type="SentDirectMessagesBuffer", session_type=session.type, buffer_title=_("Sent direct messages"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function=None, name="sent_direct_messages", sessionObject=session, account=session.db["user_name"], bufferType="dmPanel", compose_func="compose_direct_message")) +# elif i == 'sent_tweets': +# pub.sendMessage("createBuffer", buffer_type="BaseBuffer", session_type=session.type, buffer_title=_("Sent tweets"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="user_timeline", name="sent_tweets", sessionObject=session, account=session.db["user_name"], screen_name=session.db["user_name"], include_ext_alt_text=True, tweet_mode="extended")) +# elif i == 'favorites': +# pub.sendMessage("createBuffer", buffer_type="BaseBuffer", session_type=session.type, buffer_title=_("Likes"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_favorites", name="favourites", sessionObject=session, account=session.db["user_name"], sound="favourite.ogg", include_ext_alt_text=True, tweet_mode="extended")) +# elif i == 'followers': +# pub.sendMessage("createBuffer", buffer_type="PeopleBuffer", session_type=session.type, buffer_title=_("Followers"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_followers", name="followers", sessionObject=session, account=session.db["user_name"], sound="update_followers.ogg", screen_name=session.db["user_name"])) +# elif i == 'friends': +# pub.sendMessage("createBuffer", buffer_type="PeopleBuffer", session_type=session.type, buffer_title=_("Following"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_friends", name="friends", sessionObject=session, account=session.db["user_name"], screen_name=session.db["user_name"])) +# elif i == 'blocks': +# pub.sendMessage("createBuffer", buffer_type="PeopleBuffer", session_type=session.type, buffer_title=_("Blocked users"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_blocks", name="blocked", sessionObject=session, account=session.db["user_name"])) +# elif i == 'muted': +# pub.sendMessage("createBuffer", buffer_type="PeopleBuffer", session_type=session.type, buffer_title=_("Muted users"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_mutes", name="muted", sessionObject=session, account=session.db["user_name"])) +# pub.sendMessage("createBuffer", buffer_type="EmptyBuffer", session_type="base", buffer_title=_("Timelines"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, name="timelines", account=session.db["user_name"])) +# timelines_position =controller.view.search("timelines", session.db["user_name"]) +# for i in session.settings["other_buffers"]["timelines"]: +# pub.sendMessage("createBuffer", buffer_type="BaseBuffer", session_type=session.type, buffer_title=_(u"Timeline for {}").format(i,), parent_tab=timelines_position, start=False, kwargs=dict(parent=controller.view.nb, function="user_timeline", name="%s-timeline" % (i,), sessionObject=session, account=session.db["user_name"], sound="tweet_timeline.ogg", bufferType=None, user_id=i, include_ext_alt_text=True, tweet_mode="extended")) +# pub.sendMessage("createBuffer", buffer_type="EmptyBuffer", session_type="base", buffer_title=_("Likes timelines"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, name="favs_timelines", account=session.db["user_name"])) +# favs_timelines_position =controller.view.search("favs_timelines", session.db["user_name"]) +# for i in session.settings["other_buffers"]["favourites_timelines"]: +# pub.sendMessage("createBuffer", buffer_type="BaseBuffer", session_type=session.type, buffer_title=_("Likes for {}").format(i,), parent_tab=favs_timelines_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_favorites", name="%s-favorite" % (i,), sessionObject=session, account=session.db["user_name"], bufferType=None, sound="favourites_timeline_updated.ogg", user_id=i, include_ext_alt_text=True, tweet_mode="extended")) +# pub.sendMessage("createBuffer", buffer_type="EmptyBuffer", session_type="base", buffer_title=_("Followers timelines"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, name="followers_timelines", account=session.db["user_name"])) +# followers_timelines_position =controller.view.search("followers_timelines", session.db["user_name"]) +# for i in session.settings["other_buffers"]["followers_timelines"]: +# pub.sendMessage("createBuffer", buffer_type="PeopleBuffer", session_type=session.type, buffer_title=_("Followers for {}").format(i,), parent_tab=followers_timelines_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_followers", name="%s-followers" % (i,), sessionObject=session, account=session.db["user_name"], sound="new_event.ogg", user_id=i)) +# pub.sendMessage("createBuffer", buffer_type="EmptyBuffer", session_type="base", buffer_title=_("Following timelines"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, name="friends_timelines", account=session.db["user_name"])) +# friends_timelines_position =controller.view.search("friends_timelines", session.db["user_name"]) +# for i in session.settings["other_buffers"]["friends_timelines"]: +# pub.sendMessage("createBuffer", buffer_type="PeopleBuffer", session_type=session.type, buffer_title=_(u"Friends for {}").format(i,), parent_tab=friends_timelines_position, start=False, kwargs=dict(parent=controller.view.nb, function="get_friends", name="%s-friends" % (i,), sessionObject=session, account=session.db["user_name"], sound="new_event.ogg", user_id=i)) +# pub.sendMessage("createBuffer", buffer_type="EmptyBuffer", session_type="base", buffer_title=_("Lists"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, name="lists", account=session.db["user_name"])) +# lists_position =controller.view.search("lists", session.db["user_name"]) +# for i in session.settings["other_buffers"]["lists"]: +# pub.sendMessage("createBuffer", buffer_type="ListBuffer", session_type=session.type, buffer_title=_(u"List for {}").format(i), parent_tab=lists_position, start=False, kwargs=dict(parent=controller.view.nb, function="list_timeline", name="%s-list" % (i,), sessionObject=session, account=session.db["user_name"], bufferType=None, sound="list_tweet.ogg", list_id=utils.find_list(i, session.db["lists"]), include_ext_alt_text=True, tweet_mode="extended")) +# pub.sendMessage("createBuffer", buffer_type="EmptyBuffer", session_type="base", buffer_title=_("Searches"), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, name="searches", account=session.db["user_name"])) +# searches_position =controller.view.search("searches", session.db["user_name"]) +# for i in session.settings["other_buffers"]["tweet_searches"]: +# pub.sendMessage("createBuffer", buffer_type="SearchBuffer", session_type=session.type, buffer_title=_(u"Search for {}").format(i), parent_tab=searches_position, start=False, kwargs=dict(parent=controller.view.nb, function="search_tweets", name="%s-searchterm" % (i,), sessionObject=session, account=session.db["user_name"], bufferType="searchPanel", sound="search_updated.ogg", q=i, include_ext_alt_text=True, tweet_mode="extended")) +# for i in session.settings["other_buffers"]["trending_topic_buffers"]: +# pub.sendMessage("createBuffer", buffer_type="TrendsBuffer", session_type=session.type, buffer_title=_("Trending topics for %s") % (i), parent_tab=root_position, start=False, kwargs=dict(parent=controller.view.nb, name="%s_tt" % (i,), sessionObject=session, account=session.db["user_name"], trendsFor=i, sound="trends_updated.ogg"))