tomaae.homeassistant-mikrotik_router/custom_components/mikrotik_router/librouteros/query.py

# -*- coding: UTF-8 -*-
from itertools import chain
from librouteros.protocol import (
    cast_to_api,
)


class Key:

    def __init__(self, name):
        self.name = name

    def __eq__(self, other):
        yield '?={}={}'.format(self, cast_to_api(other))

    def __ne__(self, other):
        yield from self == other
        yield '?#!'

    def __lt__(self, other):
        yield '?<{}={}'.format(self, cast_to_api(other))

    def __gt__(self, other):
        yield '?>{}={}'.format(self, cast_to_api(other))

    def __str__(self):
        return str(self.name)


class Query:

    def __init__(self, path, keys, api):
        self.path = path
        self.keys = keys
        self.api = api
        self.query = tuple()

    def where(self, *args):
        self.query = tuple(chain.from_iterable(args))
        return self

    def __iter__(self):
        keys = ','.join(str(key) for key in self.keys)
        keys = '=.proplist={}'.format(keys)
        cmd = str(self.path.join('print'))
        return iter(self.api.rawCmd(cmd, keys, *self.query))


def And(left, right, *rest):
    #pylint: disable=invalid-name
    yield from left
    yield from right
    yield from chain.from_iterable(rest)
    yield '?#&'
    yield from ('?#&', ) * len(rest)


def Or(left, right, *rest):
    #pylint: disable=invalid-name
    yield from left
    yield from right
    yield from chain.from_iterable(rest)
    yield '?#|'
    yield from ('?#|', ) * len(rest)