crypticorn 2.17.0rc4__py3-none-any.whl → 2.17.0rc6__py3-none-any.whl

crypticorn/common/router/status_router.py

@@ -2,14 +2,36 @@
 This module contains the status router for the API.
 It provides endpoints for checking the status of the API and get the server's time.
 SHOULD ALLOW ACCESS TO THIS ROUTER WITHOUT AUTH.
+To enable metrics, pass enable_metrics=True and the auth_handler to the router.
+>>> status_router.enable_metrics = True
+>>> status_router.auth_handler = auth_handler
+Then include the router in the FastAPI app.
 >>> app.include_router(status_router)
 """
 
 from datetime import datetime
-from typing import Literal
 from fastapi import APIRouter, Request
+from typing import Annotated, Literal
+from fastapi import APIRouter, Response, Depends
+from prometheus_client import generate_latest, CONTENT_TYPE_LATEST
+from crypticorn.common.metrics import registry
+from crypticorn.common.auth import AuthHandler, basic_auth
 
-router = APIRouter(tags=["Status"], prefix="")
+class EnhancedApiRouter(APIRouter):
+    def __init__(self, enable_metrics: bool = False, auth_handler: AuthHandler = None, *args, **kwargs):
+        """
+        Enhanced API Router that allows for metrics and authentication.
+        If enable_metrics is True, the router will include the metrics endpoint.
+        If auth_handler is provided, the router will use the auth handler to authenticate requests with
+        """
+        super().__init__(*args, **kwargs)
+        self.enable_metrics = enable_metrics
+        self.auth_handler = auth_handler
+
+        if self.enable_metrics and not self.auth_handler:
+            raise ValueError("auth_handler must be provided if enable_metrics is True")
+
+router = EnhancedApiRouter(tags=["Status"], prefix="", enable_metrics=False, auth_handler=None)
 
 
 @router.get("/", operation_id="ping")
@@ -29,3 +51,12 @@ async def time(type: Literal["iso", "unix"] = "iso") -> str:
         return datetime.now().isoformat()
     elif type == "unix":
         return str(int(datetime.now().timestamp()))
+
+
+@router.get("/metrics", operation_id="getMetrics", include_in_schema=router.enable_metrics)
+def metrics(username: Annotated[str, Depends(basic_auth)]):
+    """
+    Get Prometheus metrics for the application. Returns plain text.
+    """
+    return Response(generate_latest(registry), media_type=CONTENT_TYPE_LATEST)
+

crypticorn/common/scopes.py

@@ -16,7 +16,7 @@ class Scope(StrEnum):
 
     # Scopes that can be purchased - these actually exist in the jwt token
     READ_PREDICTIONS = "read:predictions"
-    READ_DEXSIGNALS = "read:dexsignals"
+    READ_DEX_SIGNALS = "read:dex:signals"
 
     # Hive scopes
     READ_HIVE_MODEL = "read:hive:model"
@@ -88,5 +88,5 @@ class Scope(StrEnum):
             cls.READ_METRICS_MARKETS,
             cls.READ_KLINES,
             cls.READ_SENTIMENT,
-            cls.READ_DEXSIGNALS,
+            cls.READ_DEX_SIGNALS,
         ]

crypticorn/common/utils.py

@@ -1,11 +1,10 @@
 """General utility functions and helper methods used across the codebase."""
 
 from datetime import datetime
-from typing import Any, Type, Union
+from typing import Any, Union
 from decimal import Decimal
 import string
 import random
-from pydantic import BaseModel
 import typing_extensions
 import warnings
 

crypticorn/dex/__init__.py

@@ -0,0 +1,6 @@
+from crypticorn.dex.client import *
+from crypticorn.dex.main import DexClient
+
+__all__ = [
+    "DexClient",
+]

crypticorn/dex/client/__init__.py

@@ -0,0 +1,49 @@
+# coding: utf-8
+
+# flake8: noqa
+
+"""
+DEX AI API
+
+API for DEX AI
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+__version__ = "1.0.0"
+
+# import apis into sdk package
+from crypticorn.dex.client.api.admin_api import AdminApi
+from crypticorn.dex.client.api.signals_api import SignalsApi
+from crypticorn.dex.client.api.status_api import StatusApi
+
+# import ApiClient
+from crypticorn.dex.client.api_response import ApiResponse
+from crypticorn.dex.client.api_client import ApiClient
+from crypticorn.dex.client.configuration import Configuration
+from crypticorn.dex.client.exceptions import OpenApiException
+from crypticorn.dex.client.exceptions import ApiTypeError
+from crypticorn.dex.client.exceptions import ApiValueError
+from crypticorn.dex.client.exceptions import ApiKeyError
+from crypticorn.dex.client.exceptions import ApiAttributeError
+from crypticorn.dex.client.exceptions import ApiException
+
+# import models into sdk package
+from crypticorn.dex.client.models.api_error_identifier import ApiErrorIdentifier
+from crypticorn.dex.client.models.api_error_level import ApiErrorLevel
+from crypticorn.dex.client.models.api_error_type import ApiErrorType
+from crypticorn.dex.client.models.exception_detail import ExceptionDetail
+from crypticorn.dex.client.models.log_level import LogLevel
+from crypticorn.dex.client.models.paginated_response_signal_with_token import (
+    PaginatedResponseSignalWithToken,
+)
+from crypticorn.dex.client.models.risk import Risk
+from crypticorn.dex.client.models.signal_overview_stats import SignalOverviewStats
+from crypticorn.dex.client.models.signal_volume import SignalVolume
+from crypticorn.dex.client.models.signal_with_token import SignalWithToken
+from crypticorn.dex.client.models.token_data import TokenData
+from crypticorn.dex.client.models.token_detail import TokenDetail

crypticorn/dex/client/api/__init__.py

@@ -0,0 +1,6 @@
+# flake8: noqa
+
+# import apis into api package
+from crypticorn.dex.client.api.admin_api import AdminApi
+from crypticorn.dex.client.api.signals_api import SignalsApi
+from crypticorn.dex.client.api.status_api import StatusApi