def test_rssfeed_update_handler(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() subscription = yarss2.yarss_config.get_fresh_subscription_config( rssfeed_key="0", key="0") self.config.set_config({"subscriptions": {"0": subscription}}) # Check that last_update changes old_last_update = self.rssfeeds["0"]["last_update"] # Run the rssfeed with key 0 timer.rssfeed_update_handler("0") self.assertNotEquals(old_last_update, self.rssfeeds["0"]["last_update"]) old_last_update = self.rssfeeds["0"]["last_update"] # Run the subscription with key 0 like when the user runs it manually timer.rssfeed_update_handler(None, "0") # last_update should not have changed self.assertEquals(old_last_update, self.rssfeeds["0"]["last_update"]) # Must stop loopingcalls or test fails timer.disable_timers()
def test_disable_timers(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() timer.disable_timers() # Now verify that the timers have been stopped for key in timer.rssfeed_timers.keys(): # Is the timer running? self.assertFalse(timer.rssfeed_timers[key]["timer"].running)
def enable(self, config=None): self.log = yarss2.logger.Logger() self.torrent_handler = TorrentHandler(self.log) if config is None: self.yarss_config = YARSSConfig(self.log) else: self.yarss_config = config self.rssfeed_timer = RSSFeedTimer(self.yarss_config, self.log) self.rssfeed_timer.enable_timers() self.log.info("Enabled YaRSS2 %s" % yarss2.common.get_version())
def test_ttl_value_updated(self): config = get_test_config() config["rssfeeds"]["0"]["update_interval"] = 30 config["rssfeeds"]["0"]["obey_ttl"] = True config["rssfeeds"]["0"]["url"] = yarss2.common.get_resource( common.testdata_rssfeed_filename, path="tests") yarss_config = common.get_empty_test_config() yarss_config.set_config(config) timer = RSSFeedTimer(yarss_config, log) timer.enable_timers() def add_torrents_pass(*arg): pass timer.add_torrent_func = add_torrents_pass # Run the rssfeed with key 0 timer.rssfeed_update_handler("0") # Verify that update_interval of rssfeed in config was updated self.assertEquals( yarss_config.get_config()["rssfeeds"]["0"]["update_interval"], 60) # Verify that update_interval of the timer was updated self.assertEquals(timer.rssfeed_timers["0"]["update_interval"], 60) timer.disable_timers()
def test_ttl_value_updated(self): config = get_test_config() config["rssfeeds"]["0"]["update_interval"] = 30 config["rssfeeds"]["0"]["obey_ttl"] = True config["rssfeeds"]["0"]["url"] = yarss2.common.get_resource(common.testdata_rssfeed_filename, path="tests") yarss_config = common.get_empty_test_config() yarss_config.set_config(config) timer = RSSFeedTimer(yarss_config, log) timer.enable_timers() def add_torrents_pass(*arg): pass timer.add_torrent_func = add_torrents_pass # Run the rssfeed with key 0 timer.rssfeed_update_handler("0") # Verify that update_interval of rssfeed in config was updated self.assertEquals(yarss_config.get_config()["rssfeeds"]["0"]["update_interval"], 60) # Verify that update_interval of the timer was updated self.assertEquals(timer.rssfeed_timers["0"]["update_interval"], 60) timer.disable_timers()
def test_enable_timers(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() # Now verify the timers for key in timer.rssfeed_timers.keys(): # Is the timer running? self.assertTrue(timer.rssfeed_timers[key]["timer"].running) # Does the timer have the correct interval? interval = timer.rssfeed_timers[key]["timer"].interval self.assertEquals(self.rssfeeds[key]["update_interval"] * 60, interval) self.assertEquals(self.rssfeeds[key]["update_interval"], timer.rssfeed_timers[key]["update_interval"]) # Must stop loopingcalls or test fails timer.disable_timers()
def test_delete_timer(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() # Delete timer self.assertTrue(timer.delete_timer("0")) self.assertFalse(timer.delete_timer("-1")) self.assertEquals(len(timer.rssfeed_timers.keys()), 4) self.assertFalse(timer.rssfeed_timers.has_key("0")) # Must stop loopingcalls or test fails timer.disable_timers()
def test_interval_unchanged(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() # Set timer 0 with same interval self.assertFalse(timer.set_timer("0", 1)) # Must stop loopingcalls or test fails timer.disable_timers()
def test_rssfeed_update_queue(self): """Doesn't actually test anything". Used to manually test which threads are running with print statements""" from yarss2.rssfeed_handling import RSSFeedRunQueue self.config.set_config(get_test_config()) timer = RSSFeedTimer(self.config, log) def add_torrents_pass(*arg): pass timer.add_torrent_func = add_torrents_pass #import threading #print "TESST thread id:", threading.current_thread() timer.queue_rss_feed_update(rssfeed_key="0") timer.queue_rss_feed_update(subscription_key="1") timer.queue_rss_feed_update(rssfeed_key="1") return timer.queue_rss_feed_update(rssfeed_key="2")
def test_reschedule_timer(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() # Change interval to 60 minutes self.assertTrue(timer.set_timer("0", 60)) interval = timer.rssfeed_timers["0"]["timer"].interval self.assertEquals(60 * 60, interval) self.assertEquals(timer.rssfeed_timers["0"]["update_interval"], 60) # Must stop loopingcalls or test fails timer.disable_timers()
def test_schedule_timer(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() # Add new timer (with key "5") with interval 60 minutes self.assertTrue(timer.set_timer("5", 60)) # Verify timer values interval = timer.rssfeed_timers["5"]["timer"].interval self.assertEquals(60 * 60, interval) self.assertEquals(timer.rssfeed_timers["5"]["update_interval"], 60) # Should now be 6 timers self.assertEquals(len(timer.rssfeed_timers.keys()), 6) # Must stop loopingcalls or test fails timer.disable_timers()
def test_rssfeed_update_handler(self): timer = RSSFeedTimer(self.config, log) timer.enable_timers() subscription = yarss2.yarss_config.get_fresh_subscription_config(rssfeed_key="0", key="0") self.config.set_config({"subscriptions": {"0": subscription} }) # Check that last_update changes old_last_update = self.rssfeeds["0"]["last_update"] # Run the rssfeed with key 0 timer.rssfeed_update_handler("0") self.assertNotEquals(old_last_update, self.rssfeeds["0"]["last_update"]) old_last_update = self.rssfeeds["0"]["last_update"] # Run the subscription with key 0 like when the user runs it manually timer.rssfeed_update_handler(None, "0") # last_update should not have changed self.assertEquals(old_last_update, self.rssfeeds["0"]["last_update"]) # Must stop loopingcalls or test fails timer.disable_timers()
class Core(CorePluginBase): def enable(self, config=None): self.log = yarss2.logger.Logger() self.torrent_handler = TorrentHandler(self.log) if config is None: self.yarss_config = YARSSConfig(self.log) else: self.yarss_config = config self.rssfeed_timer = RSSFeedTimer(self.yarss_config, self.log) self.rssfeed_timer.enable_timers() self.log.info("Enabled YaRSS2 %s" % yarss2.common.get_version()) def disable(self): self.yarss_config.save() self.rssfeed_timer.disable_timers() def update(self): pass @export def initiate_rssfeed_update(self, rssfeed_key, subscription_key=None): self.rssfeed_timer.rssfeed_update_handler( rssfeed_key, subscription_key=subscription_key) @export def set_config(self, config): self.yarss_config.set_config(config) @export def get_config(self): "Returns the config dictionary" return self.yarss_config.get_config() @export def save_email_configurations(self, email_configurations): conf = {"email_configurations": email_configurations} try: self.yarss_config.set_config(conf) except ValueError as (v): self.log.error("Failed to save email configurations:" + str(v)) @export def save_subscription(self, dict_key=None, subscription_data=None, delete=False): """Saves the subscription in subscription_data. If subscription_data is None and delete=True, delete subscription with key==dict_key """ if delete: if subscription_data is not None: self.log.warn( "save_subscription called with delete=True, but rssfeed_data is not None!" ) else: self.log.info("Deleting Subscription '%s'" % self.yarss_config.get_config()["subscriptions"] [dict_key]["name"]) try: return self.yarss_config.generic_save_config( "subscriptions", dict_key=dict_key, data_dict=subscription_data, delete=delete) except ValueError as (v): self.log.error("Failed to save subscription:" + str(v)) @export def save_rssfeed(self, dict_key=None, rssfeed_data=None, delete=False): """Saves the rssfeed in rssfeed_data. If rssfeed_data is None and delete=True, delete rssfeed with key==dict_key """ try: if delete: if rssfeed_data is not None: self.log.warn( "save_rssfeed called with delete=True, but rssfeed_data is not None!" ) else: self.log.info("Stopping and deleting RSS Feed '%s'" % self.yarss_config.get_config()["rssfeeds"] [dict_key]["name"]) config = self.yarss_config.generic_save_config( "rssfeeds", dict_key=dict_key, data_dict=rssfeed_data, delete=delete) if delete is True: self.rssfeed_timer.delete_timer(dict_key) # Successfully saved rssfeed, check if timer was changed elif config: if self.rssfeed_timer.set_timer( rssfeed_data["key"], rssfeed_data["update_interval"]): self.log.info("Scheduled RSS Feed '%s' with interval %s" % (rssfeed_data["name"], rssfeed_data["update_interval"])) return config except ValueError as (v): self.log.error("Failed to save rssfeed:" + str(v)) @export def save_cookie(self, dict_key=None, cookie_data=None, delete=False): """Save cookie to config. If cookie_data is None and delete=True, delete cookie with key==dict_key""" try: return self.yarss_config.generic_save_config("cookies", dict_key=dict_key, data_dict=cookie_data, delete=delete) except ValueError as (v): self.log.error("Failed to save cookie:" + str(v)) @export def save_email_message(self, dict_key=None, message_data=None, delete=False): """Save email message to config. If message_data is None, delete message with key==dict_key""" try: return self.yarss_config.generic_save_config( "email_messages", dict_key=dict_key, data_dict=message_data, delete=delete) except ValueError as (v): self.log.error("Failed to save email message:" + str(v)) @export def add_torrent(self, torrent_url): cookie_header = get_cookie_header( self.yarss_config.get_config()["cookies"], torrent_url) id = self.torrent_handler.add_torrent(torrent_url, cookie_header=cookie_header)
class Core(CorePluginBase): def enable(self, config=None): self.log = yarss2.logger.Logger() self.torrent_handler = TorrentHandler(self.log) if config is None: self.yarss_config = YARSSConfig(self.log) else: self.yarss_config = config self.rssfeed_timer = RSSFeedTimer(self.yarss_config, self.log) self.rssfeed_timer.enable_timers() self.log.info("Enabled YaRSS2 %s" % yarss2.common.get_version()) def disable(self): self.yarss_config.save() self.rssfeed_timer.disable_timers() def update(self): pass @export def initiate_rssfeed_update(self, rssfeed_key, subscription_key=None): self.rssfeed_timer.rssfeed_update_handler(rssfeed_key, subscription_key=subscription_key) @export def set_config(self, config): self.yarss_config.set_config(config) @export def get_config(self): "Returns the config dictionary" return self.yarss_config.get_config() @export def save_email_configurations(self, email_configurations): conf = {"email_configurations": email_configurations} try: self.yarss_config.set_config(conf) except ValueError as (v): self.log.error("Failed to save email configurations:" + str(v)) @export def save_subscription(self, dict_key=None, subscription_data=None, delete=False): """Saves the subscription in subscription_data. If subscription_data is None and delete=True, delete subscription with key==dict_key """ if delete: if subscription_data is not None: self.log.warn("save_subscription called with delete=True, but rssfeed_data is not None!") else: self.log.info("Deleting Subscription '%s'" % self.yarss_config.get_config()["subscriptions"][dict_key]["name"]) try: return self.yarss_config.generic_save_config("subscriptions", dict_key=dict_key, data_dict=subscription_data, delete=delete) except ValueError as (v): self.log.error("Failed to save subscription:" + str(v)) @export def save_rssfeed(self, dict_key=None, rssfeed_data=None, delete=False): """Saves the rssfeed in rssfeed_data. If rssfeed_data is None and delete=True, delete rssfeed with key==dict_key """ try: if delete: if rssfeed_data is not None: self.log.warn("save_rssfeed called with delete=True, but rssfeed_data is not None!") else: self.log.info("Stopping and deleting RSS Feed '%s'" % self.yarss_config.get_config()["rssfeeds"][dict_key]["name"]) config = self.yarss_config.generic_save_config("rssfeeds", dict_key=dict_key, data_dict=rssfeed_data, delete=delete) if delete is True: self.rssfeed_timer.delete_timer(dict_key) # Successfully saved rssfeed, check if timer was changed elif config: if self.rssfeed_timer.set_timer(rssfeed_data["key"], rssfeed_data["update_interval"]): self.log.info("Scheduled RSS Feed '%s' with interval %s" % (rssfeed_data["name"], rssfeed_data["update_interval"])) return config except ValueError as (v): self.log.error("Failed to save rssfeed:" + str(v)) @export def save_cookie(self, dict_key=None, cookie_data=None, delete=False): """Save cookie to config. If cookie_data is None and delete=True, delete cookie with key==dict_key""" try: return self.yarss_config.generic_save_config("cookies", dict_key=dict_key, data_dict=cookie_data, delete=delete) except ValueError as (v): self.log.error("Failed to save cookie:" + str(v)) @export def save_email_message(self, dict_key=None, message_data=None, delete=False): """Save email message to config. If message_data is None, delete message with key==dict_key""" try: return self.yarss_config.generic_save_config("email_messages", dict_key=dict_key, data_dict=message_data, delete=delete) except ValueError as (v): self.log.error("Failed to save email message:" + str(v)) @export def add_torrent(self, torrent_url): cookie_header = get_cookie_header(self.yarss_config.get_config()["cookies"], torrent_url) id = self.torrent_handler.add_torrent(torrent_url, cookie_header=cookie_header)