reverse-diagrams 1.3.4__py3-none-any.whl → 2.0.0__py3-none-any.whl

src/utils/progress.py

@@ -0,0 +1,257 @@
+"""Progress tracking utilities with rich console output."""
+import logging
+from typing import List, Optional, Any, Iterator
+from contextlib import contextmanager
+
+from rich.console import Console
+from rich.progress import (
+    Progress,
+    SpinnerColumn,
+    TextColumn,
+    BarColumn,
+    TimeElapsedColumn,
+    TaskID
+)
+from rich.panel import Panel
+from rich.text import Text
+
+logger = logging.getLogger(__name__)
+
+
+class ProgressTracker:
+    """Enhanced progress tracking with rich console output."""
+    
+    def __init__(self, console: Optional[Console] = None):
+        """
+        Initialize progress tracker.
+        
+        Args:
+            console: Rich console instance (optional)
+        """
+        self.console = console or Console()
+        self.progress = Progress(
+            SpinnerColumn(),
+            TextColumn("[progress.description]{task.description}"),
+            BarColumn(),
+            TextColumn("[progress.percentage]{task.percentage:>3.0f}%"),
+            TimeElapsedColumn(),
+            console=self.console,
+            transient=False
+        )
+        self._active_tasks: List[TaskID] = []
+    
+    @contextmanager
+    def track_operation(self, description: str, total: Optional[int] = None):
+        """
+        Context manager for tracking a single operation.
+        
+        Args:
+            description: Operation description
+            total: Total number of items (optional for indeterminate progress)
+            
+        Yields:
+            Task ID for updating progress
+        """
+        with self.progress:
+            task_id = self.progress.add_task(
+                f"[cyan]{description}",
+                total=total
+            )
+            self._active_tasks.append(task_id)
+            
+            try:
+                yield task_id
+            finally:
+                if task_id in self._active_tasks:
+                    self._active_tasks.remove(task_id)
+    
+    def update_progress(self, task_id: TaskID, advance: int = 1, description: Optional[str] = None):
+        """
+        Update progress for a task.
+        
+        Args:
+            task_id: Task identifier
+            advance: Number of items to advance
+            description: Updated description (optional)
+        """
+        kwargs = {"advance": advance}
+        if description:
+            kwargs["description"] = f"[cyan]{description}"
+        
+        self.progress.update(task_id, **kwargs)
+    
+    def complete_task(self, task_id: TaskID, description: Optional[str] = None):
+        """
+        Mark a task as completed.
+        
+        Args:
+            task_id: Task identifier
+            description: Completion message (optional)
+        """
+        if description:
+            self.progress.update(task_id, description=f"[green]✓ {description}")
+        else:
+            current_desc = self.progress.tasks[task_id].description
+            self.progress.update(task_id, description=f"[green]✓ {current_desc}")
+    
+    def track_aws_operations(self, operations: List[str]) -> Iterator[TaskID]:
+        """
+        Track multiple AWS operations with progress bars.
+        
+        Args:
+            operations: List of operation descriptions
+            
+        Yields:
+            Task IDs for each operation
+        """
+        with self.progress:
+            task_ids = []
+            
+            # Create tasks for all operations
+            for operation in operations:
+                task_id = self.progress.add_task(
+                    f"[cyan]{operation}",
+                    total=None  # Indeterminate progress
+                )
+                task_ids.append(task_id)
+            
+            # Yield task IDs for updating
+            for task_id in task_ids:
+                yield task_id
+    
+    def show_summary(self, title: str, items: List[str], color: str = "blue"):
+        """
+        Display a summary panel with items.
+        
+        Args:
+            title: Panel title
+            items: List of items to display
+            color: Panel border color
+        """
+        if not items:
+            content = Text("No items to display", style="italic")
+        else:
+            content = Text("\n".join(f"• {item}" for item in items))
+        
+        panel = Panel(
+            content,
+            title=f"[bold {color}]{title}[/bold {color}]",
+            border_style=color,
+            padding=(1, 2)
+        )
+        
+        self.console.print(panel)
+    
+    def show_error(self, message: str, details: Optional[str] = None):
+        """
+        Display an error message with optional details.
+        
+        Args:
+            message: Error message
+            details: Additional error details (optional)
+        """
+        content = Text(message, style="bold red")
+        if details:
+            content.append("\n\n")
+            content.append(details, style="red")
+        
+        panel = Panel(
+            content,
+            title="[bold red]❌ Error[/bold red]",
+            border_style="red",
+            padding=(1, 2)
+        )
+        
+        self.console.print(panel)
+    
+    def show_success(self, message: str, details: Optional[str] = None):
+        """
+        Display a success message with optional details.
+        
+        Args:
+            message: Success message
+            details: Additional details (optional)
+        """
+        content = Text(message, style="bold green")
+        if details:
+            content.append("\n\n")
+            content.append(details, style="green")
+        
+        panel = Panel(
+            content,
+            title="[bold green]✅ Success[/bold green]",
+            border_style="green",
+            padding=(1, 2)
+        )
+        
+        self.console.print(panel)
+    
+    def show_warning(self, message: str, details: Optional[str] = None):
+        """
+        Display a warning message with optional details.
+        
+        Args:
+            message: Warning message
+            details: Additional details (optional)
+        """
+        content = Text(message, style="bold yellow")
+        if details:
+            content.append("\n\n")
+            content.append(details, style="yellow")
+        
+        panel = Panel(
+            content,
+            title="[bold yellow]⚠️  Warning[/bold yellow]",
+            border_style="yellow",
+            padding=(1, 2)
+        )
+        
+        self.console.print(panel)
+
+
+# Global progress tracker instance
+_progress_tracker: Optional[ProgressTracker] = None
+
+
+def get_progress_tracker() -> ProgressTracker:
+    """Get or create global progress tracker instance."""
+    global _progress_tracker
+    if _progress_tracker is None:
+        _progress_tracker = ProgressTracker()
+    return _progress_tracker
+
+
+def track_items(items: List[Any], description: str = "Processing items") -> Iterator[Any]:
+    """
+    Track progress while iterating over items.
+    
+    Args:
+        items: List of items to process
+        description: Progress description
+        
+    Yields:
+        Items from the list with progress tracking
+    """
+    tracker = get_progress_tracker()
+    
+    with tracker.track_operation(description, total=len(items)) as task_id:
+        for item in items:
+            yield item
+            tracker.update_progress(task_id)
+
+
+@contextmanager
+def track_operation(description: str, total: Optional[int] = None):
+    """
+    Context manager for tracking operations with global progress tracker.
+    
+    Args:
+        description: Operation description
+        total: Total number of items (optional)
+        
+    Yields:
+        Task ID for updating progress
+    """
+    tracker = get_progress_tracker()
+    with tracker.track_operation(description, total) as task_id:
+        yield task_id

src/version.py

@@ -1,2 +1,2 @@
 """Set version file."""
-__version__ = "1.3.4"
+__version__ = "2.0.0"

reverse_diagrams-1.3.4.dist-info/METADATA

@@ -1,247 +0,0 @@
-Metadata-Version: 2.3
-Name: reverse_diagrams
-Version: 1.3.4
-Summary: Continuous Documentation Tool - Documentation as Code Tool -   This package create reverse diagrams  based on your current state in your cloud environment using diagrams library
-Project-URL: Homepage, https://github.com/velez94/reverse_diagrams
-Project-URL: Bug Tracker, https://github.com/velez94/reverse_diagrams/issues
-Author-email: Alejandro Velez <avelez@labvel.io>
-License: Copyright [2024] [Alejandro Velez]
-        
-        Licensed under the Apache License, Version 2.0 (the "License");
-        you may not use this file except in compliance with the License.
-        You may obtain a copy of the License at
-        
-            http://www.apache.org/licenses/LICENSE-2.0
-        
-        Unless required by applicable law or agreed to in writing, software
-        distributed under the License is distributed on an "AS IS" BASIS,
-        WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-        See the License for the specific language governing permissions and
-        limitations under the License.
-License-File: LICENSE
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.8
-Requires-Dist: argcomplete>=3.4.0
-Requires-Dist: boto3>=1.26.44
-Requires-Dist: colorama>=0.4.4
-Requires-Dist: diagrams>=0.23.4
-Requires-Dist: emoji>=2.2.0
-Requires-Dist: inquirer>=3.1.4
-Requires-Dist: rich>=13.7.0
-Description-Content-Type: text/markdown
-
-<!-- START doctoc generated TOC please keep comment here to allow auto update -->
-<!-- DON'T EDIT THIS SECTION, INSTEAD RE-RUN doctoc TO UPDATE -->
-**Table of Contents**  
-
-- [Reverse Diagrams](#reverse-diagrams)
-- [Requirement](#requirement)
-- [Install](#install)
-- [Use](#use)
-  - [Subcommands](#subcommands)
-    - [watch](#watch)
-- [Service supported](#service-supported)
-  - [AWS Organizations](#aws-organizations)
-  - [Identity and Access Manager Center (SSO)](#identity-and-access-manager-center-sso)
-- [Additional Commands](#additional-commands)
-  - [watch](#watch-1)
-    - [Options](#options)
-    - [Combine the options](#combine-the-options)
-  - [Extras](#extras)
-    - [Enable autocomplete](#enable-autocomplete)
-
-<!-- END doctoc generated TOC please keep comment here to allow auto update -->
-
-# Reverse Diagrams
-
-> Continuous Documentation Tool - Documentation as Code Tool
-
-This package create diagrams and help to audit your services from your shell.
-
-![Complete demo](docs/images/complete_demo.gif)
-# Requirement
-
-AWS programmatic access using AWS CLI.  [Configuring the AWS CLI](https://docs.aws.amazon.com/cli/latest/userguide/cli-chap-configure.html)
-
-
-# Install 
-
-`pip install reverse-diagrams`
-
-# Use
-
-The following are the available options
-
-```commandline
-$ reverse_diagrams  -h
-usage: reverse_diagrams [-h] [-p PROFILE] [-od OUTPUT_DIR_PATH] [-r REGION] [-o] [-i] [-a] [-v] [-d] {watch} ...
-
-Create architecture diagram, inspect and audit your AWS services from your current state.
-
-options:
-  -h, --help            show this help message and exit
-  -p PROFILE, --profile PROFILE
-                        AWS cli profile for AWS Apis
-  -od OUTPUT_DIR_PATH, --output_dir_path OUTPUT_DIR_PATH
-                        Name of folder to save the diagrams python code files
-  -r REGION, --region REGION
-                        AWS region
-  -o, --graph_organization
-                        Set if you want to create graph for your organization
-  -i, --graph_identity  Set if you want to create graph for your IAM Center
-  -a, --auto_create     Create Automatically diagrams
-  -v, --version         Show version
-  -d, --debug           Debug Mode
-
-Commands:
-  Command and functionalities
-
-  {watch}               reverse_diagrams Commands
-    watch               Create pretty console viewexample: reverse_diagrams watch -wi diagrams/json/account_assignments.json
-
-Thanks for using reverse_diagrams!
-
-```
-For example: 
-
-```commandline
-reverse_diagrams -p labvel-master -o -i -r us-east-1
-
-❇️ Describe Organization 
-❇️ Getting Organization Info
-❇️ Listing Organizational Units 
-❇️ Getting the Account list info
-ℹ️  There are 11 Accounts in your organization
-ℹ️  The accounts are stored in diagrams/json/organizations.json 
-❇️ Creating diagrams in diagrams/code
-❇️ Getting Identity store instance info
-❇️ List groups
-ℹ️  There are 10 Groups in your Identity Store
-❇️ Get groups and Users info
-Getting groups members... ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:07
-Getting account assignments ... ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:05:23
-Create user and groups assignments ... ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00
-❇️ Getting account assignments, users and groups
-ℹ️  The accounts are stored in diagrams/json/account_assignments.json
-ℹ️  The accounts are stored in diagrams/json/groups.json
-❇️ Creating diagrams in diagrams/code
-
-```
-Then run `python3 graph_org.py` to create a png screenshot (`organizations-state.png`) for your current state.
-
-> Both files are saved into the current directory.
-
-```commandline
-$  reverse_diagrams -p labvel-master -o -r us-east-2
-Date: 2022-12-17 22:44:07.623260
-❇️ Getting Organization Info
-❇️ The Organizational Units list 
-❇️ Getting the Account list info
-Run -> python3 graph_org.py 
-
-
-$ python3 graph_org.py 
-$ ls 
-graph_org.py
-organizations-state.png
-```
-For example:
-
-![Organizations Diagram](./docs/images/organizations-state-copy.png)
-
-Now you can edit `graph_org.py` file or add to your repositories for keeping the documentation update.
-## Subcommands
-
-### watch
-
-Watch the result in console with a beautiful print style.
-```commandline
- reverse_diagrams  watch  -h
-usage: reverse_diagrams watch [-h] [-wo WATCH_GRAPH_ORGANIZATION] [-wi WATCH_GRAPH_IDENTITY] [-wa WATCH_GRAPH_ACCOUNTS_ASSIGNMENTS]
-
-Create view of diagrams in console based on kind of diagram and json file.
-
-options:
-  -h, --help            show this help message and exit
-
-Create view of diagrams in console based on kind of diagram and json file.:
-  -wo WATCH_GRAPH_ORGANIZATION, --watch_graph_organization WATCH_GRAPH_ORGANIZATION
-                        Set if you want to see graph for your organization structure summary. For example: reverse_diagrams watch watch -wi diagrams/json/organizations.json        
-  -wi WATCH_GRAPH_IDENTITY, --watch_graph_identity WATCH_GRAPH_IDENTITY
-                        Set if you want to see graph for your groups and users. For example: reverse_diagrams watch watch -wi diagrams/json/groups.json
-  -wa WATCH_GRAPH_ACCOUNTS_ASSIGNMENTS, --watch_graph_accounts_assignments WATCH_GRAPH_ACCOUNTS_ASSIGNMENTS
-                        Set if you want to see graph for your IAM Center- Accounts assignments. For example: reverse_diagrams watch watch -wi
-                        diagrams/json/account_assignments.json.jso
-```
-
-# Service supported
-
-## AWS Organizations
-
-```commandline
-reverse_diagrams -p my-profile -o -r us-east-2
-```
-## Identity and Access Manager Center (SSO)
-
-```commandline
-reverse_diagrams -p my-profile -i -r us-east-2
-```
-# Additional Commands
-
-## watch
-You can watch the configuration and summary in your shell based on json files generated previously.
-
-### Options
-
-```commandline
-$ reverse_diagrams watch -h 
-usage: reverse_diagrams watch [-h] [-wo WATCH_GRAPH_ORGANIZATION] [-wi WATCH_GRAPH_IDENTITY] [-wa WATCH_GRAPH_ACCOUNTS_ASSIGNMENTS]
-
-Create view of diagrams in console based on kind of diagram and json file.
-
-options:
-  -h, --help            show this help message and exit
-
-Create view of diagrams in console based on kind of diagram and json file.:
-  -wo WATCH_GRAPH_ORGANIZATION, --watch_graph_organization WATCH_GRAPH_ORGANIZATION
-                        Set if you want to see graph for your organization structure summary. For example: reverse_diagrams watch watch -wo diagrams/json/organizations.json
-  -wi WATCH_GRAPH_IDENTITY, --watch_graph_identity WATCH_GRAPH_IDENTITY
-                        Set if you want to see graph for your groups and users. For example: reverse_diagrams watch watch -wi diagrams/json/groups.json
-  -wa WATCH_GRAPH_ACCOUNTS_ASSIGNMENTS, --watch_graph_accounts_assignments WATCH_GRAPH_ACCOUNTS_ASSIGNMENTS
-                        Set if you want to see graph for your IAM Center- Accounts assignments. For example: reverse_diagrams watch watch -wa diagrams/json/account_assignments.json
-
-```
-
-For example, to watch account assignments: 
-
-![view Acoount assigments](docs/images/show_console_view.gif)
-
-### Combine the options
-
-```commandline
-reverse_diagrams  -p my-profile -o -i -r us-east-2
-```
-
-## Extras
-### Enable autocomplete
-Argcomplete provides easy, extensible command line tab completion of arguments for your Python application.
-
-It makes two assumptions:
-
-* You’re using bash or zsh as your shell
-
-* You’re using argparse to manage your command line arguments/options
-
-Argcomplete is particularly useful if your program has lots of options or subparsers, and if your program can dynamically suggest completions for your argument/option values (for example, if the user is browsing resources over the network).
-Run: 
-```bash
-activate-global-python-argcomplete
-```
-and to make sure that bash knows about this script, you use
-```bash
-
-echo 'eval "$(register-python-argcomplete reverse_diagrams)"' >> ~/.bashrc
-source ~/.bashrc
-
-```

reverse_diagrams-1.3.4.dist-info/RECORD

@@ -1,21 +0,0 @@
-src/__init__.py,sha256=lOE8TAJF_WdxgxEBxVz9t8dC3s3t2c21VWYPrzfznkE,17
-src/reverse_diagrams.py,sha256=YBLLz3Jg6X-5FZUTsNYi9NBWGQUaIGFeJZNs94TNCbs,4208
-src/version.py,sha256=-QqZOUBRTMXL9RNQivjJyu4d9ZqqRQfFdj6J9yLdgs8,48
-src/aws/__init__.py,sha256=lOE8TAJF_WdxgxEBxVz9t8dC3s3t2c21VWYPrzfznkE,17
-src/aws/describe_identity_store.py,sha256=PJYpAXsqCeVMPfv72Q2nTuolXMnG4dah_HRuyu0hIkY,15450
-src/aws/describe_organization.py,sha256=6154J-jnGGql_uzr5DCeSNsNEiMblkOPNq5qF1UAjEw,14268
-src/aws/describe_sso.py,sha256=If0EJ1YgrUCRQPyf0KjYhTG2J0GV-R4h-gIb7PZmoxA,4917
-src/banner/__init__.py,sha256=lOE8TAJF_WdxgxEBxVz9t8dC3s3t2c21VWYPrzfznkE,17
-src/banner/banner.py,sha256=XREWjVLp1-PHVdfBSoADgw0LvRpYU5u7NPaUraL9Els,3893
-src/dgms/__init__.py,sha256=lOE8TAJF_WdxgxEBxVz9t8dC3s3t2c21VWYPrzfznkE,17
-src/dgms/graph_mapper.py,sha256=xutxzxLCIGSBfHCAF-BnWXN21bjzusA43ZhKtXUir2E,7714
-src/dgms/graph_template.py,sha256=IOEmTk9A0xqEhtdNnaFrVDJlBF0ggW0WpxosW4nbcQo,1217
-src/reports/__init__.py,sha256=lOE8TAJF_WdxgxEBxVz9t8dC3s3t2c21VWYPrzfznkE,17
-src/reports/console_view.py,sha256=1_ET53Bn_vKlLmqr0Sih5szP5kNFCQRH4A8tXJgip2Y,5832
-src/reports/save_results.py,sha256=g-UphsQ_tfDhvNH3QNV7j68TWWvy44vbbjaNyn4KWhU,719
-src/reports/tes.py,sha256=c6TdxhAedfHjPrqaRw0nucdCEQOM9xKsgw35IOIDdYs,17691
-reverse_diagrams-1.3.4.dist-info/METADATA,sha256=iT_qL7KXmzpj0_1SzX8H2GCi5ye0evefbyKnLPyHE3w,9624
-reverse_diagrams-1.3.4.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-reverse_diagrams-1.3.4.dist-info/entry_points.txt,sha256=VZNkrc7qUDbddTCH3pGd83EhUT3PHTx9MzpAk6bb6qc,63
-reverse_diagrams-1.3.4.dist-info/licenses/LICENSE,sha256=o6nDaQ7M9xbR0L7HSJ3A-1JbBPoZro_zhVPO4-M5CAQ,571
-reverse_diagrams-1.3.4.dist-info/RECORD,,