goblin.provider

Complexity

Total Complexity

7

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	Provider.get_default_op_args()	0	3	1
A	TinkerGraph.get_hashable_id()	0	3	1
A	JanusGraph.get_hashable_id()	0	14	5

from typing import Dict, Any

class Provider:
    """Superclass for provider plugins"""
    DEFAULT_OP_ARGS: Dict[Any, Any] = {}

    @classmethod
    def get_default_op_args(cls, processor):
        return cls.DEFAULT_OP_ARGS.get(processor, dict())


class TinkerGraph(Provider):  # TODO
    """Default provider"""

    @staticmethod
    def get_hashable_id(val):
        return val

class JanusGraph(Provider):  # TODO
    """Default provider"""

    @staticmethod
    def get_hashable_id(val):
        if not isinstance(val, dict):
            return val
        type_prop = val.get("@type", None)
        if not type_prop == "janusgraph:RelationIdentifier":
            return val
        val_prop = val.get("@value", None)
        if not isinstance(val_prop, dict):
            return val
        rel_prop = val_prop.get("relationId", None)
        if not rel_prop:
            return val
        return rel_prop