mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-09-19 23:44:28 +00:00
67 lines
1.5 KiB
Python
67 lines
1.5 KiB
Python
from dataclasses import dataclass
|
|
from typing import ClassVar, List, Optional
|
|
from enum import StrEnum
|
|
import ast
|
|
import re
|
|
|
|
class MAP_TYPES(StrEnum):
|
|
key = 'key'
|
|
"""Mapping the name of one key to another key"""
|
|
|
|
class SCOPE_TYPES(StrEnum):
|
|
namespace = 'namespace'
|
|
|
|
class PHASES(StrEnum):
|
|
postload = "postload"
|
|
"""After the YAML for a model has been loaded"""
|
|
|
|
|
|
@dataclass
|
|
class Map:
|
|
scope: str
|
|
"""The namespace that the map is relevant to"""
|
|
scope_type: SCOPE_TYPES
|
|
|
|
source: str
|
|
"""The path within the schema to select the element to transform"""
|
|
target: str
|
|
"""The path where the element should end"""
|
|
|
|
transform: Optional[callable] = None
|
|
"""
|
|
Some transformation function, currently not implemented.
|
|
"""
|
|
|
|
phase: Optional[PHASES] = None
|
|
|
|
|
|
instances: ClassVar[List['Map']] = []
|
|
"""
|
|
Maps that get defined!!!
|
|
"""
|
|
|
|
def apply(self):
|
|
raise NotImplementedError('do this in a subclass')
|
|
|
|
def __post_init__(self):
|
|
self.instances.append(self)
|
|
|
|
|
|
|
|
# def replace_keys(input: dict, source: str, target: str) -> dict:
|
|
# """Recursively change keys in a dictionary"""
|
|
|
|
|
|
class KeyMap(Map):
|
|
def apply(self, input: dict) -> dict:
|
|
"""
|
|
Change all keys from source to target in a super naive way.
|
|
|
|
Convert the dictionary to a string. Do regex. parse ast
|
|
"""
|
|
input_str = str(input)
|
|
input_str = re.sub(self.source, self.target, input_str)
|
|
out = ast.literal_eval(input_str)
|
|
return out
|
|
|
|
|