# -*- coding: utf-8 -*- import time import platform if platform.system() == "Windows": import wx from wxUI import buffers, commonMessageDialogs, menus from controller import user elif platform.system() == "Linux": from gi.repository import Gtk from gtkUI import buffers, commonMessageDialogs from controller import messages import widgetUtils import output import logging from mysc.thread_utils import call_threaded from tweepy.errors import TweepyException from pubsub import pub from controller.buffers import base log = logging.getLogger("controller.buffers.twitter.trends") class TrendsBuffer(base.Buffer): def __init__(self, parent, name, sessionObject, account, trendsFor, *args, **kwargs): super(TrendsBuffer, self).__init__(parent=parent, sessionObject=sessionObject) self.trendsFor = trendsFor self.session = sessionObject self.account = account self.invisible = True self.buffer = buffers.trendsPanel(parent, name) self.buffer.account = account self.type = self.buffer.type self.bind_events() self.sound = "trends_updated.ogg" self.trends = [] self.name = name self.buffer.name = name self.compose_function = self.compose_function_ self.get_formatted_message = self.get_message self.reply = self.search_topic def start_stream(self, mandatory=False, play_sound=True, avoid_autoreading=False): # starts stream every 3 minutes. current_time = time.time() if self.execution_time == 0 or current_time-self.execution_time >= 180 or mandatory == True: self.execution_time = current_time try: data = self.session.twitter.get_place_trends(id=self.trendsFor) except TweepyException as err: log.exception("Error %s" % (str(err))) if not hasattr(self, "name_"): self.name_ = data[0]["locations"][0]["name"] self.trends = data[0]["trends"] self.put_items_on_the_list() if self.sound != None and self.session.settings["sound"]["session_mute"] == False and self.name not in self.session.settings["other_buffers"]["muted_buffers"] and play_sound == True: self.session.sound.play(self.sound) def put_items_on_the_list(self): selected_item = self.buffer.list.get_selected() self.buffer.list.clear() for i in self.trends: tweet = self.compose_function(i) self.buffer.list.insert_item(False, *tweet) self.buffer.set_position(self.session.settings["general"]["reverse_timelines"]) def compose_function_(self, trend): return [trend["name"]] def bind_events(self): log.debug("Binding events...") self.buffer.list.list.Bind(wx.EVT_CHAR_HOOK, self.get_event) widgetUtils.connect_event(self.buffer, widgetUtils.BUTTON_PRESSED, self.tweet_about_this_trend, self.buffer.tweetTrendBtn) widgetUtils.connect_event(self.buffer, widgetUtils.BUTTON_PRESSED, self.post_status, self.buffer.tweet) widgetUtils.connect_event(self.buffer.list.list, wx.EVT_LIST_ITEM_RIGHT_CLICK, self.show_menu) widgetUtils.connect_event(self.buffer.list.list, wx.EVT_LIST_KEY_DOWN, self.show_menu_by_key) widgetUtils.connect_event(self.buffer, widgetUtils.BUTTON_PRESSED, self.search_topic, self.buffer.search_topic) def get_message(self): return self.compose_function(self.trends[self.buffer.list.get_selected()])[0] def remove_buffer(self, force=False): if force == False: dlg = commonMessageDialogs.remove_buffer() else: dlg = widgetUtils.YES if dlg == widgetUtils.YES: if self.name[:-3] in self.session.settings["other_buffers"]["trending_topic_buffers"]: self.session.settings["other_buffers"]["trending_topic_buffers"].remove(self.name[:-3]) self.session.settings.write() if self.name in self.session.db: self.session.db.pop(self.name) return True elif dlg == widgetUtils.NO: return False def url(self, *args, **kwargs): self.tweet_about_this_trend() def search_topic(self, *args, **kwargs): topic = self.trends[self.buffer.list.get_selected()]["name"] pub.sendMessage("search", term=topic) def show_menu(self, ev, pos=0, *args, **kwargs): menu = menus.trendsPanelMenu() widgetUtils.connect_event(menu, widgetUtils.MENU, self.search_topic, menuitem=menu.search_topic) widgetUtils.connect_event(menu, widgetUtils.MENU, self.tweet_about_this_trend, menuitem=menu.tweetThisTrend) widgetUtils.connect_event(menu, widgetUtils.MENU, self.view, menuitem=menu.view) widgetUtils.connect_event(menu, widgetUtils.MENU, self.copy, menuitem=menu.copy) if pos != 0: self.buffer.PopupMenu(menu, pos) else: self.buffer.PopupMenu(menu, ev.GetPosition()) def view(self, *args, **kwargs): pub.sendMessage("execute-action", action="view_item") def copy(self, *args, **kwargs): pub.sendMessage("execute-action", action="copy_to_clipboard") def tweet_about_this_trend(self, *args, **kwargs): if self.buffer.list.get_count() == 0: return title = _(u"Tweet") caption = _(u"Write the tweet here") tweet = messages.tweet(self.session, title, caption, self.get_message()+ " ") tweet.message.set_cursor_at_end() if tweet.message.get_response() == widgetUtils.OK: text = tweet.message.get_text() if len(text) > 280 and tweet.message.get("long_tweet") == True: if tweet.image == None: text = twishort.create_tweet(self.session.settings["twitter"]["user_key"], self.session.settings["twitter"]["user_secret"], text) else: text = twishort.create_tweet(self.session.settings["twitter"]["user_key"], self.session.settings["twitter"]["user_secret"], text, 1) if tweet.image == None: call_threaded(self.session.api_call, call_name="update_status", status=text) else: call_threaded(self.session.api_call, call_name="update_status_with_media", status=text, media=tweet.image) if hasattr(tweet.message, "destroy"): tweet.message.destroy() def show_menu_by_key(self, ev): if self.buffer.list.get_count() == 0: return if ev.GetKeyCode() == wx.WXK_WINDOWS_MENU: self.show_menu(widgetUtils.MENU, pos=self.buffer.list.list.GetPosition()) def open_in_browser(self, *args, **kwargs): output.speak(_(u"This action is not supported in the buffer, yet."))