fastapi-factory-utilities 0.2.11__py3-none-any.whl → 0.2.12__py3-none-any.whl

fastapi_factory_utilities/core/plugins/odm_plugin/repositories.py

@@ -6,7 +6,6 @@ from collections.abc import AsyncGenerator, Callable
 from contextlib import asynccontextmanager
 from typing import Any, Generic, TypeVar, get_args
 from uuid import UUID
-from venv import create
 
 from motor.motor_asyncio import AsyncIOMotorClientSession, AsyncIOMotorDatabase
 from pydantic import BaseModel
@@ -79,7 +78,7 @@ class AbstractRepository(ABC, Generic[DocumentGenericType, EntityGenericType]):
             UnableToCreateEntityDueToDuplicateKeyError: If the entity cannot be created due to a duplicate key error.
             OperationError: If the operation fails.
         """
-        insert_time: datetime.datetime = datetime.datetime.now(tz=datetime.timezone.utc)
+        insert_time: datetime.datetime = datetime.datetime.now(tz=datetime.UTC)
         try:
             entity_dump: dict[str, Any] = entity.model_dump()
             entity_dump["created_at"] = insert_time
@@ -120,7 +119,7 @@ class AbstractRepository(ABC, Generic[DocumentGenericType, EntityGenericType]):
             ValueError: If the entity cannot be created from the document.
             OperationError: If the operation fails.
         """
-        update_time: datetime.datetime = datetime.datetime.now(tz=datetime.timezone.utc)
+        update_time: datetime.datetime = datetime.datetime.now(tz=datetime.UTC)
         try:
             entity_dump: dict[str, Any] = entity.model_dump()
             entity_dump["updated_at"] = update_time

fastapi_factory_utilities/core/plugins/opentelemetry_plugin/builder.py

@@ -1,9 +1,20 @@
 """Provides a factory function to build a objets for OpenTelemetry."""
 
 from typing import Any, Self
+from urllib.parse import ParseResult, urlparse
 
-from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import OTLPMetricExporter
-from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import (
+    OTLPMetricExporter as OTLPMetricExporterGRPC,
+)
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
+    OTLPSpanExporter as OTLPSpanExporterGRPC,
+)
+from opentelemetry.exporter.otlp.proto.http.metric_exporter import (
+    OTLPMetricExporter as OTLPMetricExporterHTTP,
+)
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
+    OTLPSpanExporter as OTLPSpanExporterHTTP,
+)
 from opentelemetry.metrics import set_meter_provider
 from opentelemetry.propagate import set_global_textmap
 from opentelemetry.propagators.b3 import B3MultiFormat
@@ -31,24 +42,30 @@ from fastapi_factory_utilities.core.utils.yaml_reader import (
     YamlFileReader,
 )
 
-from .configs import OpenTelemetryConfig
+from .configs import OpenTelemetryConfig, ProtocolEnum
 from .exceptions import OpenTelemetryPluginConfigError
 
+GRPC_PORT: int = 4317
+HTTP_PORT: int = 4318
+
 
 class OpenTelemetryPluginBuilder:
     """Configure the injection bindings for OpenTelemetryPlugin."""
 
-    def __init__(self, application: ApplicationAbstractProtocol) -> None:
+    def __init__(self, application: ApplicationAbstractProtocol, settings: OpenTelemetryConfig | None = None) -> None:
         """Instantiate the OpenTelemetryPluginFactory.
 
         Args:
             application (BaseApplicationProtocol): The application object.
+            settings (OpenTelemetryConfig | None): The OpenTelemetry configuration object.
         """
         self._application: ApplicationAbstractProtocol = application
         self._resource: Resource | None = None
-        self._config: OpenTelemetryConfig | None = None
+        self._config: OpenTelemetryConfig | None = settings
         self._meter_provider: MeterProvider | None = None
         self._tracer_provider: TracerProvider | None = None
+        self._trace_exporter: OTLPSpanExporterGRPC | OTLPSpanExporterHTTP | None = None
+        self._metric_exporter: OTLPMetricExporterGRPC | OTLPMetricExporterHTTP | None = None
 
     @property
     def resource(self) -> Resource | None:
@@ -161,12 +178,27 @@ class OpenTelemetryPluginBuilder:
 
             # TODO: Extract to a dedicated method for the exporter and period reader setup
 
+            url_parsed: ParseResult = urlparse(self._config.endpoint.unicode_string())
+
+            exporter: OTLPMetricExporterGRPC | OTLPMetricExporterHTTP
             # Setup the Exporter
-            exporter = OTLPMetricExporter(
-                endpoint=f"{self._config.endpoint.unicode_string()}",
-                timeout=self._config.timeout,
-                insecure=True if str(self._config.endpoint).startswith("http") else False,
-            )
+            if url_parsed.port == GRPC_PORT or self._config.protocol == ProtocolEnum.OTLP_GRPC:
+                exporter = OTLPMetricExporterGRPC(
+                    endpoint=f"{self._config.endpoint.unicode_string()}",
+                    timeout=self._config.timeout,
+                    insecure=True if str(self._config.endpoint).startswith("http") else False,
+                )
+            elif url_parsed.port == HTTP_PORT or self._config.protocol == ProtocolEnum.OTLP_HTTP:
+                exporter = OTLPMetricExporterHTTP(
+                    endpoint=f"{self._config.endpoint.unicode_string()}/v1/metrics",
+                    timeout=self._config.timeout,
+                )
+            else:
+                raise OpenTelemetryPluginConfigError(
+                    "The endpoint port is not supported. Use 4317 for gRPC or 4318 for HTTP or set the protocol."
+                )
+
+            self._metric_exporter = exporter
 
             # Setup the Metric Reader
             meter_config: OpenTelemetryMeterConfig = self._config.meter_config
@@ -214,12 +246,28 @@ class OpenTelemetryPluginBuilder:
             if self._config.tracer_config is None:
                 raise OpenTelemetryPluginConfigError("The tracer configuration is missing.")
 
+            exporter: OTLPSpanExporterGRPC | OTLPSpanExporterHTTP
+            url_parsed: ParseResult = urlparse(self._config.endpoint.unicode_string())
             # Setup the Exporter
-            exporter = OTLPSpanExporter(
-                endpoint=f"{self._config.endpoint.unicode_string()}",
-                timeout=self._config.timeout,
-                insecure=True if str(self._config.endpoint).startswith("http") else False,
-            )
+            if url_parsed.port == GRPC_PORT or self._config.protocol == ProtocolEnum.OTLP_GRPC:
+                insecure: bool = False if str(self._config.endpoint).startswith("https") else True
+                endpoint: str = f"{self._config.endpoint.unicode_string()}"
+                exporter = OTLPSpanExporterGRPC(
+                    endpoint=endpoint,
+                    # timeout=self._config.timeout,
+                    insecure=insecure,
+                )
+            elif url_parsed.port == HTTP_PORT or self._config.protocol == ProtocolEnum.OTLP_HTTP:
+                exporter = OTLPSpanExporterHTTP(
+                    endpoint=f"{self._config.endpoint.unicode_string()}/v1/traces",
+                    # timeout=self._config.timeout,
+                )
+            else:
+                raise OpenTelemetryPluginConfigError(
+                    "The endpoint port is not supported. Use 4317 for gRPC or 4318 for HTTP or set the protocol."
+                )
+
+            self._trace_exporter = exporter
 
             # Setup the Span Processor
             tracer_config: OpenTelemetryTracerConfig = self._config.tracer_config
@@ -261,7 +309,8 @@ class OpenTelemetryPluginBuilder:
             Self: The OpenTelemetryPluginFactory object.
         """
         self.build_resource()
-        self.build_config()
+        if self._config is None:
+            self.build_config()
         self.build_meter_provider()
         self.build_tracer_provider()
 

fastapi_factory_utilities/core/plugins/opentelemetry_plugin/configs.py

@@ -1,11 +1,19 @@
 """Provides the configuration model for the OpenTelemetry plugin."""
 
+from enum import StrEnum
 from typing import Annotated
 
 from pydantic import BaseModel, ConfigDict, Field, UrlConstraints
 from pydantic_core import Url
 
 
+class ProtocolEnum(StrEnum):
+    """Defines the protocol enum for OpenTelemetry."""
+
+    OTLP_GRPC = "otlp_grpc"
+    OTLP_HTTP = "otlp_http"
+
+
 class OpenTelemetryMeterConfig(BaseModel):
     """Provides the configuration model for the OpenTelemetry meter as sub-model."""
 
@@ -77,6 +85,11 @@ class OpenTelemetryConfig(BaseModel):
         description="The collector endpoint.",
     )
 
+    protocol: ProtocolEnum | None = Field(
+        default=None,
+        description="The protocol to use for the collector.",
+    )
+
     timeout: int = Field(
         default=TEN_SECONDS_IN_SECONDS,
         description="The timeout in seconds for the collector.",

{fastapi_factory_utilities-0.2.11.dist-info → fastapi_factory_utilities-0.2.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: fastapi_factory_utilities
-Version: 0.2.11
+Version: 0.2.12
 Summary: Consolidate libraries and utilities to create microservices in Python with FastAPI, Beanie, Httpx, AioPika and OpenTelemetry.
 License: MIT
 Keywords: python,fastapi,beanie,httpx,opentelemetry,microservices

{fastapi_factory_utilities-0.2.11.dist-info → fastapi_factory_utilities-0.2.12.dist-info}/RECORD

@@ -25,10 +25,10 @@ fastapi_factory_utilities/core/plugins/odm_plugin/configs.py,sha256=zQoJC1wLNyq2
 fastapi_factory_utilities/core/plugins/odm_plugin/depends.py,sha256=OcLsfTLzMBk_xFV6qsMy_-qFkiphEbbEuaHUooagxg8,730
 fastapi_factory_utilities/core/plugins/odm_plugin/documents.py,sha256=BFQYHxHBmTacJRfhZi2OffvT_RAFvAAiDVQAa_d6Y7w,1141
 fastapi_factory_utilities/core/plugins/odm_plugin/exceptions.py,sha256=acnKJB0lGAzDs-7-LjBap8shjP3iV1a7dw7ouPVF27o,551
-fastapi_factory_utilities/core/plugins/odm_plugin/repositories.py,sha256=5hZ5Cvdqqlnb4NITWlrg_o078B4U0U61MjAXG3VxcH8,8747
+fastapi_factory_utilities/core/plugins/odm_plugin/repositories.py,sha256=r880o--iVj3oK3yTenM5kgJwfFWKSUZa0bfnEwOukFE,8705
 fastapi_factory_utilities/core/plugins/opentelemetry_plugin/__init__.py,sha256=UsXPjiAASn5GIHW8vrF32mklxGNq8ajILV-ty4K1Tbs,4371
-fastapi_factory_utilities/core/plugins/opentelemetry_plugin/builder.py,sha256=ZStlbq7zYtlXXrTloCDmizqxCRu9g2vY5T1oM-AzxLg,10001
-fastapi_factory_utilities/core/plugins/opentelemetry_plugin/configs.py,sha256=4mMa5SrmnPY1R_gVFRtFhi9WNaTGEGZL5iNNhyjcZQ0,3448
+fastapi_factory_utilities/core/plugins/opentelemetry_plugin/builder.py,sha256=9npQImifYAbEg0lFG7KwZ8V78SNrPoaINgd8vKitdMw,12509
+fastapi_factory_utilities/core/plugins/opentelemetry_plugin/configs.py,sha256=pMG9leMB7rtdkdGFLIxXflV7bf9epGrrYPt2N97KZcM,3750
 fastapi_factory_utilities/core/plugins/opentelemetry_plugin/exceptions.py,sha256=CpsHayfQpP0zghN8y5PP6TBy-cXhHoNxBR--I86gAdE,327
 fastapi_factory_utilities/core/plugins/opentelemetry_plugin/helpers.py,sha256=qpTIzX67orJz7vy6SBIwRs24omMBoToJkhpurZRjPuk,1533
 fastapi_factory_utilities/core/protocols.py,sha256=TzZKr_KfmTphk2LL-TD2XzxNlLbihbGM2DxWMhc5lEQ,2428
@@ -64,8 +64,8 @@ fastapi_factory_utilities/example/models/books/repository.py,sha256=7K63uAsSEGZ2
 fastapi_factory_utilities/example/services/books/__init__.py,sha256=Z06yNRoA7Zg3TGN-Q9rrvJg6Bbx-qJw661MVwukV6vQ,148
 fastapi_factory_utilities/example/services/books/services.py,sha256=-x7d4hotUWLzWo5uImMjFmtNcSTHwWv2bfttIbYYKbA,5380
 fastapi_factory_utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fastapi_factory_utilities-0.2.11.dist-info/LICENSE,sha256=iO1nLzMMst6vEiqgSUrfrbetM7b0bvdzXhbed5tqG8o,1074
-fastapi_factory_utilities-0.2.11.dist-info/METADATA,sha256=VjUjf7vr94g7d-cmMgWR3yPgnBZHLFqGHmhWoypcwvw,3387
-fastapi_factory_utilities-0.2.11.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-fastapi_factory_utilities-0.2.11.dist-info/entry_points.txt,sha256=IK0VcBexXo4uXQmTrbfhhnnfq4GmXPRn0GBB8hzlsq4,101
-fastapi_factory_utilities-0.2.11.dist-info/RECORD,,
+fastapi_factory_utilities-0.2.12.dist-info/LICENSE,sha256=iO1nLzMMst6vEiqgSUrfrbetM7b0bvdzXhbed5tqG8o,1074
+fastapi_factory_utilities-0.2.12.dist-info/METADATA,sha256=1h_N321DsWiApHV0OZs9sRpo78nmxyD61x6-x80AJ00,3387
+fastapi_factory_utilities-0.2.12.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+fastapi_factory_utilities-0.2.12.dist-info/entry_points.txt,sha256=IK0VcBexXo4uXQmTrbfhhnnfq4GmXPRn0GBB8hzlsq4,101
+fastapi_factory_utilities-0.2.12.dist-info/RECORD,,