Remove enum_tools and pydantic_extra_types usage

Author: adamnsch

python-wrapper/pyproject.toml

@@ -33,8 +33,6 @@ keywords = ["graph", "visualization", "neo4j"]
 dependencies = [
     "ipython >=7, <10",
     "pydantic >=2 , <3",
-    "pydantic-extra-types >=2, <3",
-    "enum-tools==0.12.0"
 ]
 requires-python = ">=3.9"
 
@@ -147,12 +145,14 @@ select = [
     "F",  # flake8
     "I",  # isort
 ]
+exclude = ["src/neo4j_viz/pydantic_colors.py"]
 
 [tool.mypy]
 strict = true
 exclude = [
     '(^build|^\.?venv)',
     'build',
+    'src/neo4j_viz/pydantic_colors.py',
     ]
 plugins = ['pydantic.mypy']
 untyped_calls_exclude=["nbconvert"]

python-wrapper/src/neo4j_viz/colors.py

@@ -2,13 +2,11 @@
 from enum import Enum
 from typing import Any, Union
 
-import enum_tools
-from pydantic_extra_types.color import ColorType
+from neo4j_viz.pydantic_colors import ColorType
 
 ColorsType = Union[dict[Any, ColorType], Iterable[ColorType]]
 
 
-@enum_tools.documentation.document_enum
 class ColorSpace(Enum):
     """
     Describes the type of color space used by a color palette.

python-wrapper/src/neo4j_viz/node.py

@@ -4,10 +4,10 @@
 
 from pydantic import AliasChoices, AliasGenerator, BaseModel, Field, field_serializer, field_validator
 from pydantic.alias_generators import to_camel
-from pydantic_extra_types.color import Color, ColorType
 
 from .node_size import RealNumber
 from .options import CaptionAlignment
+from .pydantic_colors import Color, ColorType
 
 NodeIdType = Union[str, int]
 

python-wrapper/src/neo4j_viz/options.py

@@ -4,11 +4,9 @@
 from enum import Enum
 from typing import Any, Optional, Union
 
-import enum_tools.documentation
 from pydantic import BaseModel, Field, ValidationError, model_validator
 
 
-@enum_tools.documentation.document_enum
 class CaptionAlignment(str, Enum):
     """
     The alignment of the caption text for nodes and relationships.
@@ -19,7 +17,6 @@ class CaptionAlignment(str, Enum):
     BOTTOM = "bottom"
 
 
-@enum_tools.documentation.document_enum
 class Layout(str, Enum):
     FORCE_DIRECTED = "forcedirected"
     HIERARCHICAL = "hierarchical"
@@ -33,7 +30,6 @@ class Layout(str, Enum):
     GRID = "grid"
 
 
-@enum_tools.documentation.document_enum
 class Direction(str, Enum):
     """
     The direction in which the layout should be oriented
@@ -45,7 +41,6 @@ class Direction(str, Enum):
     DOWN = "down"
 
 
-@enum_tools.documentation.document_enum
 class Packing(str, Enum):
     """
     The packing method to be used
@@ -96,7 +91,6 @@ def construct_layout_options(layout: Layout, options: dict[str, Any]) -> Optiona
     )
 
 
-@enum_tools.documentation.document_enum
 class Renderer(str, Enum):
     """
     The renderer used to render the visualization.