qualys-mcp 2.1.4__tar.gz → 2.1.6__tar.gz

{qualys_mcp-2.1.4 → qualys_mcp-2.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qualys-mcp
-Version: 2.1.4
+Version: 2.1.6
 Summary: MCP server for Qualys security APIs - natural language interaction with vulnerability, asset, and cloud security data
 Project-URL: Homepage, https://github.com/nelssec/qualys-mcp
 Project-URL: Repository, https://github.com/nelssec/qualys-mcp

{qualys_mcp-2.1.4 → qualys_mcp-2.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "qualys-mcp"
-version = "2.1.4"
+version = "2.1.6"
 description = "MCP server for Qualys security APIs - natural language interaction with vulnerability, asset, and cloud security data"
 readme = "README.md"
 license = "MIT"

{qualys_mcp-2.1.4 → qualys_mcp-2.1.6}/qualys_mcp.py

@@ -42,7 +42,7 @@ def get_bearer_token():
         return None
 
 
-def api_get(url, gateway=False):
+def api_get(url, gateway=False, timeout=30):
     req = Request(url)
     if gateway:
         token = get_bearer_token()
@@ -51,7 +51,7 @@ def api_get(url, gateway=False):
         req.add_header('Authorization', f'Basic {BASIC_AUTH}')
     req.add_header('X-Requested-With', 'qualys-mcp')
     try:
-        with urlopen(req, timeout=60) as resp:
+        with urlopen(req, timeout=timeout) as resp:
             return resp.read()
     except:
         return None
@@ -381,34 +381,49 @@ def get_security_posture() -> dict:
     health = 100
     result = {'healthScore': 0, 'assets': {'total': 0, 'highRisk': 0},
               'vulns': {'critical': 0, 'high': 0}, 'containers': {'total': 0, 'atRisk': 0},
-              'cloud': {'accounts': 0, 'failedControls': 0}}
+              'cloud': {'accounts': 0, 'failedControls': 0}, 'errors': []}
 
-    assets = get_assets(500)
-    result['assets']['total'] = len(assets)
-    result['assets']['highRisk'] = len([a for a in assets if a.get('assetRiskScore', 0) >= 700])
-    if assets:
-        health -= int(result['assets']['highRisk'] / len(assets) * 50)
-
-    result['vulns']['critical'] = len(get_detections(5, 200))
-    result['vulns']['high'] = len(get_detections(4, 200))
-    if result['vulns']['critical'] > 50:
-        health -= 20
-    elif result['vulns']['critical'] > 10:
-        health -= 10
-
-    imgs = get_images(500)
-    result['containers']['total'] = len(imgs)
-    vuln_ids = {i.get('imageId') for i in get_images(100, 5)}
-    result['containers']['atRisk'] = len([c for c in get_containers(500) if c.get('imageId') in vuln_ids])
+    try:
+        assets = get_assets(100)
+        result['assets']['total'] = len(assets)
+        result['assets']['highRisk'] = len([a for a in assets if a.get('assetRiskScore', 0) >= 700])
+        if assets:
+            health -= int(result['assets']['highRisk'] / len(assets) * 50)
+    except:
+        result['errors'].append('assets')
 
-    for p in ['aws', 'azure', 'gcp']:
-        conns = get_connectors(p, 20)
-        result['cloud']['accounts'] += len(conns)
-        if conns:
-            acc = conns[0].get('awsAccountId') or conns[0].get('azureSubscriptionId') or conns[0].get('gcpProjectId')
-            if acc:
-                result['cloud']['failedControls'] += len([e for e in get_evaluations(acc, p, 500) if e.get('result') in ['FAIL', 'FAILED']])
+    try:
+        result['vulns']['critical'] = len(get_detections(5, 100))
+        result['vulns']['high'] = len(get_detections(4, 100))
+        if result['vulns']['critical'] > 50:
+            health -= 20
+        elif result['vulns']['critical'] > 10:
+            health -= 10
+    except:
+        result['errors'].append('vulns')
+
+    try:
+        imgs = get_images(100)
+        result['containers']['total'] = len(imgs)
+        vuln_ids = {i.get('imageId') for i in get_images(50, 5)}
+        result['containers']['atRisk'] = len([c for c in get_containers(100) if c.get('imageId') in vuln_ids])
+    except:
+        result['errors'].append('containers')
+
+    try:
+        for p in ['aws', 'azure', 'gcp']:
+            conns = get_connectors(p, 5)
+            if conns:
+                result['cloud']['accounts'] += len(conns)
+                acc = conns[0].get('awsAccountId') or conns[0].get('azureSubscriptionId') or conns[0].get('gcpProjectId')
+                if acc:
+                    evals = get_evaluations(acc, p, 100)
+                    result['cloud']['failedControls'] += len([e for e in evals if e.get('result') in ['FAIL', 'FAILED']])
+    except:
+        result['errors'].append('cloud')
 
+    if not result['errors']:
+        del result['errors']
     result['healthScore'] = max(0, health)
     return result