from abc import ABC, abstractmethod class CommandMapper(object): def __init__(self, base=None, mappings=None, static=None): self.base = base self.mappings = {} if mappings is None else mappings self.static = static def map(self, values): args = [self.mappings[k].map(v) for k, v in values.items() if k in self.mappings] args = [a for a in args if a != ""] options = " ".join(args) command = "{0} {1}".format(self.base, options) if self.static is not None: command += " " + self.static return command def setMapping(self, key, mapping): self.mappings[key] = mapping return self def setMappings(self, mappings): for k, v in mappings.items(): self.setMapping(k, v) return self def setBase(self, base): self.base = base return self def setStatic(self, static): self.static = static return self def keys(self): return self.mappings.keys() class CommandMapping(ABC): @abstractmethod def map(self, value): pass class Flag(CommandMapping): def __init__(self, flag): self.flag = flag def map(self, value): if value is not None and value: return self.flag else: return "" class Option(CommandMapping): def __init__(self, option): self.option = option self.spacer = " " def map(self, value): if value is not None: if isinstance(value, str) and " " in value: template = '{option}{spacer}"{value}"' else: template = "{option}{spacer}{value}" return template.format(option=self.option, spacer=self.spacer, value=value) else: return "" def setSpacer(self, spacer): self.spacer = spacer return self class Argument(CommandMapping): def map(self, value): return value