ostium-python-sdk 0.1.8__tar.gz → 0.1.9__tar.gz

{ostium_python_sdk-0.1.8 → ostium_python_sdk-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ostium-python-sdk
-Version: 0.1.8
+Version: 0.1.9
 Summary: A python based SDK developed for interacting with Ostium, a leveraged trading application for trading currencies, commodities, indices, crypto and more.
 Home-page: https://github.com/0xOstium/ostium-python-sdk
 Author: ami@ostium.io

{ostium_python_sdk-0.1.8 → ostium_python_sdk-0.1.9}/ostium_python_sdk/sdk.py

@@ -3,10 +3,11 @@ import os
 from web3 import Web3
 from .ostium import Ostium
 from .config import NetworkConfig
+from typing import Union
 
 
 class OstiumSDK:
-    def __init__(self, network="mainnet", private_key: str = None, rpc_url: str = None):
+    def __init__(self, network: Union[str, NetworkConfig], private_key: str = None, rpc_url: str = None):
         load_dotenv()
         self.private_key = private_key or os.getenv('PRIVATE_KEY')
         if not self.private_key:
@@ -15,20 +16,28 @@ class OstiumSDK:
 
         self.rpc_url = rpc_url or os.getenv('RPC_URL')
         if not self.rpc_url:
+            network_name = "mainnet" if isinstance(
+                network, str) and network == "mainnet" else "testnet"
             raise ValueError(
-                f"No RPC URL provided for {network}. Please provide via constructor or RPC_URL environment variable")
+                f"No RPC URL provided for {network_name}. Please provide via constructor or RPC_URL environment variable")
 
         # Initialize Web3
         self.w3 = Web3(Web3.HTTPProvider(self.rpc_url))
 
         # Get network configuration
-        if network == "mainnet":
-            self.network_config = NetworkConfig.mainnet()
-        elif network == "testnet":
-            self.network_config = NetworkConfig.testnet()
+        if isinstance(network, NetworkConfig):
+            self.network_config = network
+        elif isinstance(network, str):
+            if network == "mainnet":
+                self.network_config = NetworkConfig.mainnet()
+            elif network == "testnet":
+                self.network_config = NetworkConfig.testnet()
+            else:
+                raise ValueError(
+                    f"Unsupported network: {network}. Use 'mainnet' or 'testnet'")
         else:
             raise ValueError(
-                f"Unsupported network: {network}. Use 'mainnet' or 'testnet'")
+                "Network must be either a NetworkConfig instance or a string ('mainnet' or 'testnet')")
 
         # Initialize Ostium instance
         self.ostium = Ostium(

{ostium_python_sdk-0.1.8 → ostium_python_sdk-0.1.9}/ostium_python_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ostium-python-sdk
-Version: 0.1.8
+Version: 0.1.9
 Summary: A python based SDK developed for interacting with Ostium, a leveraged trading application for trading currencies, commodities, indices, crypto and more.
 Home-page: https://github.com/0xOstium/ostium-python-sdk
 Author: ami@ostium.io

{ostium_python_sdk-0.1.8 → ostium_python_sdk-0.1.9}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="ostium-python-sdk",
-    version="0.1.8",
+    version="0.1.9",
     packages=find_packages(),
     install_requires=[
         "web3>=6.0.0",