reverse-diagrams 0.2.6__py3-none-any.whl → 1.0.0__py3-none-any.whl

{reverse_diagrams-0.2.6.dist-info → reverse_diagrams-1.0.0.dist-info}/METADATA

@@ -1,41 +1,32 @@
 Metadata-Version: 2.1
 Name: reverse_diagrams
-Version: 0.2.6
-Summary: Continuous Documentation Tool - Documentation as Code Tool -   This package create reverse diagrams  based on your current state in your cloud environment
+Version: 1.0.0
+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: MIT License
+License: Copyright [2024] [Alejandro Velez]
         
-        Copyright (c) [2023] [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
         
-        Permission is hereby granted, free of charge, to any person obtaining a copy
-        of this software and associated documentation files (the "Software"), to deal
-        in the Software without restriction, including without limitation the rights
-        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-        copies of the Software, and to permit persons to whom the Software is
-        furnished to do so, subject to the following conditions:
+            http://www.apache.org/licenses/LICENSE-2.0
         
-        The above copyright notice and this permission notice shall be included in all
-        copies or substantial portions of the Software.
-        
-        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-        SOFTWARE.
+        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: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.8
 Requires-Dist: boto3>=1.26.44
 Requires-Dist: colorama>=0.4.4
-Requires-Dist: diagrams>=0.22.0
+Requires-Dist: diagrams>=0.23.4
 Requires-Dist: emoji>=2.2.0
-Requires-Dist: jsonschema>=3.2.0
+Requires-Dist: rich>=13.7.0
 Description-Content-Type: text/markdown
 
 <!-- START doctoc generated TOC please keep comment here to allow auto update -->
@@ -156,3 +147,25 @@ reverse_diagrams -c aws -p my-profile -i -r us-east-2
 reverse_diagrams -c aws -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.0.0.dist-info/RECORD

@@ -0,0 +1,20 @@
+src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/reverse_diagrams.py,sha256=xC4alfmJ4CZcgnjShXE13ZvehhSuxSkrQxhs4LX1FVU,9496
+src/version.py,sha256=GdBzLFWoHZyrTuCHs8_6o2940bb1o4TwZpxxsOIS-kI,48
+src/aws/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/aws/describe_identity_store.py,sha256=2mmniQowKhatfg7ekLdJASz424j0gkP9TJ-ArEeb8Po,8838
+src/aws/describe_organization.py,sha256=S9r-lCy3NX1e1qpLw_gMl_-nbeE5RXcISLMdThDOp7w,4731
+src/aws/describe_sso.py,sha256=LMejRRupMckB6r2akK8Nyu35AFzyvprwh4QnzAQvmoA,4832
+src/banner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/banner/banner.py,sha256=aGFVaW7RVdhgMi8b3YeC6MQodhJd01MO3iTTHlQRw8s,979
+src/dgms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/dgms/graph_mapper.py,sha256=vzBgjdFzO7nseqTXp0Z68-f5OgtZaDsThfMNy8Q7kjg,7402
+src/dgms/graph_template.py,sha256=QMTsbYPt9uZkWS9r7h8UDc7NP7BKcCP1j523soeQrOQ,1176
+src/export_report/export_csv.py,sha256=7j53bcuXK1x6ciGzq88psfyguhQWo36uVFYxp0y8p34,92
+src/reports/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/reports/save_results.py,sha256=eOYGmp1-2Kfaf5k-cUZ7UIAQoBm59ta8sxlDUoZCfK4,689
+reverse_diagrams-1.0.0.dist-info/METADATA,sha256=udbRPGZhORkHUAyUOYW9DLhqjb6dt6QcjPytoEaWUTQ,5528
+reverse_diagrams-1.0.0.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+reverse_diagrams-1.0.0.dist-info/entry_points.txt,sha256=VZNkrc7qUDbddTCH3pGd83EhUT3PHTx9MzpAk6bb6qc,63
+reverse_diagrams-1.0.0.dist-info/licenses/LICENSE,sha256=o6nDaQ7M9xbR0L7HSJ3A-1JbBPoZro_zhVPO4-M5CAQ,571
+reverse_diagrams-1.0.0.dist-info/RECORD,,

reverse_diagrams-1.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,13 @@
+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.

src/aws/describe_identity_store.py

@@ -1,29 +1,29 @@
-import boto3
-from .describe_sso import list_account_assignments
-from colorama import Fore
 import logging
 
+from colorama import Fore
+from rich.progress import track
+
+from .describe_sso import client, list_account_assignments
+
+
+def list_groups_pag(identity_store_id, region, next_token: str = None):
+    identity_client = client("identitystore", region_name=region)
 
-def list_groups_pag(identity_store_id, client=boto3.client('identitystore', region_name="us-east-2"),
-                    next_token: str = None):
-    paginator = client.get_paginator('list_groups')
+    paginator = identity_client.get_paginator("list_groups")
     response_iterator = paginator.paginate(
         IdentityStoreId=identity_store_id,
-        PaginationConfig={
-            'MaxItems': 1000,
-            'PageSize': 4,
-            'StartingToken': next_token
-        }
+        PaginationConfig={"MaxItems": 1000, "PageSize": 4, "StartingToken": next_token},
     )
     response = response_iterator.build_full_result()
     logging.info(response_iterator.build_full_result())
     return response["Groups"]
 
 
-def list_groups(identity_store_id, client=boto3.client('identitystore', region_name="us-east-2"), ):
-    groups = client.list_groups(
-        IdentityStoreId=identity_store_id,
-        MaxResults=20
+def list_groups(identity_store_id, region):
+    identity_client = client("identitystore", region_name=region)
+
+    groups = identity_client.list_groups(
+        IdentityStoreId=identity_store_id, MaxResults=20
     )
 
     logging.info(groups)
@@ -32,7 +32,11 @@ def list_groups(identity_store_id, client=boto3.client('identitystore', region_n
 
     if len(groups["Groups"]) >= 20:
         logging.info("Paginating ...")
-        ad_groups = list_groups_pag(identity_store_id=identity_store_id, client=client, next_token=groups["NextToken"])
+        ad_groups = list_groups_pag(
+            identity_store_id=identity_store_id,
+            region=region,
+            next_token=groups["NextToken"],
+        )
         for ad in ad_groups:
             l_groups.append(ad)
         logging.info(f"You have {len(l_groups)} Groups")
@@ -40,39 +44,62 @@ def list_groups(identity_store_id, client=boto3.client('identitystore', region_n
     return l_groups
 
 
-def list_users(identity_store_id, client=boto3.client('identitystore', region_name="us-east-2"), ):
-    response = client.list_users(
+# define pagination list user function
+def list_users_pag(identity_store_id, region, next_token: str = None):
+    identity_client = client("identitystore", region_name=region)
+    paginator = identity_client.get_paginator("list_users")
+    response_iterator = paginator.paginate(
         IdentityStoreId=identity_store_id,
+        PaginationConfig={"MaxItems": 1000, "PageSize": 4, "StartingToken": next_token},
+    )
+    return response_iterator["Users"]
 
+
+def list_users(identity_store_id, region):
+    identity_client = client("identitystore", region_name=region)
+
+    response = identity_client.list_users(
+        IdentityStoreId=identity_store_id,
     )
+    # create pagination option
+    users = response["Users"]
+    if len(users) >= 20:
+        logging.info("Paginating ...")
+        ad_users = list_users_pag(
+            identity_store_id=identity_store_id,
+            region=region,
+            next_token=response["NextToken"],
+        )
+        for ad in ad_users:
+            users.append(ad)
+        logging.info(f"You have {len(users)} Users")
 
-    return response["Users"]
+    return users
 
 
-def get_members_pag(identity_store_id, client=boto3.client('identitystore', region_name="us-east-2"),
-                    next_token: str = None):
-    paginator = client.get_paginator('list_group_memberships')
+def get_members_pag(identity_store_id, region, next_token: str = None):
+    identity_client = client("identitystore", region_name=region)
+
+    paginator = identity_client.get_paginator("list_group_memberships")
     response_iterator = paginator.paginate(
         IdentityStoreId=identity_store_id,
-        PaginationConfig={
-            'MaxItems': 1000,
-            'PageSize': 4,
-            'StartingToken': next_token
-        }
+        PaginationConfig={"MaxItems": 1000, "PageSize": 4, "StartingToken": next_token},
     )
     response = response_iterator.build_full_result()
     logging.info(response_iterator.build_full_result())
     return response["GroupMemberships"]
 
 
-def get_members(identity_store_id, groups, client=boto3.client('identitystore', region_name="us-east-2")):
+def get_members(identity_store_id, groups, region):
+    l_client = client("identitystore", region_name=region)
     group_members = []
-    for g in groups:
-        response = client.list_group_memberships(
+    for g, y in zip(
+        groups, track(range(len(groups) - 1), description="Getting groups members...")
+    ):
+        response = l_client.list_group_memberships(
             IdentityStoreId=identity_store_id,
             GroupId=g["GroupId"],
             MaxResults=20,
-
         )
         members = response["GroupMemberships"]
         logging.info(members)
@@ -81,18 +108,21 @@ def get_members(identity_store_id, groups, client=boto3.client('identitystore',
 
         if len(members) >= 20:
             logging.info("Paginating ...")
-            ad_members = get_members_pag(identity_store_id=identity_store_id, client=client,
-                                         next_token=response["NextToken"])
+            ad_members = get_members_pag(
+                identity_store_id=identity_store_id,
+                region=region,
+                next_token=response["NextToken"],
+            )
             for ad in ad_members:
                 members.append(ad)
-            logging.info(f"You have {len(ad_members)} Members")
-            print(f"You have {len(ad_members)} Members")
+                logging.info(f"You have {len(ad_members)} Members")
 
         group_members.append(
-            {"group_id": g["GroupId"],
-             "group_name": g["DisplayName"],
-             "members": members
-             }
+            {
+                "group_id": g["GroupId"],
+                "group_name": g["DisplayName"],
+                "members": members,
+            }
         )
 
     return group_members
@@ -103,7 +133,7 @@ def list_group_memberships(identitystore_client, group_name, pagination=True):
     Lists memberships for a group in an AWS SSO identity store.
 
     Args:
-        identitystore_client (boto3.client): Boto3 SSO identity store client
+        identitystore_client (client): Boto3 SSO identity store client
         group_name (str): Name of the group to list memberships for
         pagination (bool): Whether to enable result pagination (default: True)
 
@@ -111,16 +141,16 @@ def list_group_memberships(identitystore_client, group_name, pagination=True):
         list: List of member objects
     """
 
-    params = {'GroupName': group_name}
+    params = {"GroupName": group_name}
     members = []
 
     if pagination:
-        paginator = identitystore_client.get_paginator('list_group_memberships')
+        paginator = identitystore_client.get_paginator("list_group_memberships")
         for page in paginator.paginate(**params):
-            members.extend(page['Members'])
+            members.extend(page["Members"])
     else:
         response = identitystore_client.list_group_memberships(**params)
-        members.extend(response['Members'])
+        members.extend(response["Members"])
 
     return members
 
@@ -143,7 +173,7 @@ def l_groups_to_d_groups(l_groups: list = None):
     """
     names = []
     for a in l_groups:
-        names.append(a['group_name'])
+        names.append(a["group_name"])
     logging.info(names)
 
     d_user_groups = dict(zip(names, l_groups))
@@ -151,36 +181,72 @@ def l_groups_to_d_groups(l_groups: list = None):
     return d_user_groups
 
 
-def extend_account_assignments(accounts_list, permissions_sets, store_arn,
-                               client_sso=boto3.client('identitystore', region_name="us-east-2")):
+def extend_account_assignments(accounts_list, permissions_sets, store_arn, region):
     account_assignments = []
-    for p in permissions_sets:
-
+    for p, y in zip(
+        permissions_sets,
+        track(
+            range(len(permissions_sets) - 1),
+            description="Getting account assignments ...",
+        ),
+    ):
         for ac in accounts_list:
-            assign = list_account_assignments(instance_arn=store_arn, account_id=ac["Id"], client=client_sso,
-                                              permission_set_arn=p)
+            assign = list_account_assignments(
+                instance_arn=store_arn,
+                account_id=ac["Id"],
+                region=region,
+                permission_set_arn=p,
+            )
             logging.debug(f"AccountAssignments  {assign}")
             for a in assign:
                 account_assignments.append(a)
     return account_assignments
 
 
-def add_users_and_groups_assign(account_assignments_list, user_and_group_list, user_list,
-                                list_permissions_set_arn_name):
-    for a in account_assignments_list:
+def add_users_and_groups_assign(
+    account_assignments_list,
+    user_and_group_list,
+    user_list,
+    list_permissions_set_arn_name,
+):
+    for a, y in zip(
+        account_assignments_list,
+        track(
+            range(len(account_assignments_list) - 1),
+            description="Create user and groups assignments ...",
+        ),
+    ):
         for g in user_and_group_list:
-            if len(a) > 0 and a['PrincipalType'] == 'GROUP' and g["group_id"] == a['PrincipalId']:
-                print(Fore.YELLOW +
-                      f"Account {a['AccountId']} assign to {a['PrincipalType']} {g['group_name']} with permission set {list_permissions_set_arn_name[a['PermissionSetArn']]} or {a['PermissionSetArn']}" + Fore.RESET)
-
-                a["GroupName"] = g['group_name']
-                a["PermissionSetName"] = list_permissions_set_arn_name[a['PermissionSetArn']]
+            if (
+                len(a) > 0
+                and a["PrincipalType"] == "GROUP"
+                and g["group_id"] == a["PrincipalId"]
+            ):
+                logging.info(
+                    Fore.YELLOW
+                    + f"Account {a['AccountId']} assign to {a['PrincipalType']} {g['group_name']} with permission set {list_permissions_set_arn_name[a['PermissionSetArn']]} or {a['PermissionSetArn']}"
+                    + Fore.RESET
+                )
+
+                a["GroupName"] = g["group_name"]
+                a["PermissionSetName"] = list_permissions_set_arn_name[
+                    a["PermissionSetArn"]
+                ]
         for u in user_list:
-            if len(a) > 0 and a['PrincipalType'] == 'USER' and u["UserId"] == a['PrincipalId']:
-                print(Fore.YELLOW +
-                      f"Account {a['AccountId']} assign to {a['PrincipalType']} {u['UserName']} with permission set {a['PermissionSetArn']} or {list_permissions_set_arn_name[a['PermissionSetArn']]}" + Fore.RESET)
-                a["UserName"] = u['UserName']
-                a["PermissionSetName"] = list_permissions_set_arn_name[a['PermissionSetArn']]
+            if (
+                len(a) > 0
+                and a["PrincipalType"] == "USER"
+                and u["UserId"] == a["PrincipalId"]
+            ):
+                logging.info(
+                    Fore.YELLOW
+                    + f"Account {a['AccountId']} assign to {a['PrincipalType']} {u['UserName']} with permission set {a['PermissionSetArn']} or {list_permissions_set_arn_name[a['PermissionSetArn']]}"
+                    + Fore.RESET
+                )
+                a["UserName"] = u["UserName"]
+                a["PermissionSetName"] = list_permissions_set_arn_name[
+                    a["PermissionSetArn"]
+                ]
     logging.debug(f"Account Assignments --> {account_assignments_list}")
     return account_assignments_list
 
@@ -190,7 +256,6 @@ def order_accounts_assignments_list(accounts_dict, account_assignments):
     for ac in accounts_dict:
         final_account_assignments[ac["Name"]] = []
         for a in account_assignments:
-
             if ac["Id"] == a["AccountId"]:
                 final_account_assignments[ac["Name"]].append(a)
 

src/aws/describe_organization.py

@@ -1,36 +1,79 @@
-import boto3
 import logging
 
+from colorama import Fore
 
-def describe_organization(client=boto3.client('organizations')):
-    organization = client.describe_organization()
+from .describe_sso import client
+
+
+def describe_organization(region):
+    """
+    Describe the organization.
+
+    :param region: AWS Region
+    :return:
+    """
+    print(f"{Fore.GREEN}❇️ Describe Organization {Fore.RESET}")
+    org_client = client("organizations", region_name=region)
+    organization = org_client.describe_organization()
     organization = organization["Organization"]
     return organization
 
 
-def list_roots(client=boto3.client('organizations')):
-    roots = client.list_roots(
-
-    )
+def list_roots(region):
+    """
+    List the roots
+    :param region: AWS Region
+    :return:
+    """
+    org_client = client("organizations", region_name=region)
+    roots = org_client.list_roots()
     return roots["Roots"]
 
 
-def list_organizational_units(parent_id, client=boto3.client('organizations'), org_units=[]):
-    ous = client.list_organizational_units_for_parent(
+# def list organizational units pagination
+def list_organizational_units_pag(parent_id, region, next_token=None):
+    org_client = client("organizations", region_name=region)
+    paginator = org_client.get_paginator("list_organizational_units_for_parent")
+    response_iterator = paginator.paginate(
         ParentId=parent_id,
+        PaginationConfig={"MaxItems": 1000, "PageSize": 4, "StartingToken": next_token},
+    )
+    return response_iterator["OrganizationalUnits"]
+
 
+def list_organizational_units(parent_id, region, org_units=None):
+    org_client = client("organizations", region_name=region)
+    if org_units is None:
+        org_units = []
+    ous = org_client.list_organizational_units_for_parent(
+        ParentId=parent_id,
+        MaxResults=20,
     )
+    ous = ous["OrganizationalUnits"]
+
+    if len(ous) >= 20:
+        logging.info("Paginating ...")
+        add_ous = list_organizational_units_pag(
+            parent_id, region, next_token=ous["NextToken"]
+        )
+        for ou in add_ous:
+            ous.append(ou)
+        logging.debug(add_ous)
 
-    for o in ous["OrganizationalUnits"]:
+    for o in ous:
         org_units.append(o)
-    logging.debug("The parent Id is: ", parent_id)
+    logging.debug(
+        f"The parent Id is: {parent_id}",
+    )
     logging.debug(ous)
     if len(ous) > 0:
         for ou in ous["OrganizationalUnits"]:
             logging.debug(ou)
             if "Id" in ou.keys():
-                logging.debug("Search nested for: ", ou["Name"])
-                ous_next = list_organizational_units(ou["Id"], client=client, org_units=org_units)
+                logging.debug(f"Search nested for: {ou['Name']}")
+                ous_next = list_organizational_units(
+                    ou["Id"], region=region, org_units=org_units
+                )
                 logging.debug(ous_next)
                 if len(ous_next) > 0:
                     logging.debug("Find Netsted")
@@ -38,50 +81,61 @@ def list_organizational_units(parent_id, client=boto3.client('organizations'), o
     return org_units
 
 
-def list_parents(child_id, client=boto3.client('organizations')):
-    response = client.list_parents(
-        ChildId=child_id,
+def list_parents(child_id, region):
+    """
+    List the parents of a child.
 
+    :param child_id:
+    :param region:
+    :return:
+    """
+    org_client = client("organizations", region_name=region)
+    response = org_client.list_parents(
+        ChildId=child_id,
     )
     return response["Parents"]
 
 
-def index_ous(list_ous, client=boto3.client('organizations')):
+def index_ous(list_ous, region):
+    """
+    Index the parents of a child.
+
+    :param list_ous:
+    :param region:
+    :return list_ous:
+
+    """
+    org_client = client("organizations", region_name=region)
     for ou in list_ous:
         if "Id" in ou.keys() and len(ou) > 0:
-            response = client.list_parents(
+            response = org_client.list_parents(
                 ChildId=ou["Id"],
-
             )
             logging.debug(response["Parents"])
             ou["Parents"] = response["Parents"]
     return list_ous
 
 
-def list_accounts_pag(client=boto3.client('organizations'), next_token: str = None):
-    paginator = client.get_paginator('list_accounts')
+def list_accounts_pag(region, next_token: str = None):
+    org_client = client("organizations", region_name=region)
+    paginator = org_client.get_paginator("list_accounts")
     response_iterator = paginator.paginate(
-        PaginationConfig={
-            'MaxItems': 1000,
-            'PageSize': 20,
-            'StartingToken': next_token
-        }
+        PaginationConfig={"MaxItems": 1000, "PageSize": 20, "StartingToken": next_token}
     )
     response = response_iterator.build_full_result()
     logging.info(response_iterator.build_full_result())
     return response["Accounts"]
 
 
-def list_accounts(client=boto3.client('organizations')):
-    accounts = client.list_accounts(
-
-    )
+def list_accounts(region):
+    org_client = client("organizations", region_name=region)
+    accounts = org_client.list_accounts()
     logging.info(accounts)
     l_account = accounts["Accounts"]
     logging.info(len(accounts["Accounts"]))
     if len(accounts["Accounts"]) >= 20:
         logging.info("Paginating ...")
-        ad_accounts = list_accounts_pag(client=client, next_token=accounts["NextToken"])
+        ad_accounts = list_accounts_pag(region=region, next_token=accounts["NextToken"])
         for ad in ad_accounts:
             l_account.append(ad)
         logging.info(f"You Organizations have {len(l_account)} Accounts")
@@ -89,14 +143,15 @@ def list_accounts(client=boto3.client('organizations')):
     return l_account
 
 
-def index_accounts(list_account):
+def index_accounts(list_account, region):
     accounts = []
-    client = boto3.client('organizations')
+    org_client = client("organizations", region_name=region)
     for a in list_account:
-        response = client.list_parents(
+        response = org_client.list_parents(
             ChildId=a["Id"],
-
         )
 
-        accounts.append({"account": a["Id"], "name": a["Name"], "parents": response["Parents"]})
+        accounts.append(
+            {"account": a["Id"], "name": a["Name"], "parents": response["Parents"]}
+        )
     return accounts