pygeai 0.1.6__py3-none-any.whl → 0.6.0b15__py3-none-any.whl

pygeai/cli/commands/organization.py

@@ -1,31 +1,52 @@
-import sys
-
 from pygeai.cli.commands import Command, Option, ArgumentsEnum
 from pygeai.cli.commands.builders import build_help_text
+from pygeai.cli.commands.options import DETAIL_OPTION, PROJECT_NAME_OPTION, PROJECT_ID_OPTION, SUBSCRIPTION_TYPE_OPTION, \
+    USAGE_LIMIT_USAGE_UNIT_OPTION, USAGE_LIMIT_SOFT_LIMIT_OPTION, USAGE_LIMIT_HARD_LIMIT_OPTION, \
+    USAGE_LIMIT_RENEWAL_STATUS_OPTION, PROJECT_DESCRIPTION_OPTION
 from pygeai.cli.texts.help import ORGANIZATION_HELP_TEXT
 from pygeai.core.common.exceptions import MissingRequirementException
+from pygeai.core.plugins.clients import PluginClient
+from pygeai.core.utils.console import Console
 from pygeai.organization.clients import OrganizationClient
 
 
-def show_organization_help():
+def show_help():
     """
     Displays help text in stdout
     """
     help_text = build_help_text(organization_commands, ORGANIZATION_HELP_TEXT)
-    sys.stdout.write(help_text)
+    Console.write_stdout(help_text)
 
 
-def get_assistant_list(detail: str = "summary"):
-    client = OrganizationClient()
-    result = client.get_assistant_list(detail)
-    sys.stdout.write(f"Assistant list: \n{result}\n")
+def list_assistants(option_list: list):
+    organization_id = None
+    project_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "organization_id":
+            organization_id = option_arg
+        if option_flag.name == "project_id":
+            project_id = option_arg
+
+        if not organization_id and project_id:
+            raise MissingRequirementException("Organization ID and Project ID are required.")
+
+    client = PluginClient()
+    result = client.list_assistants(organization_id=organization_id, project_id=project_id)
+
+    Console.write_stdout(f"Assistant list: \n{result}")
 
 
 assistants_list_options = [
     Option(
-        "detail",
-        ["--detail", "-d"],
-        "Defines the level of detail required. The available options are summary (default) or full (optional).",
+        "organization_id",
+        ["--organization-id", "--oid"],
+        "UUID of the organization",
+        True
+    ),
+    Option(
+        "project_id",
+        ["--project-id", "--pid"],
+        "UUID of the project",
         True
     ),
 ]
@@ -42,22 +63,12 @@ def get_project_list(option_list: list):
 
     client = OrganizationClient()
     result = client.get_project_list(detail, name)
-    sys.stdout.write(f"Project list: \n{result}\n")
+    Console.write_stdout(f"Project list: \n{result}")
 
 
 project_list_options = [
-    Option(
-        "detail",
-        ["--detail", "-d"],
-        "Defines the level of detail required. The available options are summary (default) or full (optional).",
-        True
-    ),
-    Option(
-        "name",
-        ["--name", "-n"],
-        "Searches by project name (equals) (optional)",
-        True
-    ),
+    DETAIL_OPTION,
+    PROJECT_NAME_OPTION,
 ]
 
 
@@ -72,16 +83,11 @@ def get_project_detail(option_list: list):
 
     client = OrganizationClient()
     result = client.get_project_data(project_id=project_id)
-    sys.stdout.write(f"Project detail: \n{result}\n")
+    Console.write_stdout(f"Project detail: \n{result}")
 
 
 project_detail_options = [
-    Option(
-        "project_id",
-        ["--project-id", "--id"],
-        "GUID of the project (required)",
-        True
-    ),
+    PROJECT_ID_OPTION,
 ]
 
 
@@ -135,7 +141,7 @@ def create_project(option_list: list):
 
     client = OrganizationClient()
     result = client.create_project(name, email, description)
-    sys.stdout.write(f"New project: \n{result}\n")
+    Console.write_stdout(f"New project: \n{result}")
 
 
 create_project_options = [
@@ -157,36 +163,11 @@ create_project_options = [
         "Project administrator's email",
         True
     ),
-    Option(
-        "subscription_type",
-        ["--subscription-type"],
-        "string: Options: Freemium, Daily, Weekly, Monthly)",
-        True
-    ),
-    Option(
-        "usage_unit",
-        ["--usage-unit"],
-        "string: Options: Requests, Cost)",
-        True
-    ),
-    Option(
-        "soft_limit",
-        ["--soft-limit"],
-        "number: Soft limit for usage (lower threshold))",
-        True
-    ),
-    Option(
-        "hard_limit",
-        ["--hard-limit"],
-        "number: Hard limit for usage (upper threshold)). Must be greater or equal to --soft-limit.",
-        True
-    ),
-    Option(
-        "renewal_status",
-        ["--renewal-status"],
-        "string: Options: Renewable, NonRenewable). If --subscription-type is Freemium, this must be NonRenewable",
-        True
-    )
+    SUBSCRIPTION_TYPE_OPTION,
+    USAGE_LIMIT_USAGE_UNIT_OPTION,
+    USAGE_LIMIT_SOFT_LIMIT_OPTION,
+    USAGE_LIMIT_HARD_LIMIT_OPTION,
+    USAGE_LIMIT_RENEWAL_STATUS_OPTION
 ]
 
 
@@ -209,28 +190,13 @@ def update_project(option_list: list):
 
     client = OrganizationClient()
     result = client.update_project(project_id, name, description)
-    sys.stdout.write(f"Updated project: \n{result}\n")
+    Console.write_stdout(f"Updated project: \n{result}")
 
 
 update_project_options = [
-    Option(
-        "project_id",
-        ["--project-id", "--id"],
-        "GUID of the project (required)",
-        True
-    ),
-    Option(
-        "name",
-        ["--name", "-n"],
-        "Name of the new project",
-        True
-    ),
-    Option(
-        "description",
-        ["--description", "-d"],
-        "Description of the new project",
-        True
-    ),
+    PROJECT_ID_OPTION,
+    PROJECT_NAME_OPTION,
+    PROJECT_DESCRIPTION_OPTION,
 ]
 
 
@@ -245,16 +211,11 @@ def delete_project(option_list: list):
 
     client = OrganizationClient()
     result = client.delete_project(project_id)
-    sys.stdout.write(f"Deleted project: \n{result}\n")
+    Console.write_stdout(f"Deleted project: \n{result}")
 
 
 delete_project_options = [
-    Option(
-        "project_id",
-        ["--project-id", "--id"],
-        "GUID of the project (required)",
-        True
-    ),
+    PROJECT_ID_OPTION,
 ]
 
 
@@ -269,16 +230,11 @@ def get_project_tokens(option_list: list):
 
     client = OrganizationClient()
     result = client.get_project_tokens(project_id)
-    sys.stdout.write(f"Project tokens: \n{result}\n")
+    Console.write_stdout(f"Project tokens: \n{result}")
 
 
 get_project_tokens_options = [
-    Option(
-        "project_id",
-        ["--project-id", "--id"],
-        "GUID of the project (required)",
-        True
-    ),
+    PROJECT_ID_OPTION,
 ]
 
 
@@ -302,7 +258,7 @@ def export_request_data(option_list: list):
 
     client = OrganizationClient()
     result = client.export_request_data(assistant_name, status, skip, count)
-    sys.stdout.write(f"Request data: \n{result}\n")
+    Console.write_stdout(f"Request data: \n{result}")
 
 
 export_request_data_options = [
@@ -332,12 +288,437 @@ export_request_data_options = [
     )
 ]
 
+
+def get_memberships(option_list: list):
+    email = None
+    start_page = 1
+    page_size = 20
+    order_key = None
+    order_direction = "desc"
+    role_types = None
+
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "email":
+            email = option_arg
+        if option_flag.name == "start_page":
+            start_page = int(option_arg)
+        if option_flag.name == "page_size":
+            page_size = int(option_arg)
+        if option_flag.name == "order_key":
+            order_key = option_arg
+        if option_flag.name == "order_direction":
+            order_direction = option_arg
+        if option_flag.name == "role_types":
+            role_types = option_arg
+
+    client = OrganizationClient()
+    result = client.get_memberships(email, start_page, page_size, order_key, order_direction, role_types)
+    Console.write_stdout(f"Memberships: \n{result}")
+
+
+get_memberships_options = [
+    Option(
+        "email",
+        ["--email", "-e"],
+        "Email address of the user (optional, case-insensitive)",
+        True
+    ),
+    Option(
+        "start_page",
+        ["--start-page"],
+        "Page number for pagination (default: 1)",
+        True
+    ),
+    Option(
+        "page_size",
+        ["--page-size"],
+        "Number of items per page (default: 20)",
+        True
+    ),
+    Option(
+        "order_key",
+        ["--order-key"],
+        "Field for sorting (only 'organizationName' supported)",
+        True
+    ),
+    Option(
+        "order_direction",
+        ["--order-direction"],
+        "Sort direction: asc or desc (default: desc)",
+        True
+    ),
+    Option(
+        "role_types",
+        ["--role-types"],
+        "Comma-separated list: backend, frontend (optional, case-insensitive)",
+        True
+    ),
+]
+
+
+def get_project_memberships(option_list: list):
+    email = None
+    start_page = 1
+    page_size = 20
+    order_key = None
+    order_direction = "desc"
+    role_types = None
+
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "email":
+            email = option_arg
+        if option_flag.name == "start_page":
+            start_page = int(option_arg)
+        if option_flag.name == "page_size":
+            page_size = int(option_arg)
+        if option_flag.name == "order_key":
+            order_key = option_arg
+        if option_flag.name == "order_direction":
+            order_direction = option_arg
+        if option_flag.name == "role_types":
+            role_types = option_arg
+
+    client = OrganizationClient()
+    result = client.get_project_memberships(email, start_page, page_size, order_key, order_direction, role_types)
+    Console.write_stdout(f"Project memberships: \n{result}")
+
+
+get_project_memberships_options = [
+    Option(
+        "email",
+        ["--email", "-e"],
+        "Email address of the user (optional, case-insensitive)",
+        True
+    ),
+    Option(
+        "start_page",
+        ["--start-page"],
+        "Page number for pagination (default: 1)",
+        True
+    ),
+    Option(
+        "page_size",
+        ["--page-size"],
+        "Number of items per page (default: 20)",
+        True
+    ),
+    Option(
+        "order_key",
+        ["--order-key"],
+        "Field for sorting (only 'projectName' supported)",
+        True
+    ),
+    Option(
+        "order_direction",
+        ["--order-direction"],
+        "Sort direction: asc or desc (default: desc)",
+        True
+    ),
+    Option(
+        "role_types",
+        ["--role-types"],
+        "Comma-separated list: backend, frontend (optional, case-insensitive)",
+        True
+    ),
+]
+
+
+def get_project_roles(option_list: list):
+    project_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
+
+    if not project_id:
+        raise MissingRequirementException("Cannot retrieve project roles without project-id")
+
+    client = OrganizationClient()
+    result = client.get_project_roles(project_id)
+    Console.write_stdout(f"Project roles: \n{result}")
+
+
+get_project_roles_options = [
+    PROJECT_ID_OPTION,
+]
+
+
+def get_project_members(option_list: list):
+    project_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
+
+    if not project_id:
+        raise MissingRequirementException("Cannot retrieve project members without project-id")
+
+    client = OrganizationClient()
+    result = client.get_project_members(project_id)
+    Console.write_stdout(f"Project members: \n{result}")
+
+
+get_project_members_options = [
+    PROJECT_ID_OPTION,
+]
+
+
+def get_organization_members(option_list: list):
+    organization_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "organization_id":
+            organization_id = option_arg
+
+    if not organization_id:
+        raise MissingRequirementException("Cannot retrieve organization members without organization-id")
+
+    client = OrganizationClient()
+    result = client.get_organization_members(organization_id)
+    Console.write_stdout(f"Organization members: \n{result}")
+
+
+get_organization_members_options = [
+    Option(
+        "organization_id",
+        ["--organization-id", "--oid"],
+        "GUID of the organization (required)",
+        True
+    ),
+]
+
+
+def add_project_member(option_list: list):
+    project_id = None
+    user_email = None
+    roles = []
+    batch_file = None
+
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
+        if option_flag.name == "user_email":
+            user_email = option_arg
+        if option_flag.name == "roles":
+            roles = [r.strip() for r in option_arg.split(",")]
+        if option_flag.name == "batch_file":
+            batch_file = option_arg
+
+    client = OrganizationClient()
+
+    if batch_file:
+        add_project_member_in_batch(client, batch_file)
+    else:
+        if not (project_id and user_email and roles):
+            raise MissingRequirementException("Cannot add project member without project-id, user email, and roles")
+        result = client.add_project_member(project_id, user_email, roles)
+        Console.write_stdout(f"User invitation sent: \n{result}")
+
+
+def add_project_member_in_batch(client: OrganizationClient, batch_file: str):
+    import csv
+    import os
+
+    if not os.path.exists(batch_file):
+        raise MissingRequirementException(f"Batch file not found: {batch_file}")
+
+    successful = 0
+    failed = 0
+    errors = []
+
+    try:
+        with open(batch_file, 'r') as f:
+            csv_reader = csv.reader(f)
+            for line_num, row in enumerate(csv_reader, start=1):
+                if len(row) < 3:
+                    error_msg = f"Line {line_num}: Invalid format - expected at least 3 columns (project_id, email, role1, ...)"
+                    errors.append(error_msg)
+                    failed += 1
+                    continue
+
+                project_id = row[0].strip()
+                email = row[1].strip()
+                roles = [r.strip() for r in row[2:] if r.strip()]
+
+                if not (project_id and email and roles):
+                    error_msg = f"Line {line_num}: Missing required fields (project_id={project_id}, email={email}, roles={roles})"
+                    errors.append(error_msg)
+                    failed += 1
+                    continue
+
+                try:
+                    client.add_project_member(project_id, email, roles)
+                    successful += 1
+                except Exception as e:
+                    error_msg = f"Line {line_num}: Failed to add {email} to project {project_id}: {str(e)}"
+                    errors.append(error_msg)
+                    failed += 1
+
+        Console.write_stdout(f"Batch processing complete: {successful} successful, {failed} failed")
+        if errors:
+            Console.write_stdout("\nErrors:")
+            for error in errors:
+                Console.write_stdout(f"  - {error}")
+    except Exception as e:
+        raise MissingRequirementException(f"Failed to read batch file: {str(e)}")
+
+
+add_project_member_options = [
+    Option(
+        "project_id",
+        ["--project-id", "--pid"],
+        "GUID of the project (required unless --batch is used)",
+        True
+    ),
+    Option(
+        "user_email",
+        ["--email", "-e"],
+        "Email address of the user to invite (required unless --batch is used)",
+        True
+    ),
+    Option(
+        "roles",
+        ["--roles", "-r"],
+        "Comma-separated list of role names or GUIDs (e.g., 'Project member,Project administrator') (required unless --batch is used)",
+        True
+    ),
+    Option(
+        "batch_file",
+        ["--batch", "-b"],
+        "Path to CSV file with format: project_id,email,role1,role2,... (one invitation per line)",
+        True
+    ),
+]
+
+
+def create_organization(option_list: list):
+    name = None
+    email = None
+
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "name":
+            name = option_arg
+
+        if option_flag.name == "admin_email":
+            email = option_arg
+
+    if not (name and email):
+        raise MissingRequirementException("Cannot create organization without name and administrator's email")
+
+    client = OrganizationClient()
+    result = client.create_organization(name, email)
+    Console.write_stdout(f"New organization: \\n{result}")
+
+
+create_organization_options = [
+    Option(
+        "name",
+        ["--name", "-n"],
+        "Name of the new organization",
+        True
+    ),
+    Option(
+        "admin_email",
+        ["--email", "-e"],
+        "Organization administrator's email",
+        True
+    ),
+]
+
+
+def get_organization_list(option_list: list):
+    start_page = None
+    page_size = None
+    order_key = None
+    order_direction = "desc"
+    filter_key = None
+    filter_value = None
+
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "start_page":
+            start_page = int(option_arg)
+        if option_flag.name == "page_size":
+            page_size = int(option_arg)
+        if option_flag.name == "order_key":
+            order_key = option_arg
+        if option_flag.name == "order_direction":
+            order_direction = option_arg
+        if option_flag.name == "filter_key":
+            filter_key = option_arg
+        if option_flag.name == "filter_value":
+            filter_value = option_arg
+
+    client = OrganizationClient()
+    result = client.get_organization_list(start_page, page_size, order_key, order_direction, filter_key, filter_value)
+    Console.write_stdout(f"Organization list: \\n{result}")
+
+
+get_organization_list_options = [
+    Option(
+        "start_page",
+        ["--start-page"],
+        "Page number for pagination",
+        True
+    ),
+    Option(
+        "page_size",
+        ["--page-size"],
+        "Number of items per page",
+        True
+    ),
+    Option(
+        "order_key",
+        ["--order-key"],
+        "Field for sorting (only 'name' supported)",
+        True
+    ),
+    Option(
+        "order_direction",
+        ["--order-direction"],
+        "Sort direction: asc or desc (default: desc)",
+        True
+    ),
+    Option(
+        "filter_key",
+        ["--filter-key"],
+        "Field for filtering (only 'name' supported)",
+        True
+    ),
+    Option(
+        "filter_value",
+        ["--filter-value"],
+        "Value to filter by",
+        True
+    ),
+]
+
+
+def delete_organization(option_list: list):
+    organization_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "organization_id":
+            organization_id = option_arg
+
+    if not organization_id:
+        raise MissingRequirementException("Cannot delete organization without organization-id")
+
+    client = OrganizationClient()
+    result = client.delete_organization(organization_id)
+    Console.write_stdout(f"Deleted organization: \\n{result}")
+
+
+delete_organization_options = [
+    Option(
+        "organization_id",
+        ["--organization-id", "--oid"],
+        "GUID of the organization (required)",
+        True
+    ),
+]
+
 organization_commands = [
     Command(
         "help",
         ["help", "h"],
         "Display help text",
-        show_organization_help,
+        show_help,
         ArgumentsEnum.NOT_AVAILABLE,
         [],
         []
@@ -346,7 +727,7 @@ organization_commands = [
         "assistants_list",
         ["list-assistants"],
         "List assistant information",
-        get_assistant_list,
+        list_assistants,
         ArgumentsEnum.OPTIONAL,
         [],
         assistants_list_options
@@ -414,4 +795,85 @@ organization_commands = [
         [],
         export_request_data_options
     ),
+    Command(
+        "get_memberships",
+        ["get-memberships"],
+        "Get user memberships across organizations and projects",
+        get_memberships,
+        ArgumentsEnum.OPTIONAL,
+        [],
+        get_memberships_options
+    ),
+    Command(
+        "get_project_memberships",
+        ["get-project-memberships"],
+        "Get user project memberships within an organization",
+        get_project_memberships,
+        ArgumentsEnum.OPTIONAL,
+        [],
+        get_project_memberships_options
+    ),
+    Command(
+        "get_project_roles",
+        ["get-project-roles"],
+        "Get all roles supported by a project",
+        get_project_roles,
+        ArgumentsEnum.REQUIRED,
+        [],
+        get_project_roles_options
+    ),
+    Command(
+        "get_project_members",
+        ["get-project-members"],
+        "Get all members and their roles for a project",
+        get_project_members,
+        ArgumentsEnum.REQUIRED,
+        [],
+        get_project_members_options
+    ),
+    Command(
+        "get_organization_members",
+        ["get-organization-members"],
+        "Get all members and their roles for an organization",
+        get_organization_members,
+        ArgumentsEnum.REQUIRED,
+        [],
+        get_organization_members_options
+    ),
+    Command(
+        "add_project_member",
+        ["add-project-member", "apm"],
+        "Add a user to a project by sending an invitation email",
+        add_project_member,
+        ArgumentsEnum.REQUIRED,
+        [],
+        add_project_member_options
+    ),
+    Command(
+        "create_organization",
+        ["create-organization"],
+        "Create new organization",
+        create_organization,
+        ArgumentsEnum.REQUIRED,
+        [],
+        create_organization_options
+    ),
+    Command(
+        "organization_list",
+        ["list-organizations"],
+        "List organization information",
+        get_organization_list,
+        ArgumentsEnum.OPTIONAL,
+        [],
+        get_organization_list_options
+    ),
+    Command(
+        "delete_organization",
+        ["delete-organization"],
+        "Delete existing organization",
+        delete_organization,
+        ArgumentsEnum.REQUIRED,
+        [],
+        delete_organization_options
+    ),
 ]