openwebrx-clone/owrx/command.py
2020-10-25 16:53:18 +01:00

80 lines
1.9 KiB
Python

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 str(value)