mandos.model.mappings

Complexity

Total Complexity

8

Size/Duplication

Total Lines	42
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	RegexMapParser.from_resource()	0	3	1
A	RegexMapParser.from_path()	0	9	3
A	RegexMap.__init__()	0	2	1
A	RegexMap.get()	0	6	3

import re

Missing module docstring

from pathlib import Path
from typing import Mapping

from mandos.model import MandosResources


class RegexMap:

Missing class docstring

    def __init__(self, dct: Mapping[re.Pattern, str]):
        self._dct = dct

    def get(self, s: str) -> str:

Argument name "s" doesn't conform to snake_case naming style ('([^\\W\\dA-Z][^\\WA-Z]2,|_[^\\WA-Z]*|__[^\\WA-Z\\d_][^\\WA-Z]+__)$' pattern)

Missing function or method docstring

        for pat, fixed in self._dct.items():
            sub = pat.sub(fixed, s)
            if sub != s:
                return sub
        return s


class RegexMapParser:

Missing class docstring

    @classmethod
    def from_resource(cls, name: str):

Missing function or method docstring

        return cls.from_path(MandosResources.path("mappings", name, suffix=".regexes"))

    @classmethod
    def from_path(cls, path: Path):

Missing function or method docstring

        dct = {}
        for line in path.read_text(encoding="utf8").splitlines():
            if not line.startswith("#"):
                key, value = line.split("--->")
                key, value = key.strip(), value.strip()
                dct[re.compile(key)] = value
        return RegexMap(dct)


if __name__ == "__main__":
    mp = RegexMapParser.from_resource("@targets_neuro.regex")
    print(mp.get("Dopamine D3 receptor"))


__all__ = ["RegexMap", "RegexMapParser"]