refactor: removed instance export from plugin and fixed code to adhere to POSIX standard

Author: AnishRane

plugins/dpsn/dpsn_plugin_gamesdk/dpsn_plugin.py

@@ -1,15 +1,10 @@
 import os
-from dotenv import load_dotenv
 from dpsn_client.client import DpsnClient, DPSNError
 from datetime import datetime
 from game_sdk.game.custom_types import Function, Argument, FunctionResultStatus
 from typing import Dict, Any, Callable, Tuple, Optional
 import json
 import logging
-
-# Load .env variables
-load_dotenv()
-
 # Configure logging for the plugin (optional, but good practice)
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger = logging.getLogger("DpsnPlugin")
@@ -285,7 +280,5 @@ def _handle_client_error(self, error: DPSNError):
             self._initialized = False
          else:
              logger.warning("Client error received, but plugin was already not marked as initialized.")
-             
-# Create plugin instance
-plugin = DpsnPlugin()
+    
 

plugins/dpsn/examples/dpsn_agent.py

@@ -7,13 +7,17 @@
 import threading
 import signal
 from typing import Dict, Any, Tuple
-
-parent_dir = str(Path(__file__).parent.parent)
-sys.path.append(parent_dir)
-
+from dotenv import load_dotenv
 from game_sdk.game.agent import Agent, WorkerConfig
 from game_sdk.game.custom_types import FunctionResult, FunctionResultStatus, Function, Argument
-from dpsn_plugin_gamesdk.dpsn_plugin import plugin
+from dpsn_plugin_gamesdk.dpsn_plugin import DpsnPlugin
+load_dotenv()
+
+dpsn_plugin = DpsnPlugin(
+    dpsn_url=os.getenv("DPSN_URL"),
+    pvt_key=os.getenv("PVT_KEY")
+)
+
 
 # Global message counter and timestamp
 message_count = 0
@@ -134,7 +138,7 @@ def exit_program():
 )
 
 # Set the callback in the plugin instance *before* running the agent
-plugin.set_message_callback(handle_incoming_message)
+dpsn_plugin.set_message_callback(handle_incoming_message)
 # --- End Message Handler Setup ---
 
 def get_agent_state_fn(function_result: FunctionResult, current_state: dict) -> dict:
@@ -187,14 +191,15 @@ def get_worker_state(function_result: FunctionResult, current_state: dict) -> di
     worker_description="Worker specialized in managing DPSN topic subscriptions, unsubscriptions, message handling, and shutdown.",
     get_state_fn=get_worker_state,
     action_space=[
-        plugin.get_function("subscribe"),
-        plugin.get_function("unsubscribe"),
-        plugin.get_function("shutdown"),
+        dpsn_plugin.get_function("subscribe"),
+        dpsn_plugin.get_function("unsubscribe"),
+        dpsn_plugin.get_function("shutdown"),
         check_status_function,
         complete_task_function  # Add the task completion function
     ],
 )
 
+
 # Initialize the agent
 agent = Agent(
     api_key=os.environ.get("GAME_API_KEY"),
@@ -253,4 +258,4 @@ def get_worker_state(function_result: FunctionResult, current_state: dict) -> di
 except Exception as e:
     print(f"Error running agent:{e}")
     import traceback
-    traceback.print_exc()
+    traceback.print_exc()

plugins/dpsn/examples/dpsn_worker.py

@@ -1,24 +1,25 @@
 import sys
 import os
+from dotenv import load_dotenv
 from pathlib import Path
 from typing import Dict, Any, List
 from datetime import datetime
 import time # Import time for sleep
-
-# Add the parent directory to Python path
-parent_dir = str(Path(__file__).parent.parent)
-sys.path.append(parent_dir)
+load_dotenv()
 
 # Import FunctionResultStatus to check the status enum
 from game_sdk.game.custom_types import FunctionResultStatus 
-from dpsn_plugin_gamesdk.dpsn_plugin import plugin
+from dpsn_plugin_gamesdk.dpsn_plugin import DpsnPlugin
 
 class DpsnWorker:
     """
     DPSN Worker for processing market data and executing trades (Synchronous Version)
     """
     def __init__(self):
-        self.plugin = plugin
+        self.plugin = DpsnPlugin(
+            dpsn_url=os.getenv("DPSN_URL"),
+            pvt_key=os.getenv("PVT_KEY")
+        )
         self.trades: List[Dict[str, Any]] = []
         self.is_running = False
 
@@ -128,4 +129,5 @@ def main():
 
 if __name__ == "__main__":
     # Run main directly without asyncio
-    main() 
+    main()
+    

plugins/dpsn/examples/test_dpsn_game_functions.py

@@ -3,23 +3,24 @@
 from pathlib import Path
 import time
 from datetime import datetime
+from game_sdk.game.custom_types import Function, Argument, FunctionResultStatus
+from dotenv import load_dotenv
+load_dotenv()
 
-# Add the parent directory to Python path
-parent_dir = str(Path(__file__).parent.parent)
-sys.path.append(parent_dir)
 
-from dpsn_plugin_gamesdk.dpsn_plugin import plugin
+
+from dpsn_plugin_gamesdk.dpsn_plugin import DpsnPlugin
+
+dpsn_plugin=DpsnPlugin(
+     dpsn_url=os.getenv("DPSN_URL"),
+    pvt_key=os.getenv("PVT_KEY")
+)
 
 def test_dpsn_connection():
     """Test DPSN connection and basic functionality"""
     print("\n🔄 Testing DPSN Connection...")
 
-    # Initialize DPSN client (without options since the method doesn't accept them)
-    result = plugin.initialize()
-    if not result["success"]:
-        print(f"❌ Failed to initialize DPSN: {result.get('error')}")
-        return False
-    
+  
     # Wait for connection to stabilize
     time.sleep(1)
     print("✅ DPSN initialized successfully")
@@ -36,13 +37,13 @@ def handle_message(message_data):
         print(f"Received message on {topic}: {payload}")
 
     # Set the callback
-    plugin.set_message_callback(handle_message)
+    dpsn_plugin.set_message_callback(handle_message)
 
     # Test topic
     topic = "0xe14768a6d8798e4390ec4cb8a4c991202c2115a5cd7a6c0a7ababcaf93b4d2d4/SOLUSDT/ohlc"
 
     print(f"Subscribing to topic: {topic}")
-    result = plugin.subscribe(topic)
+    result = dpsn_plugin.subscribe(topic)
     if not result["success"]:
         print(f"❌ Failed to subscribe to topic: {result.get('error')}")
         return False
@@ -52,26 +53,24 @@ def handle_message(message_data):
 
     try:
         while True:
-            if not plugin.client.dpsn_broker.is_connected():
+            if not dpsn_plugin.client.dpsn_broker.is_connected():
                 print("Connection lost, attempting to reconnect...")
-                plugin.initialize()
+                dpsn_plugin.initialize()
                 time.sleep(1)
-                plugin.subscribe(topic)
+                dpsn_plugin.subscribe(topic)
             time.sleep(1)
 
     except KeyboardInterrupt:
         print("\n⚠️ Test interrupted by user")
         return True
 
-    return True
-
 def test_shutdown():
     """Test graceful shutdown"""
     print("\n🔄 Testing Shutdown...")
 
-    result = plugin.shutdown()
-    if not result["success"]:
-        print(f"❌ Failed to shutdown: {result.get('error')}")
+    status,message,extra = dpsn_plugin.shutdown()
+    if status is not FunctionResultStatus.DONE:
+        print(f"❌ Failed to shutdown")
         return False
 
     print("✅ Shutdown successful")
@@ -86,7 +85,6 @@ def main():
         if not test_dpsn_connection():
             return
 
-        # Test subscription and message reception
         if not test_subscribe_and_receive():
             return