onetick-py 1.162.2__py3-none-any.whl

locator_parser/acl.py

@@ -0,0 +1,73 @@
+from locator_parser.common import Entity
+from locator_parser.actions import DoNothing
+
+
+class Allow(Entity):
+    TAG = "ALLOW"
+    SINGLE = True
+
+
+class DB(Entity):
+    TAG = "DB"
+    HAS_PROPERTIES = True
+    CHILDREN = [Allow]
+
+
+class DBs(Entity):
+    TAG = "DATABASES"
+    CHILDREN = [DB]
+
+
+class User(Entity):
+    TAG = "USER"
+    SINGLE = True
+
+
+class Role(Entity):
+    TAG = "ROLE"
+    HAS_PROPERTIES = True
+    CHILDREN = [User]
+
+
+class Roles(Entity):
+    TAG = "ROLES"
+    CHILDREN = [Role]
+
+
+class EP(Entity):
+    TAG = "EP"
+    HAS_PROPERTIES = True
+    CHILDREN = [Allow]
+
+
+class EPs(Entity):
+    TAG = "EVENT_PROCESSORS"
+    CHILDREN = [EP]
+
+
+def parse_acl(reader, writer, action=DoNothing()):
+    writer.refresh()
+    reader.set_writer(writer)
+
+    roles_p = Roles()
+    databases_p = DBs()
+    eps_p = EPs()
+
+    for _ in reader:
+        roles_p(reader, writer, action)
+        databases_p(reader, writer, action)
+        eps_p(reader, writer, action)
+
+
+# ----------------------------------------- #
+# set parent
+for _, cls in list(globals().items()):
+    try:
+        if issubclass(cls, Entity) and cls != Entity:
+            for child in cls.CHILDREN:
+                child.PARENT = Entity
+
+                if not cls.HAS_PROPERTIES and not cls.SINGLE:
+                    child.PARENT = cls
+    except Exception:
+        continue

locator_parser/actions.py

@@ -0,0 +1,266 @@
+from locator_parser.common import Action, Entity
+from collections import OrderedDict
+
+
+class Add(Action):
+    """
+    Add entity.
+
+    Example:
+    raw_db = Add(RawDB(id='PRIMARY'))
+    raw_db.add_where(DB, id='DB_2')
+    """
+
+    def __init__(self, obj):
+        self.what_to_add = obj.to_lines()
+
+        super(Add, self).__init__()
+
+        if obj.__class__.PARENT is not Entity:
+            self._add_conditions[(obj.__class__.PARENT.TAG.lower(), None, None)] = False
+
+    def do(self, obj, all_lines, properties, writer):
+        writer.add(self.what_to_add)
+
+
+class Modify(Action):
+    """
+    Modify entity properties.
+    If property doesn't exist, then it will be added.
+    If property is set to None, then it will be removed.
+
+    Example:
+    action = Modify(symbology='TDEQ', db_archive_tmp_dir=None)
+    action.add_where(DB, id="DB_1")
+    """
+
+    def __init__(self, **kwargs):
+        if len(kwargs) > 1:
+            raise Exception("It is not supported to set multiple keys yet!")
+
+        field, value = "", ""
+        delete = False
+
+        for k, v in kwargs.items():
+            if v is None:
+                delete = True
+
+            field, value = k, str(v)
+
+        self.init(field, value, delete)
+
+        super(Modify, self).__init__()
+
+    def init(self, field, value, delete):
+        self.field = field.lower()
+        self.value = '"' + value + '"'
+        self.delete = delete
+
+    def do(self, obj, all_lines, properties_lines, writer):
+        if not obj.__class__.HAS_PROPERTIES:
+            raise Exception("Entity with tag '" + obj.__class__.TAG + "' does not expect to have properties!")
+
+        new_properties = list(properties_lines)
+
+        if not hasattr(obj, self.field):
+            for line in new_properties:
+                end_pos = line.find(">")
+
+                if end_pos != -1:
+                    if line.find("/>") != -1 and line.find("/>") < end_pos:
+                        end_pos = line.find("/>")
+                    # if
+
+                    line = line[:end_pos] + " %s=%s" % (self.field, self.value) + line[end_pos:]
+                    break
+
+            new_properties[-1] = line
+        else:
+            for inx, line in enumerate(new_properties):
+                pos = line.lower().find(self.field + "=")
+
+                if pos != -1:
+                    # modify property
+                    t_line = line[pos + len(self.field):].split()[0].split("/>")[0].split(">")[0].split("=")[1]
+
+                    if self.delete:
+                        line = line.replace(line[pos: pos + len(self.field) + 1] + t_line, "")
+                    else:
+                        line = line.replace(
+                            line[pos: pos + len(self.field) + 1] + t_line,
+                            line[pos: pos + len(self.field) + 1] + self.value,
+                        )
+
+                    break
+
+            new_properties[inx] = line
+
+        writer.modify(len(all_lines), properties_lines, new_properties)
+
+
+class Delete(Action):
+    """
+    Delete entity
+
+    Example:
+    action = Delete()
+    action.add_where(DB, id="DB_1")
+    """
+
+    def do(self, obj, all_lines, properites_lines, writer):
+        writer.delete(all_lines)
+
+
+class Get(Action):
+    """
+    Populate self.result object with properites of found object
+
+    Example:
+    action = Get()
+    action.add_where(DB, id='LB')
+    """
+
+    def do(self, obj, all_lines, properties_lines, writer):
+        self.lines = all_lines
+
+        # self.result = obj.__class__()
+        properties_dict = {}
+
+        properties_lines = " ".join(properties_lines)
+
+        # variables to concat space separated
+        # values
+        res_token = ""
+        double_quotes_count = 0
+        single_quotes_count = 0
+
+        for token in filter(lambda x: x != "" and x != "<" and x != "/>" and x != ">", properties_lines.split()[1:]):
+            # ---------------------
+            # Logic to combine values in quotes are separated
+            # by space
+            if '"' in token:
+                double_quotes_count += token.count('"')
+            if "'" in token:
+                single_quotes_count += token.count("'")
+
+            if double_quotes_count > 0 or single_quotes_count > 0:
+                if (double_quotes_count + single_quotes_count) % 2 == 0:
+                    res_token += token
+                    double_quotes_count = 0
+                    single_quotes_count = 0
+                else:
+                    res_token += token
+                    continue
+            else:
+                res_token = token
+            # --------------------
+
+            t_list = res_token.split("=")
+
+            # refresh to do not concat
+            # already ready token
+            res_token = ""
+            double_quotes_count = 0
+            single_quotes_count = 0
+
+            key, value = t_list[0], t_list[1]
+
+            if value[-2:] == "/>":
+                value = value[:-2]
+
+            if value[-1:] == ">" or value[-1:] == "/":
+                value = value[:-1]
+
+            properties_dict[key.lower()] = value.replace('"', "")
+            # setattr(self.result, key.lower(), value.replace('"', ''))
+
+        self.result = obj.__class__(**properties_dict)
+
+        for key, value in properties_dict.items():
+            setattr(self.result, key, value)
+
+
+class GetAll(Action):
+
+    def __init__(self):
+        super(GetAll, self).__init__()
+        self.result = []
+
+    def do(self, obj, all_lines, properties_lines, writer):
+        get = Get()
+        get.do(obj, all_lines, properties_lines, writer)
+
+        self.result.append(get.result)
+
+    def apply(self, tag, entity):
+        if self.executed:
+            self.just_executed = False
+            self.executed = False
+
+            # Base is one that we are looking for; this entity has only
+            # tag specified.
+            # Here is logic to refresh the base, because we have to go
+            # through the subtree.
+            is_base = False
+            for key, value in reversed(self.conditions.items()):
+                t, f, v = key
+
+                if f is None and v is None:
+                    is_base = True
+
+                if is_base:
+                    self.conditions[key] = False
+                    # leave, because we support only the last open-ended base
+                    break
+
+        # first time it updates and then does nothing
+        self.conditions.update(self._add_conditions)
+        self._add_conditions = OrderedDict()
+
+        tag = tag.lower()
+
+        for key, value in self.conditions.items():
+            t, f, v = key
+
+            if tag == t:
+                if f:
+                    if hasattr(entity, f) and getattr(entity, f) == '"' + v + '"':
+                        self.conditions[(t, f, v)] = True
+                    else:
+                        # if we move to the same similar section, but with another key,
+                        # then set it to False, because in the previous section it might be
+                        # set to True
+                        self.conditions[(t, f, v)] = False
+                else:
+                    self.conditions[(t, f, v)] = True
+            else:
+                if value:
+                    continue
+
+            if not self.conditions[(t, f, v)]:
+                break
+
+        self.executed = self.is_result()
+        self.just_executed = self.executed
+
+
+class Show(Action):
+    """
+    Print object and all children objects
+
+    Example:
+    action = Show()
+    action.add_where(DB, id='S_ORDERS_LB')
+    """
+
+    def do(self, obj, all_lines, properties_lines, writer):
+        print("\n".join(all_lines))
+
+
+class DoNothing(Action):
+    """
+    Default action that does nothing
+    """
+
+    def do(self, *args, **kwargs):
+        pass

locator_parser/common.py

@@ -0,0 +1,365 @@
+import re
+from collections import OrderedDict
+from abc import ABC, abstractmethod
+from typing import List, Type
+
+
+def get_properties(declaration):
+    """
+    Construct dictionary from properties in format
+    <prop_1=value_1 prop_2=value_2 ... >
+    or
+    <prop_1=value_1 prop_2=value_2 ... />
+    and return it back
+    """
+    tokens = declaration.split()
+
+    properties = {
+        token.split("=")[0].lower(): token.split("=")[1].replace("/>", "").replace(">", "")
+        for token in tokens[1:]
+        if "=" in token
+    }
+
+    return properties
+
+
+class Generator(object):
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        self.current = next(self.generator)
+        return self.current
+
+
+class Action(ABC):
+
+    def __init__(self):
+        self.conditions = OrderedDict()
+        # additional conditions that we can deduce
+        # it's applicable for the sections without properties
+        #
+        # For example, if you add DB, then you can forget that
+        # it's situated in the single <databases> section; that
+        # information we retrieve from the entities inheritance
+        # put here and use along with conditions dict
+        self._add_conditions = OrderedDict()
+
+        self.executed = False
+        self.just_executed = False
+
+    @abstractmethod
+    def do(self):
+        pass
+
+    def clone(self, *args, **kwargs):
+        obj = self.__class__(*args, **kwargs)
+        obj.conditions = dict(self.conditions)
+
+        for key, _ in obj.conditions.items():
+            obj.conditions[key] = False
+
+        return obj
+
+    def apply(self, tag, entity):
+        if self.executed:
+            self.just_executed = False
+            return
+
+        # first time it updates and then does nothing
+        self.conditions.update(self._add_conditions)
+        self._add_conditions = OrderedDict()
+
+        tag = tag.lower()
+
+        for key, value in self.conditions.items():
+            t, f, v = key
+
+            if tag == t:
+                if f:
+                    if hasattr(entity, f) and getattr(entity, f) == '"' + v + '"':
+                        self.conditions[(t, f, v)] = True
+                    else:
+                        # if we move to the same similar section, but with another key,
+                        # then set it to False, because in the previous section it might be
+                        # set to True
+                        self.conditions[(t, f, v)] = False
+                else:
+                    self.conditions[(t, f, v)] = True
+            else:
+                if value:
+                    continue
+
+            if not self.conditions[(t, f, v)]:
+                break
+
+        self.executed = self.is_result()
+        self.just_executed = self.executed
+
+    def is_result(self):
+        for v in self.conditions.values():
+            if not v:
+                return False
+
+        return True
+
+    def add_where(self, tag, **kwargs):
+        if type(tag) is str:
+            tag = tag
+        else:
+            # for case when it's class name
+            tag = tag.TAG
+
+        if len(kwargs) > 1:
+            raise Exception("It is not support to set multiple keys yet!")
+
+        if len(kwargs) == 0:
+            self.conditions[(tag.lower(), None, None)] = False
+        else:
+            self.conditions[(tag.lower(), list(kwargs.keys())[0].lower(), list(kwargs.values())[0])] = False
+
+
+def apply_actions(func, reader, writer, actions=[], flush=False):
+    from locator_parser.io import LinesReader
+    from locator_parser.actions import DoNothing
+
+    if len(actions) == 0:
+        actions.append(DoNothing())
+
+    func(reader, writer, actions[0])
+
+    for action in actions[1:]:
+        lines = "\n".join(map(lambda x: x.replace("\n", ""), writer.lines))
+        new_reader = LinesReader(lines)
+
+        func(new_reader, writer, action)
+
+    if flush:
+        writer.flush()
+
+    for action in actions:
+        if not action.executed:
+            return False
+
+    return True
+
+
+class Writer(ABC):
+
+    NOTHING_MSG = 0
+    NEW_MSG = 1
+    MODIFY_MSG = 2
+    DELETE_MSG = 3
+
+    def __init__(self):
+        self.refresh()
+
+    def refresh(self):
+        self.__gen = self.__generator__()
+        # init generator to send further messages
+        next(self.__gen)
+
+        self.lines = []
+
+    def __generator__(self):
+        while True:
+            msg_type = yield
+
+            if msg_type == Writer.NOTHING_MSG:
+                # since second value is a just next line
+                line = yield
+                self.lines.append(line)
+            elif msg_type == Writer.NEW_MSG:
+                lines = yield
+
+                assert type(lines) is list
+
+                self.lines = self.lines[:-1] + lines + self.lines[-1:]
+            elif msg_type == Writer.MODIFY_MSG:
+                pos, orig_lines, new_lines = yield
+
+                for inx, pair in enumerate(zip(orig_lines, new_lines)):
+                    orig_line, new_line = pair
+
+                    self.lines[-pos + inx] = self.lines[-pos + inx].replace(orig_line, new_line)
+            elif msg_type == Writer.DELETE_MSG:
+                lines_to_delete = yield
+
+                for _ in range(len(lines_to_delete)):
+                    self.lines.pop()
+
+    def __next__(self):
+        return self.__send(None)
+
+    def __send(self, arg):
+        """private method to not allow user to send something directly"""
+        self.__gen.send(arg)
+
+    def put_as_is(self, line):
+        self.__send(Writer.NOTHING_MSG)
+        self.__send(line)
+
+    def add(self, lines):
+        self.__send(Writer.NEW_MSG)
+        self.__send(lines)
+
+    def modify(self, pos, orig_lines, new_lines):
+        self.__send(Writer.MODIFY_MSG)
+        self.__send((pos, orig_lines, new_lines))
+
+    def delete(self, lines):
+        self.__send(Writer.DELETE_MSG)
+        self.__send(lines)
+
+    @abstractmethod
+    def flush(self):
+        pass
+
+
+class Reader(Generator):
+
+    def __init__(self):
+        self.generator = self.__generator__()
+        self.writer = None
+        self.current = None
+
+    def set_writer(self, writer):
+        self.__writer = writer
+
+    def __generator__(self):
+        for line in self.iterable_object:
+            # put just coming line as is into the writer
+            self.__writer.put_as_is(line)
+
+            line = line.strip()
+
+            if line == "" or line[0:1] == "#":
+                continue
+
+            yield line
+
+
+class Entity(Generator, ABC):
+
+    # Tag to identify entity
+    TAG = ""
+    # Whether entity has properties
+    HAS_PROPERTIES = False
+    # Whether it is single line properties,
+    # it means that entity ends with '/>' instead of '</TAG>'
+    SINGLE = False
+    # children entities
+    CHILDREN: List[Type["Entity"]] = []
+
+    def __init__(self, **kwargs):
+        if self.__class__.SINGLE:
+            self.__class__.HAS_PROPERTIES = True
+
+        self.__tag = self.__class__.TAG.lower()
+
+        # define condition to identify section
+        pattern = "<\\s*" + self.__tag
+        if self.__class__.HAS_PROPERTIES:
+            pattern += "\\s+"
+        else:
+            pattern += "\\s*>"
+
+        self.__cond = lambda x: re.match(pattern, x.lower())
+        self.__lines = []
+        self.__properties_lines = []
+        self.__do = False
+        self.__custom_properties = kwargs
+
+    def to_lines(self):
+        result = []
+        line = "<" + self.__class__.TAG + " "
+
+        for key, value in self.__custom_properties.items():
+            line += key + '="' + str(value) + '" '
+
+        if self.__class__.SINGLE:
+            line += "/>"
+            result.append(line)
+        else:
+            line += ">"
+            result.append(line)
+
+            offset = "    "
+            for child in self.__class__.CHILDREN:
+                if not child.SINGLE and not child.HAS_PROPERTIES:
+                    result.append(offset + "<" + child.TAG + ">")
+                    result.append(offset + "</" + child.TAG + ">")
+
+            result.append("</" + self.__class__.TAG + ">")
+
+        return result
+
+    def __process(self, reader, writer, action):
+        line = reader.current
+        pattern = "</\\s*" + self.__tag + "\\s*>"
+
+        while True:
+            self.__lines.append(line)
+
+            if self.__class__.SINGLE:
+                if line.find("/>") != -1:
+                    # the final line for single and we have to return it
+                    yield line
+
+                    if action.just_executed:
+                        self.finish(writer, action)
+
+                    return
+            else:
+                if re.match(pattern, line.lower()):
+                    self.finish(writer, action)
+                    return
+
+            line = next(reader)
+
+            yield line
+
+    def finish(self, writer, action):
+        if self.__do:
+            action.do(self, self.__lines, self.__properties_lines, writer)
+
+    def __read_properties__(self):
+        line = self.current
+        self.__properties_lines.append(line)
+        declare = line
+
+        while line.find(">") == -1:
+            line = next(self)
+            self.__properties_lines.append(line)
+            declare += " " + line
+
+        properties = get_properties(declare)
+        self.__dict__.update(properties)
+
+    def __process_children__(self, reader, writer, action):
+        for _ in self:
+            for child_p in self.__class__.CHILDREN:
+                child_p()(self, writer, action)
+
+    def __call__(self, reader, writer, action):
+        if self.__cond(reader.current):
+            self.generator = self.__process(reader, writer, action)
+            self.current = reader.current
+
+            self.__read_properties__()
+            action.apply(self.__class__.TAG, self)
+
+            if action.just_executed:
+                self.__do = True
+
+            self.parse(self, writer, action)
+
+            self.__process_children__(self, writer, action)
+
+            return True
+
+        return False
+
+    def parse(self, reader, writer, action):
+        pass