From f4c50f8ab5bcd4ead21759ea49840f549ac442d2 Mon Sep 17 00:00:00 2001
From: Dmitry Selyutin <ghostmansd@gmail.com>
Date: Wed, 21 Jun 2023 19:48:47 +0300
Subject: [PATCH] walker: introduce Path classes

---
 src/mdis/walker.py | 48 +++++++++++++++++++++++++++++++---------------
 1 file changed, 33 insertions(+), 15 deletions(-)

diff --git a/src/mdis/walker.py b/src/mdis/walker.py
index 4d1c33d..5509355 100644
--- a/src/mdis/walker.py
+++ b/src/mdis/walker.py
@@ -13,30 +13,48 @@ class WalkerMeta(dispatcher.DispatcherMeta):
     pass
 
 
-class PartId(enum.Enum):
-    Index = enum.auto()
-    Attribute = enum.auto()
-    Hash = enum.auto()
+class GenericPath:
+    def __init__(self, path):
+        self.__path = path
+        return super().__init__()
 
-    def __call__(self, part):
-        return {
-            PartId.Index: "[{part}]",
-            PartId.Attribute: ".{part}",
-            PartId.Hash: "{{{part}}}",
-        }[self].format(part=part)
+    def __str__(self):
+        return self.__path.__str__()
+
+    def __repr__(self):
+        return f"{self.__class__.__name__}({str(self)})"
+
+    @property
+    def path(self):
+        return self.__path
+
+
+class IndexPath(GenericPath):
+    def __str__(self):
+        return f"[{self.path}]"
+
+
+class AttributePath(GenericPath):
+    def __str__(self):
+        return f".{self.path}]"
+
+
+class HashPath(GenericPath):
+    def __str__(self):
+        return f"{{{self.path}}}"
 
 
 class Walker(dispatcher.Dispatcher, metaclass=WalkerMeta):
     @dispatcher.Hook(tuple, list)
     def dispatch_ordered_sequence(self, instance):
         for (index, item) in enumerate(instance):
-            yield (item, instance, index, PartId.Index)
+            yield (item, instance, index, IndexPath)
             yield from self(item)
 
     @dispatcher.Hook(set, frozenset)
     def dispatch_unordered_sequence(self, instance):
         for item in instance:
-            yield (item, instance, item, PartId.Hash)
+            yield (item, instance, item, HashPath)
             yield from self(item)
 
     @dispatcher.Hook(dataclasses.is_dataclass)
@@ -44,15 +62,15 @@ class Walker(dispatcher.Dispatcher, metaclass=WalkerMeta):
         for field in dataclasses.fields(instance):
             key = field.name
             value = getattr(instance, key)
-            yield (value, instance, key, PartId.Attribute)
+            yield (value, instance, key, AttributePath)
             yield from self(value)
 
     @dispatcher.Hook(dict)
     def dispatch_mapping(self, instance):
         for (key, value) in instance.items():
-            yield (key, instance, key, PartId.Hash)
+            yield (key, instance, key, HashPath)
             yield from self(key)
-            yield (value, instance, key, PartId.Index)
+            yield (value, instance, key, IndexPath)
             yield from self(value)
 
     @dispatcher.Hook(object)
-- 
2.30.2