diff --git a/lib/models.py b/lib/models.py index 8bfc065..d1a43ad 100644 --- a/lib/models.py +++ b/lib/models.py @@ -142,7 +142,12 @@ def import_gobject_from_vivod(self, vivod, rec): try: newdikt = subdikt.copy() newdikt['object_hash'] = object_hash - if subclass(**newdikt).is_valid() is False: + + sub, params = subclass(**newdikt), [] + if isinstance(sub, Watchdog): + params = [dashd] + + if sub.is_valid(*params) is False: govobj.vote_delete(vivod) return (govobj, None) diff --git a/lib/vivod.py b/lib/vivod.py index 2ad4487..9590868 100644 --- a/lib/vivod.py +++ b/lib/vivod.py @@ -15,7 +15,9 @@ class VivoDaemon(): def __init__(self, **kwargs): - host = kwargs.get('host', '127.0.0.1') + host = config.sentinel_cfg.get('vivo_host', None) + if host is None: + host = kwargs.get('host', '127.0.0.1') user = kwargs.get('user') password = kwargs.get('password') port = kwargs.get('port') diff --git a/lib/vivolib.py b/lib/vivolib.py index 713e37d..4028e14 100644 --- a/lib/vivolib.py +++ b/lib/vivolib.py @@ -10,6 +10,7 @@ import binascii from misc import printdbg, epoch2str import time +import config def is_valid_vivo_address(address, network='mainnet'): @@ -250,8 +251,9 @@ def did_we_vote(output): voted = False err_msg = '' + _, config_filename = os.path.split(config.vivo_conf) try: - detail = output.get('detail').get('vivo.conf') + detail = output.get('detail').get(config_filename) result = detail.get('result') if 'errorMessage' in detail: err_msg = detail.get('errorMessage')