utils_devops 0.1.161__tar.gz → 0.1.166__tar.gz

{utils_devops-0.1.161 → utils_devops-0.1.166}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: utils_devops
-Version: 0.1.161
+Version: 0.1.166
 Summary: Lightweight DevOps utilities for automation scripts: config editing (YAML/JSON/INI/.env), templating, diffing, and CLI tools
 License: MIT
 Keywords: devops,automation,nginx,cli,jinja2,yaml,config,diff,templating,logging,docker,compose,file-ops

{utils_devops-0.1.161 → utils_devops-0.1.166}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "utils_devops"
-version = "0.1.161"  # Bumped for new string features + diffing
+version = "0.1.166"  # Bumped for new string features + diffing
 description = "Lightweight DevOps utilities for automation scripts: config editing (YAML/JSON/INI/.env), templating, diffing, and CLI tools"
 authors = ["Hamed Sheikhan <sh.sheikhan.m@gmail.com>"]
 license = "MIT"

{utils_devops-0.1.161 → utils_devops-0.1.166}/src/utils_devops/extras/docker_ops.py

@@ -82,6 +82,7 @@ class ContainerInfo:
     status: str
     service: Optional[str] = None
     ports: Dict[str, Any] = field(default_factory=dict)
+    exit_code: Optional[int] = None
 
 # Core Docker operations
 def get_docker_client():
@@ -615,36 +616,82 @@ def compose_restart(compose_file: str, remove_volumes: bool = False,
 
     return None
 
+import json
+import re
+from typing import List, Optional
+
+_EXITED_RE = re.compile(r'Exited\s*\((\d+)\)', re.I)
+
 def compose_ps(compose_file: str, project_name: Optional[str] = None, env_file: Optional[str] = None) -> List[ContainerInfo]:
-    """List compose services status using detected compose command"""
-    result = run_compose_command(compose_file, ["ps" ,"-a", "--format", "json"], project_name , env_file)
-    
-    containers = []
-    try:
-        data = json.loads(result.stdout)
-        for item in data if isinstance(data, list) else [data]:
+    """List compose services status using detected compose command. Returns ContainerInfo items
+    and attempts to populate `exit_code` when available (JSON ExitCode or parsed from Status)."""
+    result = run_compose_command(compose_file, ["ps", "-a", "--format", "json"], project_name, env_file)
+    logger.info(f"DEBUG: compose_ps stdout: {result.stdout}")  # Use logger instead of print if available
+    containers: List[ContainerInfo] = []
+    lines = result.stdout.splitlines()
+    items = []
+    for line in lines:
+        if line.strip():
+            try:
+                items.append(json.loads(line))
+            except json.JSONDecodeError:
+                pass  # ignore bad lines
+    if items:
+        for item in items:
+            state_str = item.get('State') or ''
+            full_status = item.get('Status') or state_str or ''
+            exit_code = item.get('ExitCode') or item.get('Exit')
+            if exit_code is not None:
+                try:
+                    exit_code = int(exit_code)
+                except ValueError:
+                    exit_code = None
+            if exit_code is None:
+                m = _EXITED_RE.search(full_status)
+                if m:
+                    try:
+                        exit_code = int(m.group(1))
+                    except ValueError:
+                        exit_code = None
             containers.append(ContainerInfo(
-                id=item.get('ID', ''),
-                name=item.get('Name', ''),
-                image=item.get('Image', ''),
-                status=item.get('State', ''),
-                service=item.get('Service', '')
+                id=item.get('ID', '') or item.get('Id', ''),
+                name=item.get('Name', '') or item.get('Names', ''),
+                image=item.get('Image', '') or item.get('ImageName', ''),
+                status=state_str,
+                service=item.get('Service', '') or item.get('Labels', {}).get('com.docker.compose.service', ''),
+                exit_code=exit_code
             ))
-    except json.JSONDecodeError:
-        # Fallback to text parsing
-        lines = result.stdout.strip().splitlines()[1:]  # Skip header
-        for line in lines:
-            parts = line.split()
-            if len(parts) >= 4:
-                containers.append(ContainerInfo(
-                    id=parts[0],
-                    name=parts[1],
-                    image=parts[2],
-                    status=' '.join(parts[3:])
-                ))
-    
+    else:
+        # Fallback to text parsing (if not JSON format)
+        if len(lines) > 0 and 'NAME' in lines[0].upper():  # Check for header
+            for line in lines[1:]:  # skip header
+                parts = re.split(r'\s{2,}', line.strip())
+                if len(parts) >= 6:
+                    name = parts[0]
+                    image = parts[1]
+                    command = parts[2]
+                    service = parts[3]
+                    created = parts[4]
+                    status = parts[5]
+                    ports = ' '.join(parts[6:]) if len(parts) > 6 else ''
+                    exit_code = None
+                    m = _EXITED_RE.search(status)
+                    if m:
+                        try:
+                            exit_code = int(m.group(1))
+                        except ValueError:
+                            exit_code = None
+                    containers.append(ContainerInfo(
+                        id='',  # ID not in text
+                        name=name,
+                        image=image,
+                        status=status,
+                        service=service,
+                        exit_code=exit_code
+                    ))
     return containers
 
+
 # Add a function to check compose version
 def get_compose_version() -> Dict[str, str]:
     """Get detected compose command and version"""
@@ -2222,6 +2269,7 @@ def playwright_test_compose(
     keep_compose_up: bool = False,
     dry_run: bool = False,
     project_name: Optional[str] = None,
+    ui: Optional[bool] = False,
     logger: Optional[logger] = None,
 ) -> Dict[str, Any]:
     """
@@ -2368,7 +2416,8 @@ def playwright_test_compose(
         # STEP 5: Log capture
         pw_services = []
         logs_acc = {}
-        
+        if ui :
+            capture_logs = False
         if not dry_run and capture_logs:
             logger.info(f"{STICKERS['logs']} STEP 5: Capturing logs")
             
@@ -2420,54 +2469,66 @@ def playwright_test_compose(
                 
                 # Check if all services have finished
                 ps = compose_ps(playwright_compose_file, project_name=project_name_playwright)
-                running_count = 0
-                
-                for container in ps:
-                    service_name = container.service or ''
-                    if service_name in pw_services:
-                        status = (container.status or '').lower()
+
+                running = []
+                exited = []
+
+                for c in ps:
+                    if c.service in pw_services:
+                        status = (c.status or '').lower()
                         if 'running' in status or 'up' in status:
-                            running_count += 1
-                
-                if running_count == 0:
-                    finished = True
-                    logger.info(f"{STICKERS['complete']} All services completed")
-                    logger.info("")
-                else:
+                            running.append(c)
+                        elif 'exited' in status:
+                            exited.append(c)
+
+                if running:
                     time.sleep(health_interval)
+                else:
+                    finished = True
+
             
             result['logs_captured'] = logs_acc
         
         # STEP 6: Analysis
         logger.info(f"{STICKERS['analysis']} STEP 6: Analyzing results")
-        
+
         if not dry_run:
-            # Get exit codes
-            ps = compose_ps(playwright_compose_file, project_name=project_name_playwright)
-            exit_codes = {}
-            
-            for container in ps:
-                service_name = container.service or ''
-                if service_name in pw_services:
-                    exit_code = getattr(container, 'exit_code', None) or 1
-                    exit_codes[service_name] = exit_code
-            
-            # Check overall success
-            all_success = all(code == 0 for code in exit_codes.values())
-            
-            if all_success:
-                result['success'] = True
-                logger.info(f"{STICKERS['success']} " + "="*50)
-                logger.info(f"{STICKERS['success']} TESTS PASSED")
-                logger.info(f"{STICKERS['success']} Test ID: {test_id}")
-                logger.info(f"{STICKERS['success']} " + "="*50)
+            if not ui :
+                # Get exit codes
+                ps = compose_ps(playwright_compose_file, project_name=project_name_playwright)
+                exit_codes = {}
+                
+                for container in ps:
+                    service_name = container.service or ''
+                    if service_name in pw_services:
+                        exit_code = container.exit_code if container.exit_code is not None else 1  # Default to 1 if unknown
+                        exit_codes[service_name] = exit_code
+                
+                if not exit_codes:
+                    logger.error("No Playwright containers found during analysis — assuming failure")
+                    exit_codes = {pw_services[0]: 1} if pw_services else {'unknown': 1}
+                    result['error'] = 'No containers found for analysis - likely exited with error'
+                
+                # Check overall success
+                all_success = all(code == 0 for code in exit_codes.values())
+                
+                if all_success:
+                    result['success'] = True
+                    logger.info(f"{STICKERS['success']} " + "="*50)
+                    logger.info(f"{STICKERS['success']} TESTS PASSED")
+                    logger.info(f"{STICKERS['success']} Test ID: {test_id}")
+                    logger.info(f"{STICKERS['success']} " + "="*50)
+                else:
+                    result['success'] = False
+                    result['error'] = 'Playwright tests failed' if 'error' not in result else result['error']
+                    logger.error(f"{STICKERS['error']} " + "="*50)
+                    logger.error(f"{STICKERS['error']} TESTS FAILED")
+                    logger.error(f"{STICKERS['error']} Exit codes: {exit_codes}")
+                    logger.error(f"{STICKERS['error']} " + "="*50)
             else:
-                result['success'] = False
-                result['error'] = 'Playwright tests failed'
-                logger.error(f"{STICKERS['error']} " + "="*50)
-                logger.error(f"{STICKERS['error']} TESTS FAILED")
-                logger.error(f"{STICKERS['error']} Exit codes: {exit_codes}")
-                logger.error(f"{STICKERS['error']} " + "="*50)
+                result['success'] = True
+                logger.info(f"{STICKERS['info']} Ui conteiner is Up")
+                
         else:
             result['success'] = True
             logger.info(f"{STICKERS['info']} Analysis skipped (dry run)")
@@ -2478,6 +2539,8 @@ def playwright_test_compose(
         logger.info("")
         
         # Cleanup
+        if ui :
+            keep_compose_up = True
         if not keep_compose_up and not dry_run:
             logger.info(f"{STICKERS['cleanup']} Cleaning up")
             logger.info(f"{STICKERS['info']} Stopping Playwright stack...")