feat:

--Added Node Status Check

main.py

@@ -5,6 +5,9 @@ from fastapi import FastAPI,Depends, Header, HTTPException, status
 import psycopg2
 from jwt import ExpiredSignatureError, InvalidTokenError
 import jwt
+from pydantic import BaseModel
+
+
 load_dotenv()
 app = FastAPI()
 
@@ -12,6 +15,14 @@ SECRET_KEY = os.getenv("SECRET_KEY")
 ALGORITHM = os.getenv("ALGORITHM", "HS256")
 
 
+class NodeStatusRequest(BaseModel):
+    node_name: str
+    host: str
+    port: int
+    dbname: str
+    user: str
+    password: str
+
 
 
 def verify_token(
@@ -92,28 +103,35 @@ if __name__ == "__main__":
 
 
 
-DB_CONFIG = {
+#{
-    "host": "172.16.10.178",
+#     "host": "172.16.10.178",
-    "port": 5444,
+#     "node_name = "node3",
-    "dbname": "postgres",
+#     "port": 5444,
-    "user": "postgres",
+#     "dbname": "postgres",
-    "password": "postgres",
+#     "user": "postgres",
-    "connect_timeout": 3,
+#     "password": "postgres",
-}
+#     "connect_timeout": 3,
-
+# }
-NODE_NAME = "node3"
 
 
 @app.get("/node/status")
-def node_status(token_data=Depends(verify_token)):
+def node_status(data: NodeStatusRequest,token_data=Depends(verify_token)):
 
     try:
-        conn = psycopg2.connect(**DB_CONFIG)
+        db_config = {
+            "host": data.host,
+            "port": data.port,
+            "dbname": data.dbname,
+            "user": data.user,
+            "password": data.password,
+            "connect_timeout": 3,
+        }
+        conn = psycopg2.connect(**db_config)
 
         conn.close()
 
         return {
-            "node": NODE_NAME,
+            "node": data.node_name,
             "online": True,
             "status": "online"
         }
@@ -121,7 +139,7 @@ def node_status(token_data=Depends(verify_token)):
     except Exception as e:
 
         return {
-            "node": NODE_NAME,
+            "node": data.node_name,
             "online": False,
             "status": "offline",
             "error": str(e)