add

django_ltree/managers.py

@@ -11,20 +11,24 @@
 class TreeManager(models.Manager):
     def get_queryset(self) -> TreeQuerySet["TreeModel"]:
         """Returns a queryset with the models ordered by `path`"""
+
         return TreeQuerySet(model=self.model, using=self._db).order_by("path")
 
     def roots(self) -> TreeQuerySet["TreeModel"]:
         """Returns the roots of a given model"""
+
         return self.filter().roots()
 
     def children(self, path: str) -> TreeQuerySet["TreeModel"]:
         """Returns the childrens of a given object"""
+
         return self.filter().children(path)
 
     def create_child(
         self, parent: "TreeModel" = None, **kwargs
     ) -> TreeQuerySet["TreeModel"]:
         """Creates a tree child with or without parent"""
+
         paths_in_use = parent.children() if parent else self.roots()
         prefix = parent.path if parent else None
         path_generator = PathGenerator(

django_ltree/paths.py

@@ -6,7 +6,7 @@
 
 
 class PathGenerator(object):
-    def __init__(self, prefix=None, skip=None):
+    def __init__(self, prefix: str = None, skip: list[str] = None):
         combinations = string.digits + string.ascii_letters
 
         self.skip_paths = [] if skip is None else skip[:]
@@ -54,5 +54,3 @@ def guess_the_label_size(path_size: int, combination_size: int) -> int:
             label_size += 1
 
         return label_size
-
-

pyproject.toml

@@ -41,6 +41,18 @@ furo = "^2024.1.29"
 myst-parser = "^2.0.0"
 sphinx-reload = "^0.2.0"
 
+
+[tool.poetry.group.dev.dependencies]
+ruff = "^0.2.2"
+mypy = "^1.8.0"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.mypy]
+mypy_path = "./django_ltree/"
+namespace_packages = false
+show_error_codes = true
+strict = true
+warn_unreachable = true