mirror of
https://github.com/samsonjs/vdirsyncer.git
synced 2026-03-25 08:55:50 +00:00
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
import logging
|
|
|
|
import click
|
|
|
|
from . import AppContext
|
|
from .. import exceptions
|
|
from ..utils import expand_path, synchronized
|
|
|
|
SUFFIX = '.fetch'
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def expand_fetch_params(config):
|
|
config = dict(config)
|
|
for key in list(config):
|
|
if not key.endswith(SUFFIX):
|
|
continue
|
|
|
|
newkey = key[:-len(SUFFIX)]
|
|
if newkey in config:
|
|
raise ValueError('Can\'t set {} and {}.'.format(key, newkey))
|
|
config[newkey] = _fetch_value(config[key], key)
|
|
del config[key]
|
|
|
|
return config
|
|
|
|
|
|
@synchronized()
|
|
def _fetch_value(opts, key):
|
|
if not isinstance(opts, list):
|
|
raise ValueError('Invalid value for {}: Expected a list, found {!r}.'
|
|
.format(key, opts))
|
|
if not opts:
|
|
raise ValueError('Expected list of length > 0.')
|
|
|
|
try:
|
|
ctx = click.get_current_context().find_object(AppContext)
|
|
if ctx is None:
|
|
raise RuntimeError()
|
|
password_cache = ctx.fetched_params
|
|
except RuntimeError:
|
|
password_cache = {}
|
|
|
|
cache_key = tuple(opts)
|
|
if cache_key in password_cache:
|
|
rv = password_cache[cache_key]
|
|
logger.debug('Found cached value for {!r}.'.format(opts))
|
|
if isinstance(rv, BaseException):
|
|
raise rv
|
|
return rv
|
|
|
|
strategy = opts[0]
|
|
try:
|
|
strategy_fn = STRATEGIES[strategy]
|
|
except KeyError:
|
|
raise exceptions.UserError('Unknown strategy: {}'.format(strategy))
|
|
|
|
logger.debug('Fetching value for {} with {} strategy.'
|
|
.format(key, strategy))
|
|
try:
|
|
rv = strategy_fn(*opts[1:])
|
|
except (click.Abort, KeyboardInterrupt) as e:
|
|
password_cache[cache_key] = e
|
|
raise
|
|
else:
|
|
if not rv:
|
|
raise exceptions.UserError('Empty value for {}, this most likely '
|
|
'indicates an error.'
|
|
.format(key))
|
|
password_cache[cache_key] = rv
|
|
return rv
|
|
|
|
|
|
def _strategy_command(*command):
|
|
import subprocess
|
|
command = (expand_path(command[0]),) + command[1:]
|
|
try:
|
|
stdout = subprocess.check_output(command, universal_newlines=True)
|
|
return stdout.strip('\n')
|
|
except OSError as e:
|
|
raise exceptions.UserError('Failed to execute command: {}\n{}'
|
|
.format(' '.join(command), str(e)))
|
|
|
|
|
|
def _strategy_prompt(text):
|
|
return click.prompt(text, hide_input=True)
|
|
|
|
|
|
STRATEGIES = {
|
|
'command': _strategy_command,
|
|
'prompt': _strategy_prompt,
|
|
}
|