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/search/__init__.py

@@ -1 +1,33 @@
-"""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

@@ -24,3 +24,16 @@
     # String
     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):