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

src/reverse_diagrams.py

@@ -1,169 +1,325 @@
-import os
-
-import boto3
-import argparse
-import logging
-import emoji
-from colorama import Fore
-from datetime import datetime
-from .aws.describe_organization import describe_organization, \
-    list_accounts, index_accounts, list_roots, list_organizational_units, index_ous
-from .aws.describe_identity_store import order_accounts_assignments_list, extend_account_assignments, \
-    list_groups, get_members, list_users, complete_group_members, add_users_and_groups_assign, l_groups_to_d_groups
-from .aws.describe_sso import list_instances, extends_permissions_set, list_permissions_set
-from .dgms.graph_mapper import create_mapper, create_sso_mapper_complete, create_sso_mapper, create_file
-from .dgms.graph_template import graph_template, graph_template_sso, graph_template_sso_complete
-from .banner.banner import get_version
-
-__version__ = "0.2.5"
-
-
-def main() -> int:
-    """
-    Crete architecture diagram from your current state
-    :return:
-    """
-
-    # Initialize parser
-    parser = argparse.ArgumentParser()
-    parser.add_argument("-c", "--cloud",
-                        help="Cloud Provider, aws, gcp, azure", default="aws")
-    parser.add_argument("-p", "--profile",
-                        help="AWS cli profile for Access Analyzer Api", default=None)
-    parser.add_argument("-od", "--output_dir_path",
-                        help="Name of folder to save the diagrams python code files", default=None)
-    parser.add_argument("-r", "--region",
-                        help="AWS cli profile for Access Analyzer Api", default="us-east-2")
-    parser.add_argument("-o", "--graph_organization",
-                        help="Set if you want to create graph for your organization", action='store_true')
-    parser.add_argument("-i", "--graph_identity",
-                        help="Set if you want to create graph for your IAM Center", action='store_true')
-
-    parser.add_argument("-v", "--version",
-                        help="Show version", action='store_true')
-    parser.add_argument("-d", "--debug",
-                        help="Debug Mode", action='store_true')
-
-    # Read arguments from command line
-    args = parser.parse_args()
-    logging.info(f"The arguments are {args}")
-    if args.debug:
-        logging.basicConfig(level=logging.DEBUG)
-
-    if args.output_dir_path:
-        diagrams_path = args.output_dir_path
-    else:
-        diagrams_path = "."
-
-    if args.cloud == "aws":
-        if args.profile:
-            profile = args.profile
-            if profile is not None:
-                boto3.setup_default_session(profile_name=profile)
-
-            logging.info(f"Profile is: {profile}")
-        if args.region:
-            region = args.region
-
-        if args.graph_organization:
-            create_file(template_content=graph_template, file_name="graph_org.py", directory_path=diagrams_path)
-
-            client_org = boto3.client('organizations')
-            organization = describe_organization(client_org)
-            print(Fore.BLUE + emoji.emojize(":sparkle: Getting Organization Info" + Fore.RESET))
-            logging.debug(organization)
-            logging.debug("The Roots Info")
-            roots = list_roots(client=client_org)
-            logging.debug(roots)
-            print(Fore.BLUE + emoji.emojize(":sparkle: The Organizational Units list " + Fore.RESET))
-            logging.debug("The Organizational Units list ")
-            ous = list_organizational_units(parent_id=roots[0]["Id"], client=client_org)
-            logging.debug(ous)
-            logging.debug("The Organizational Units list with parents info")
-            i_ous = index_ous(ous, client=client_org)
-            logging.debug(i_ous)
-            print(Fore.BLUE + emoji.emojize(":sparkle: Getting the Account list info" + Fore.RESET))
-            l_accounts = list_accounts(client_org)
-            logging.debug(l_accounts)
-            logging.debug("The Account list with parents info")
-            print(Fore.YELLOW + emoji.emojize(
-                f":information:  There are {len(l_accounts)} Accounts in your organization" + Fore.RESET))
-            i_accounts = index_accounts(l_accounts)
-            logging.debug(i_accounts)
-
-            create_mapper(template_file="graph_org.py", org=organization, root_id=roots[0]["Id"], list_ous=ous,
-                          list_accounts=i_accounts)
-
-            print(
-                Fore.YELLOW + emoji.emojize(f":sparkles:   Run -> python3 {diagrams_path}/graph_org.py " + Fore.RESET))
-
-        if args.graph_identity:
-            create_file(template_content=graph_template_sso, file_name="graph_sso.py", directory_path=diagrams_path)
-            create_file(template_content=graph_template_sso_complete, file_name="graph_sso_complete.py",
-                        directory_path=diagrams_path)
-
-            client_identity = boto3.client('identitystore', region_name=region)
-            client_sso = boto3.client('sso-admin', region_name=region)
-
-            store_instances = list_instances(client=client_sso)
-            print(Fore.BLUE + emoji.emojize(":sparkle: Getting Identity store instance info" + Fore.RESET))
-            logging.debug(store_instances)
-            store_id = store_instances[0]["IdentityStoreId"]
-            store_arn = store_instances[0]["InstanceArn"]
-            print(Fore.BLUE + emoji.emojize(":sparkle: List groups" + Fore.RESET))
-            l_groups = list_groups(store_id, client=client_identity)
-            logging.debug(l_groups)
-            print(Fore.YELLOW + emoji.emojize(
-                f":information:  There are {len(l_groups)} Groups in your Identity Store" + Fore.RESET))
-
-            print(Fore.BLUE + emoji.emojize(":sparkle: Get groups and Users info" + Fore.RESET))
-
-            m_groups = get_members(store_id, l_groups, client=client_identity)
-
-            logging.debug(m_groups)
-
-            logging.debug("Extend Group Members")
-            l_users = list_users(store_id, client=client_identity)
-            logging.debug(l_users)
-            c_users_and_groups = complete_group_members(m_groups, l_users)
-            d_groups = l_groups_to_d_groups(l_groups=c_users_and_groups)
-
-            logging.debug(c_users_and_groups)
-            # Get Account assingments
-            permissions_set = list_permissions_set(instance_arn=store_arn, client=client_sso)
-            l_permissions_set_arn_name = extends_permissions_set(permissions_sets=permissions_set, store_arn=store_arn,
-                                                                 client_sso=client_sso)
-
-            client_org = boto3.client('organizations')
-            l_accounts = list_accounts(client_org)
-            account_assignments = extend_account_assignments(accounts_list=l_accounts,
-                                                             permissions_sets=l_permissions_set_arn_name,
-                                                             client_sso=client_sso,
-                                                             store_arn=store_arn)
-
-            account_assignments = add_users_and_groups_assign(account_assignments_list=account_assignments,
-                                                              user_and_group_list=c_users_and_groups,
-                                                              user_list=l_users,
-                                                              list_permissions_set_arn_name=l_permissions_set_arn_name)
-            print(Fore.BLUE + emoji.emojize(":sparkle: Getting account assignments, users and groups" + Fore.RESET))
-            f_accounts = order_accounts_assignments_list(accounts_dict=l_accounts,
-                                                         account_assignments=account_assignments)
-            f_path= os.path.join(diagrams_path, "graph_sso_complete.py")
-            create_sso_mapper_complete(template_file=f_path,
-                                       acc_assignments=f_accounts,
-                                       d_groups=d_groups)
-
-            f_path = os.path.join(diagrams_path, "graph_sso.py")
-            create_sso_mapper(template_file=f_path, group_and_members=c_users_and_groups)
-            print(Fore.YELLOW + emoji.emojize(
-                f":sparkles:  Run -> python3 {diagrams_path}/graph_sso_complete.py " + Fore.RESET))
-    else:
-        print(Fore.RED + emoji.emojize(":warning: " + f"The cloud provider {args.cloud} is no available" + Fore.RESET))
-    if args.version:
-        get_version(version=__version__)
-
-    return 0
-
-
-if __name__ == '__main__':
-    main()
+import argparse
+import logging
+import os
+
+import argcomplete
+import emoji
+from boto3 import setup_default_session
+from colorama import Fore
+
+from .aws.describe_identity_store import (
+    add_users_and_groups_assign,
+    complete_group_members,
+    extend_account_assignments,
+    get_members,
+    l_groups_to_d_groups,
+    list_groups,
+    list_users,
+    order_accounts_assignments_list,
+)
+from .aws.describe_organization import (
+    describe_organization,
+    index_accounts,
+    index_ous,
+    list_accounts,
+    list_organizational_units,
+    list_roots,
+)
+from .aws.describe_sso import (
+    extends_permissions_set,
+    list_instances,
+    list_permissions_set,
+)
+from .banner.banner import get_version
+from .dgms.graph_mapper import (
+    create_file,
+    create_mapper,
+    create_sso_mapper,
+    create_sso_mapper_complete,
+)
+from .dgms.graph_template import (
+    graph_template,
+    graph_template_sso,
+    graph_template_sso_complete,
+)
+from .reports.save_results import save_results
+from .version import __version__
+
+
+def graph_organizations(diagrams_path, region, auto):
+    """
+    Create organizations Graph.
+
+    :param diagrams_path:
+    :param region:
+    :return:
+    """
+    template_file = "graph_org.py"
+    code_path = f"{diagrams_path}/code"
+    json_path = f"{diagrams_path}/json"
+    create_file(
+        template_content=graph_template,
+        file_name=template_file,
+        directory_path=code_path,
+    )
+
+    organization = describe_organization(region=region)
+    print(Fore.BLUE + emoji.emojize(":sparkle: Getting Organization Info" + Fore.RESET))
+    logging.debug(organization)
+    logging.debug("The Roots Info")
+    roots = list_roots(region=region)
+    logging.debug(roots)
+
+    print(
+        Fore.BLUE
+        + emoji.emojize(":sparkle: Listing Organizational Units " + Fore.RESET)
+    )
+    logging.debug("The Organizational Units list ")
+    ous = list_organizational_units(parent_id=roots[0]["Id"], region=region)
+    logging.debug(ous)
+    logging.debug("The Organizational Units list with parents info")
+    i_ous = index_ous(ous, region=region)
+    logging.debug(i_ous)
+
+    print(
+        Fore.BLUE
+        + emoji.emojize(":sparkle: Getting the Account list info" + Fore.RESET)
+    )
+    l_accounts = list_accounts(region=region)
+    logging.debug(l_accounts)
+    logging.debug("The Account list with parents info")
+
+    print(
+        Fore.YELLOW
+        + emoji.emojize(
+            f":information:  There are {len(l_accounts)} Accounts in your organization"
+            + Fore.RESET
+        )
+    )
+    i_accounts = index_accounts(l_accounts, region=region)
+    logging.debug(i_accounts)
+
+    file_name = "organizations.json"
+    save_results(results=i_accounts, filename=file_name, directory_path=json_path)
+
+    create_mapper(
+        template_file=f"{code_path}/{template_file}",
+        org=organization,
+        root_id=roots[0]["Id"],
+        list_ous=ous,
+        list_accounts=i_accounts,
+    )
+    if auto:
+        print(f"{Fore.GREEN} ❇️ Creating diagrams in {code_path}")
+        command = os.system(f"cd {code_path} && python3 {template_file}")
+        if command != 0:
+            print(Fore.RED + "❌ Error creating diagrams")
+            print(command)
+    else:
+        print(
+            Fore.YELLOW
+            + emoji.emojize(
+                f":sparkles: Run -> python3 {code_path}/graph_org.py " + Fore.RESET
+            )
+        )
+
+
+def graph_identity_center(diagrams_path, region, auto):
+    """
+    Create Identity center diagram.
+
+    :param diagrams_path:
+    :param region:
+    :return:
+    """
+
+    template_file = "graph_sso.py"
+    template_file_complete = "graph_sso_complete.py"
+
+    code_path = f"{diagrams_path}/code"
+    json_path = f"{diagrams_path}/json"
+
+    create_file(
+        template_content=graph_template_sso,
+        file_name=template_file,
+        directory_path=code_path,
+    )
+    create_file(
+        template_content=graph_template_sso_complete,
+        file_name=template_file_complete,
+        directory_path=code_path,
+    )
+
+    store_instances = list_instances(region=region)
+    print(
+        Fore.BLUE
+        + emoji.emojize(":sparkle: Getting Identity store instance info" + Fore.RESET)
+    )
+    logging.debug(store_instances)
+    store_id = store_instances[0]["IdentityStoreId"]
+    store_arn = store_instances[0]["InstanceArn"]
+
+    print(Fore.BLUE + emoji.emojize(":sparkle: List groups" + Fore.RESET))
+    l_groups = list_groups(store_id, region=region)
+    logging.debug(l_groups)
+    print(
+        Fore.YELLOW
+        + emoji.emojize(
+            f":information:  There are {len(l_groups)} Groups in your Identity Store"
+            + Fore.RESET
+        )
+    )
+
+    print(Fore.BLUE + emoji.emojize(":sparkle: Get groups and Users info" + Fore.RESET))
+    m_groups = get_members(store_id, l_groups, region=region)
+    logging.debug(m_groups)
+    logging.debug("Extend Group Members")
+    l_users = list_users(store_id, region=region)
+    logging.debug(l_users)
+    c_users_and_groups = complete_group_members(m_groups, l_users)
+    d_groups = l_groups_to_d_groups(l_groups=c_users_and_groups)
+
+    logging.debug(c_users_and_groups)
+    # Get Account assignments
+    permissions_set = list_permissions_set(instance_arn=store_arn, region=region)
+    l_permissions_set_arn_name = extends_permissions_set(
+        permissions_sets=permissions_set, store_arn=store_arn, region=region
+    )
+
+    l_accounts = list_accounts(region=region)
+    account_assignments = extend_account_assignments(
+        accounts_list=l_accounts,
+        permissions_sets=l_permissions_set_arn_name,
+        region=region,
+        store_arn=store_arn,
+    )
+
+    account_assignments = add_users_and_groups_assign(
+        account_assignments_list=account_assignments,
+        user_and_group_list=c_users_and_groups,
+        user_list=l_users,
+        list_permissions_set_arn_name=l_permissions_set_arn_name,
+    )
+
+    print(
+        Fore.BLUE
+        + emoji.emojize(
+            ":sparkle: Getting account assignments, users and groups" + Fore.RESET
+        )
+    )
+    f_accounts = order_accounts_assignments_list(
+        accounts_dict=l_accounts, account_assignments=account_assignments
+    )
+    f_path = os.path.join(code_path, template_file_complete)
+    create_sso_mapper_complete(
+        template_file=f_path, acc_assignments=f_accounts, d_groups=d_groups
+    )
+
+    save_results(
+        results=f_accounts,
+        filename="account_assignments.json",
+        directory_path=json_path,
+    )
+    save_results(results=d_groups, filename="groups.json", directory_path=json_path)
+
+    f_path = os.path.join(code_path, template_file)
+
+    create_sso_mapper(template_file=f_path, group_and_members=c_users_and_groups)
+
+    if auto:
+        print(f"{Fore.GREEN} ❇️ Creating diagrams in {code_path}")
+        command_1 = os.system(f"cd {code_path} && python3 {template_file}")
+        command = os.system(f"cd {code_path} && python3 {template_file_complete}")
+
+        if command != 0 or command_1 != 0:
+            print(Fore.RED + "❌ Error creating diagrams")
+            print(command)
+    else:
+        print(
+            Fore.YELLOW
+            + emoji.emojize(
+                f":sparkles:  Run -> python3 {code_path}/{template_file_complete}"
+                f" or python3 {code_path}/{template_file}" + Fore.RESET
+            )
+        )
+
+
+def main() -> int:
+    """
+    Crete architecture diagram from your current state.
+
+    :return:
+    """
+
+    # Initialize parser
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument(
+        "-p", "--profile", help="AWS cli profile for AWS Apis", default=None
+    )
+    parser.add_argument(
+        "-od",
+        "--output_dir_path",
+        help="Name of folder to save the diagrams python code files",
+        default="diagrams",
+    )
+    parser.add_argument("-r", "--region", help="AWS region", default="us-east-2")
+    parser.add_argument(
+        "-o",
+        "--graph_organization",
+        help="Set if you want to create graph for your organization",
+        action="store_true",
+    )
+    parser.add_argument(
+        "-i",
+        "--graph_identity",
+        help="Set if you want to create graph for your IAM Center",
+        action="store_true",
+    )
+    parser.add_argument(
+        "-a",
+        "--auto_create",
+        help="Create Automatically diagrams",
+        action="store_true",
+        default=True,
+    )
+    parser.add_argument("-v", "--version", help="Show version", action="store_true")
+    parser.add_argument("-d", "--debug", help="Debug Mode", action="store_true")
+
+    # Read arguments from command line
+    args = parser.parse_args()
+    # Add autocomplete
+    argcomplete.autocomplete(parser)
+    logging.info(f"The arguments are {args}")
+    if args.debug:
+        logging.basicConfig(level=logging.DEBUG)
+
+    diagrams_path = args.output_dir_path
+
+    region = args.region
+
+    if args.profile:
+        profile = args.profile
+        if profile is not None:
+            setup_default_session(profile_name=profile)
+
+        logging.info(f"Profile is: {profile}")
+
+    if args.graph_organization:
+        graph_organizations(
+            diagrams_path=diagrams_path, region=region, auto=args.auto_create
+        )
+
+    if args.graph_identity:
+        graph_identity_center(
+            diagrams_path=diagrams_path, region=region, auto=args.auto_create
+        )
+
+    if args.version:
+        get_version(version=__version__)
+
+    return 0
+
+
+if __name__ == "__main__":
+    main()

src/version.py

@@ -0,0 +1,2 @@
+"""Set version file."""
+__version__ = "1.0.0"

docs/graph_org.py

@@ -1,8 +0,0 @@
-
-from diagrams import Diagram, Cluster
-
-from diagrams.aws.management import Organizations, OrganizationsAccount, OrganizationsOrganizationalUnit
-
-with Diagram("Organizations-State", show=False, direction="TB"):
-    ou = OrganizationsOrganizationalUnit("OU")
-    oa = OrganizationsAccount("Account")

docs/graph_sso.py

@@ -1,39 +0,0 @@
-
-from diagrams import Diagram, Cluster
-
-from diagrams.aws.management import Organizations, OrganizationsAccount, OrganizationsOrganizationalUnit
-from diagrams.aws.general import Users, User
-
-with Diagram("SSO-State", show=False, direction="TB"):
-    gg = Users("Group")
-    uu= User("User")
-
-    with Cluster('Groups'):
-
-        with Cluster("AWSAccountFactory"):
-
-                gg_0= [User("velez94@protonma\nil.com"),]
-
-        gg_1= Users("AWSAuditAccountA\ndmins")
-
-        gg_2= Users("AWSLogArchiveAdm\nins")
-
-        with Cluster("AWSControlTowerAdmins"):
-
-                gg_3= [User("velez94@protonma\nil.com"),]
-
-        gg_4= Users("AWSLogArchiveVie\nwers")
-
-        gg_5= Users("AWSSecurityAudit\nors")
-
-        gg_6= Users("AWSSecurityAudit\nPowerUsers")
-
-        with Cluster("DevSecOps_Admins"):
-
-                gg_7= [User("DevSecOpsAdm"),]
-
-        gg_8= Users("AWSServiceCatalo\ngAdmins")
-
-        with Cluster("SecOps_Adms"):
-
-                gg_9= [User("w.alejovl+secops\n-labs@gmail.com"),]