File: traversal.py

package info (click to toggle)
python-a38 0.1.8-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 440 kB
  • sloc: python: 4,065; xml: 174; makefile: 80; sh: 14
file content (36 lines) | stat: -rw-r--r-- 985 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from contextlib import contextmanager
from typing import TYPE_CHECKING, Optional

if TYPE_CHECKING:
    from . import fields


class Annotation:
    def __init__(self, prefix: Optional[str], field: "fields.Field"):
        self.prefix = prefix
        self.field = field

    @property
    def qualified_field(self) -> str:
        if self.prefix is None:
            return self.field.name
        elif self.field.name is None:
            return self.prefix
        else:
            return self.prefix + "." + self.field.name


class Traversal:
    def __init__(self, prefix: Optional[str] = None):
        self.prefix = prefix

    def with_prefix(self, prefix: str) -> "Traversal":
        raise NotImplementedError("Traversal subclasses must implement with_prefix")

    @contextmanager
    def subfield(self, name: str):
        if self.prefix is None:
            prefix = name
        else:
            prefix = self.prefix + "." + name
        yield self.with_prefix(prefix)