[#2333] [Console] Fix 'set and then get' in config command

* The get method was returning old config information so use correct
 core get callback.
 * Remove redundant deferred in set method
This commit is contained in:
Calum Lind 2015-08-28 16:53:09 +01:00
parent 2b08ed06af
commit 78fcf1781a
1 changed files with 23 additions and 29 deletions

View File

@ -34,8 +34,6 @@
#
#
from twisted.internet import defer
from deluge.ui.console.main import BaseCommand
import deluge.ui.console.colors as colors
from deluge.ui.client import client
@ -106,35 +104,33 @@ class Command(BaseCommand):
return self._get_config(*args, **options)
def _get_config(self, *args, **options):
deferred = defer.Deferred()
config = component.get("CoreConfig")
keys = config.keys()
keys.sort()
s = ""
for key in keys:
if args and key not in args:
continue
color = "{!white,black,bold!}"
value = config[key]
if type(value) in colors.type_color:
color = colors.type_color[type(value)]
def _on_get_config(config):
keys = config.keys()
keys.sort()
s = ""
for key in keys:
if args and key not in args:
continue
color = "{!white,black,bold!}"
value = config[key]
if type(value) in colors.type_color:
color = colors.type_color[type(value)]
# We need to format dicts for printing
if isinstance(value, dict):
import pprint
value = pprint.pformat(value, 2, 80)
new_value = []
for line in value.splitlines():
new_value.append("%s%s" % (color, line))
value = "\n".join(new_value)
# We need to format dicts for printing
if isinstance(value, dict):
import pprint
value = pprint.pformat(value, 2, 80)
new_value = []
for line in value.splitlines():
new_value.append("%s%s" % (color, line))
value = "\n".join(new_value)
s += " %s: %s%s\n" % (key, color, value)
s += " %s: %s%s\n" % (key, color, value)
self.console.write(s)
self.console.write(s)
return config
return client.core.get_config().addCallback(_on_get_config)
def _set_config(self, *args, **options):
deferred = defer.Deferred()
config = component.get("CoreConfig")
key = options["set"][0]
val = simple_eval(options["set"][1] + " " .join(args))
@ -152,11 +148,9 @@ class Command(BaseCommand):
def on_set_config(result):
self.console.write("{!success!}Configuration value successfully updated.")
deferred.callback(True)
self.console.write("Setting %s to %s.." % (key, val))
client.core.set_config({key: val}).addCallback(on_set_config)
return deferred
return client.core.set_config({key: val}).addCallback(on_set_config)
def complete(self, text):
return [ k for k in component.get("CoreConfig").keys() if k.startswith(text) ]