115 refactor search and enhance api

src/monggregate/__init__.py

@@ -4,9 +4,10 @@
 from monggregate.dollar import S, SS
 from monggregate.pipeline import Pipeline
 
+__all__ = ["Expression", "S", "SS", "Pipeline"]
 
 __version__ = "0.21.0"
 __author__ = "Vianney Mixtur"
 __contact__ = "[email protected]"
-__copyright__ = "Copyright © 2022 Vianney Mixtur"
+__copyright__ = "Copyright © 2022-2024 Vianney Mixtur"
 __license__ = "MIT"

src/monggregate/pipeline.py

@@ -10,7 +10,6 @@
 from monggregate.base import BaseModel
 from monggregate.stages import (
     AnyStage,
-    Stage,
     BucketAuto,
     GranularityEnum,
     Bucket,
@@ -35,17 +34,16 @@
     Unset,
     VectorSearch,
 )
-from monggregate.stages.search.base import SearchBase, OperatorLiteral
+from monggregate.stages.search.base import OperatorLiteral
 from monggregate.search.operators import OperatorMap
 from monggregate.search.operators.compound import Compound, ClauseType
 from monggregate.search.collectors.facet import Facet, FacetType
 from monggregate.search.commons import CountOptions, HighlightOptions
 from monggregate.operators import MergeObjects
 from monggregate.dollar import ROOT
-from monggregate.utils import StrEnum
 
 
 
 class Pipeline(BaseModel): # pylint: disable=too-many-public-methods
     """
     MongoDB aggregation pipeline abstraction

src/monggregate/search/__init__.py

@@ -1 +1,61 @@
-"""Search Package"""
+"""Search Package"""
+
+from monggregate.search.commons import (
+    FuzzyOptions,
+    CountOptions,
+    HighlightOptions,
+    HightlightOutput,
+    CountResults,
+)
+from monggregate.search.collectors import (
+    Facet,
+    Facets,
+    FacetBucket,
+    FacetBuckets,
+    FacetResult,
+    StringFacet,
+    NumericFacet,
+    DateFacet,
+    FacetName,
+)
+from monggregate.search.operators import (
+    Autocomplete,
+    Compound,
+    Equals,
+    Exists,
+    MoreLikeThis,
+    Range,
+    Regex,
+    Text,
+    Wildcard,
+    AnyOperator,
+    OperatorMap,
+)
+
+__all__ = [
+    "FuzzyOptions",
+    "CountOptions",
+    "HighlightOptions",
+    "HightlightOutput",
+    "CountResults",
+    "Facet",
+    "Facets",
+    "FacetBucket",
+    "FacetBuckets",
+    "FacetResult",
+    "StringFacet",
+    "NumericFacet",
+    "DateFacet",
+    "FacetName",
+    "Autocomplete",
+    "Compound",
+    "Equals",
+    "Exists",
+    "MoreLikeThis",
+    "Range",
+    "Regex",
+    "Text",
+    "Wildcard",
+    "AnyOperator",
+    "OperatorMap",
+]

src/monggregate/search/collectors/__init__.py

@@ -22,5 +22,18 @@
     DateFacet,
     Facets,
     # String
-    FacetName, 
+    FacetName,
 )
+
+
+__all__ = [
+    "Facet",
+    "FacetBucket",
+    "FacetBuckets",
+    "FacetResult",
+    "StringFacet",
+    "NumericFacet",
+    "DateFacet",
+    "Facets",
+    "FacetName",
+]

src/monggregate/search/commons/__init__.py

@@ -3,3 +3,5 @@
 from monggregate.search.commons.count import CountOptions, CountResults
 from monggregate.search.commons.fuzzy import FuzzyOptions
 from monggregate.search.commons.highlight import HighlightOptions, HightlightOutput
+
+__all__ = ["CountOptions", "CountResults", "FuzzyOptions", "HighlightOptions", "HightlightOutput"]

src/monggregate/stages/search/base.py

@@ -12,7 +12,7 @@
 
 from monggregate.base import pyd, BaseModel
 from monggregate.stages.stage import Stage
-from monggregate.search.collectors import Facet, Facets
+from monggregate.search.collectors import Facet
 from monggregate.search.operators import(
     Autocomplete,
     Compound,

src/monggregate/stages/search/search.py

@@ -68,31 +68,13 @@
 
 """
 
-from datetime import datetime
-from typing import Any, Callable, Literal
 try:
     from typing import Self
 except ImportError:
     from typing_extensions import Self
 
-from monggregate.base import pyd
-from monggregate.stages.stage import Stage
-from monggregate.stages.search.base import SearchConfig, SearchBase
-from monggregate.search.collectors import Facet, Facets
-from monggregate.search.operators import(
-    Autocomplete,
-    Compound,
-    Equals,
-    Exists,
-    MoreLikeThis,
-    Range,
-    Regex,
-    Text,
-    Wildcard,
-    AnyOperator
-)
-from monggregate.search.operators.compound import ClauseType
-from monggregate.search.commons import FuzzyOptions
+from monggregate.stages.search.base import SearchBase
+
 
 
 # Classes

src/monggregate/stages/search/search_meta.py

@@ -63,7 +63,7 @@
 
 """
 
-from monggregate.stages.search.base import SearchConfig, SearchBase
+from monggregate.stages.search.base import SearchBase
 
 
 class SearchMeta(SearchBase):