cherry-shared2 0.1.30__tar.gz → 0.1.31__tar.gz

{cherry_shared2-0.1.30 → cherry_shared2-0.1.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cherry_shared2
-Version: 0.1.30
+Version: 0.1.31
 Summary: Cherry Bot shared utilities
 Author: headria
 Classifier: Development Status :: 4 - Beta

{cherry_shared2-0.1.30 → cherry_shared2-0.1.31}/cherry_shared/InfoService.py

@@ -1,6 +1,7 @@
 import json
 import logging
 from typing import Any, Dict, Literal, Optional
+from urllib.parse import urlparse
 import aiohttp
 from cherry_shared.launchpads import LaunchPad
 from cherry_shared.types.dexscreener import Pair
@@ -9,9 +10,22 @@ from cherry_shared.types.launchpad import LaunchpadToken
 
 class InfoService:
 
-    def __init__(self, info_url: str, helper_url: str, logger: logging.Logger = None):
+    _public_buybot_paths = {
+        "/buybot/ping",
+        "/buybot/trendingslot/init",
+        "/buybot/trendingslot/verify",
+    }
+
+    def __init__(
+        self,
+        info_url: str,
+        helper_url: str,
+        logger: logging.Logger = None,
+        api_key: Optional[str] = None,
+    ):
         self.info_url = info_url
         self.helper_url = helper_url
+        self._api_key = api_key
         if logger is None:
             self._logger = logging.getLogger(__name__)
         else:
@@ -28,6 +42,10 @@ class InfoService:
     ):
         if headers is None:
             headers = {"Content-Type": "application/json"}
+        if self._api_key:
+            path = urlparse(url).path
+            if path.startswith("/buybot") and path not in self._public_buybot_paths:
+                headers = {**headers, "x-api-key": self._api_key}
         # Define a custom timeout
         custom_timeout = aiohttp.ClientTimeout(total=timeout)
 

{cherry_shared2-0.1.30 → cherry_shared2-0.1.31}/cherry_shared2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cherry_shared2
-Version: 0.1.30
+Version: 0.1.31
 Summary: Cherry Bot shared utilities
 Author: headria
 Classifier: Development Status :: 4 - Beta

{cherry_shared2-0.1.30 → cherry_shared2-0.1.31}/setup.py

@@ -10,7 +10,7 @@ if readme_file.exists():
 
 setup(
     name="cherry_shared2",
-    version="0.1.30",
+    version="0.1.31",
     packages=find_packages(),
     install_requires=[
         "web3",