Fix status reporting.

This commit is contained in:
John Garland 2009-07-06 06:49:34 +00:00
parent b3a3b9e4f1
commit c53af6af33
1 changed files with 10 additions and 3 deletions

View File

@ -75,6 +75,8 @@ class Core(CorePluginBase):
def enable(self): def enable(self):
log.debug('Blocklist: Plugin enabled..') log.debug('Blocklist: Plugin enabled..')
self.is_downloading = False
self.is_importing = False
self.has_imported = False self.has_imported = False
self.up_to_date = False self.up_to_date = False
self.num_blocked = 0 self.num_blocked = 0
@ -82,6 +84,8 @@ class Core(CorePluginBase):
self.core = component.get("Core") self.core = component.get("Core")
update_now = False
self.config = deluge.configmanager.ConfigManager("blocklist.conf", DEFAULT_PREFS) self.config = deluge.configmanager.ConfigManager("blocklist.conf", DEFAULT_PREFS)
if self.config["load_on_start"]: if self.config["load_on_start"]:
if self.config["last_update"]: if self.config["last_update"]:
@ -90,7 +94,7 @@ class Core(CorePluginBase):
"%a, %d %b %Y %H:%M:%S GMT") "%a, %d %b %Y %H:%M:%S GMT")
check_period = datetime.timedelta(days=self.config["check_after_days"]) check_period = datetime.timedelta(days=self.config["check_after_days"])
if not self.config["last_update"] or last_update + check_period >= now: if not self.config["last_update"] or last_update + check_period >= now:
d = self.check_import() update_now = True
else: else:
self.use_cache = True self.use_cache = True
d = self.import_list() d = self.import_list()
@ -99,7 +103,7 @@ class Core(CorePluginBase):
# This function is called every 'check_after_days' days, to download # This function is called every 'check_after_days' days, to download
# and import a new list if needed. # and import a new list if needed.
self.update_timer = LoopingCall(self.check_import) self.update_timer = LoopingCall(self.check_import)
self.update_timer.start(self.config["check_after_days"] * 24 * 60 * 60) self.update_timer.start(self.config["check_after_days"] * 24 * 60 * 60, update_now)
def disable(self): def disable(self):
self.config.save() self.config.save()
@ -149,7 +153,7 @@ class Core(CorePluginBase):
else: else:
status["state"] = "Idle" status["state"] = "Idle"
status["up_to_date"] = False status["up_to_date"] = self.up_to_date
status["num_blocked"] = self.num_blocked status["num_blocked"] = self.num_blocked
status["file_progress"] = self.file_progress status["file_progress"] = self.file_progress
status["file_type"] = self.config["list_type"] status["file_type"] = self.config["list_type"]
@ -190,6 +194,7 @@ class Core(CorePluginBase):
log.debug("Attempting to download blocklist %s" % url) log.debug("Attempting to download blocklist %s" % url)
log.debug("Sending headers: %s" % headers) log.debug("Sending headers: %s" % headers)
self.up_to_date = False
self.is_downloading = True self.is_downloading = True
return download_file(url, deluge.configmanager.get_config_dir("blocklist.download"), on_retrieve_data, headers) return download_file(url, deluge.configmanager.get_config_dir("blocklist.download"), on_retrieve_data, headers)
@ -219,6 +224,7 @@ class Core(CorePluginBase):
log.debug("Blocklist is up-to-date!") log.debug("Blocklist is up-to-date!")
d = threads.deferToThread(update_info, d = threads.deferToThread(update_info,
deluge.configmanager.get_config_dir("blocklist.cache")) deluge.configmanager.get_config_dir("blocklist.cache"))
self.up_to_date = True
self.use_cache = True self.use_cache = True
f.trap(f.type) f.trap(f.type)
elif self.failed_attempts < self.config["try_times"]: elif self.failed_attempts < self.config["try_times"]:
@ -235,6 +241,7 @@ class Core(CorePluginBase):
log.debug("Latest blocklist is already imported") log.debug("Latest blocklist is already imported")
return True return True
self.is_importing = True
self.num_blocked = 0 self.num_blocked = 0
# TODO: Open blocklist with appropriate reader # TODO: Open blocklist with appropriate reader