agilicus 1.251.4__py3-none-any.whl → 1.253.0__py3-none-any.whl

agilicus/__init__.py

@@ -19,68 +19,9 @@ from .agilicus_api import exceptions  # noqa
 from . import patches  # noqa
 from .pagination.pagination import get_many_entries
 
-ApiClient = patches.patched_api_client()
-
-
-def find_guid(obj: dict):
-    guid = obj.get("id", None)
-    if guid:
-        return guid
-    md = obj.get("metadata", {})
-    guid = md.get("id", None)
-    if guid:
-        return guid
-    raise Exception(f"GUID cannot be found in obj {obj}")
+from .create import create_or_update, add_list_resources, AddInfo, AddResult, find_guid
 
-
-def create_or_update(
-    obj,
-    create_method: Callable,
-    update_method: Callable = None,
-    to_dict=True,
-    guid_finder=find_guid,
-) -> [dict, str]:
-    """A helper method that handles duplicate (409)
-    creation of objects. On 409, if the update_method is provided,
-    it will apply a PUT on the resource to update it with
-    the new requested data. The guid is searched for in the object
-    returned from the 409, and this is then provided
-    to the update method as an argument, along with the original
-    object that should be applied.
-        param: obj: the object to be created or updated
-        param: create_method(obj, ...)
-        param: update_method(guid, obj, ...)
-        returns: tuple, the object, with the status code.
-
-        Note the status code could be:
-           409: a duplicate, no update was performed
-           201: a create was succesfully made
-           200: a duplicate occured, and the object was updated accordingly
-    """
-    result = None
-    try:
-        result = create_method(obj)
-        if to_dict:
-            return result.to_dict(), 201
-        else:
-            return result, 201
-    except ApiException as exc:
-        if exc.status == 409:
-            body = exc.body
-            if not body:
-                raise
-            result = json.loads(body)
-            if update_method:
-                guid = guid_finder(result)
-                if to_dict:
-                    return update_method(guid, obj).to_dict(), 200
-                else:
-                    return update_method(guid, obj), 200
-            else:
-                return result, 409
-        else:
-            raise
-    return result
+ApiClient = patches.patched_api_client()
 
 
 def GetClient(

agilicus/agilicus_api/api_client.py

@@ -77,7 +77,7 @@ class ApiClient(object):
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/1.251.4/python'
+        self.user_agent = 'OpenAPI-Generator/1.253.0/python'
 
     def __enter__(self):
         return self

agilicus/agilicus_api/configuration.py

@@ -387,7 +387,7 @@ class Configuration(object):
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: 2024.04.11\n"\
-               "SDK Package Version: 1.251.4".\
+               "SDK Package Version: 1.253.0".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self):

agilicus/agilicus_api_README.md

@@ -4,7 +4,7 @@ Agilicus is API-first. Modern software is controlled by other software, is open,
 The `agilicus_api` package is automatically generated by the [OpenAPI Generator](https://openapi-generator.tech) project:
 
 - API version: 2024.04.11
-- Package version: 1.251.4
+- Package version: 1.253.0
 - Build package: org.openapitools.codegen.languages.PythonClientCodegen
 For more information, please visit [https://www.agilicus.com/api](https://www.agilicus.com/api)
 

agilicus/create.py

@@ -0,0 +1,108 @@
+import json
+
+from dataclasses import dataclass
+from typing import Callable, Optional, Any, Union
+from agilicus_api import ApiException
+
+
+def find_guid(obj: dict):
+    guid = obj.get("id", None)
+    if guid:
+        return guid
+    md = obj.get("metadata", {})
+    guid = md.get("id", None)
+    if guid:
+        return guid
+    raise Exception(f"GUID cannot be found in obj {obj}")
+
+
+def create_or_update(
+    obj,
+    create_method: Callable,
+    update_method: Optional[Callable] = None,
+    to_dict=True,
+    guid_finder=find_guid,
+) -> tuple[Union[dict, object], int]:
+    """A helper method that handles duplicate (409)
+    creation of objects. On 409, if the update_method is provided,
+    it will apply a PUT on the resource to update it with
+    the new requested data. The guid is searched for in the object
+    returned from the 409, and this is then provided
+    to the update method as an argument, along with the original
+    object that should be applied.
+        param: obj: the object to be created or updated
+        param: create_method(obj, ...)
+        param: update_method(guid, obj, ...)
+        returns: tuple, the object, with the status code.
+
+        Note the status code could be:
+           409: a duplicate, no update was performed
+           201: a create was succesfully made
+           200: a duplicate occured, and the object was updated accordingly
+    """
+    result = None
+    try:
+        result = create_method(obj)
+        if to_dict:
+            return result.to_dict(), 201
+        else:
+            return result, 201
+    except ApiException as exc:
+        if exc.status == 409:
+            body = exc.body
+            if not body:
+                raise
+            result = json.loads(body)
+            if update_method:
+                guid = guid_finder(result)
+                if to_dict:
+                    return update_method(guid, obj).to_dict(), 200
+                else:
+                    return update_method(guid, obj), 200
+            else:
+                return result, 409
+        else:
+            raise
+    return result
+
+
+def add_list_resources(objs, add_info):
+    results: list[AddResult] = []
+    for obj in objs:
+        result, status = create_or_update(
+            obj,
+            add_info.create_fn,
+            add_info.replace_fn,
+            to_dict=False,
+            guid_finder=add_info.guid_finder,
+        )
+        results.append(
+            AddResult(
+                obj=result,
+                result_name=add_info.name_getter(result),
+                result_id=add_info.guid_finder(result),
+                created=status != 409,
+            )
+        )
+
+    return results
+
+
+class AddInfo:
+    def __init__(self):
+        super().__init__()
+        self.name_getter = name_in_spec
+        self.guid_finder = find_guid
+
+
+@dataclass
+class AddResult:
+    obj: Any
+    result_name: str
+    created: bool
+    result_id: Optional[str] = None
+    exc: Optional[str] = None
+
+
+def name_in_spec(obj):
+    return obj.spec.name

agilicus/policy/policies.py

@@ -11,6 +11,7 @@ from agilicus import (
     PolicyTemplateInstance,
     PolicyTemplateInstanceSpec,
     MFAPolicyTemplate,
+    SourceInfoPolicyTemplate,
 )
 
 from ..output.table import (
@@ -132,3 +133,54 @@ def format_policy_templates(ctx, templates):
     ]
 
     return format_table(ctx, templates, columns)
+
+
+def set_source_info_policy(
+    ctx,
+    name,
+    action,
+    source_subnet,
+    iso_country_code,
+    invert,
+    log_message=None,
+    label=None,
+    description=None,
+    **kwargs,
+):
+    org_id = get_org_from_input_or_ctx(ctx, **kwargs)
+
+    token = context.get_token(ctx)
+    apiclient = context.get_apiclient(ctx, token)
+
+    tmpl = SourceInfoPolicyTemplate(
+        action=action,
+        source_subnets=list(source_subnet or []),
+        iso_country_codes=list(iso_country_code or []),
+        invert=invert,
+        labels=[LabelName(la) for la in (label or [])],
+        template_type="source_info",
+    )
+
+    if log_message:
+        tmpl.log_message = log_message
+
+    spec = PolicyTemplateInstanceSpec(
+        org_id=org_id,
+        name=name,
+        template=tmpl,
+    )
+
+    if description is not None:
+        spec.description = description
+
+    tmpl = PolicyTemplateInstance(spec=spec)
+    templates_api = apiclient.policy_templates_api
+    resp, _ = create_or_update(
+        tmpl,
+        lambda obj: templates_api.create_policy_template_instance(obj),
+        lambda guid, obj: templates_api.replace_policy_template_instance(
+            guid, policy_template_instance=obj
+        ),
+        to_dict=False,
+    )
+    return resp

agilicus/policy/policy_main.py

@@ -15,6 +15,21 @@ def cli_command_set_multifactor_policy(ctx, **kwargs):
     output_entry(ctx, policies.set_multifactor_policy(ctx, **kwargs).to_dict())
 
 
+@click.command(name="set-source-info-policy")
+@click.option("--name", required=True)
+@click.option("--action", type=click.Choice(["allow", "deny"]), required=True)
+@click.option("--label", multiple=True, type=str)
+@click.option("--description", default=None)
+@click.option("--org-id", default=None)
+@click.option("--source-subnet", multiple=True, type=str)
+@click.option("--iso-country-code", multiple=True, type=str)
+@click.option("--log-message", default=None)
+@click.option("--invert", type=bool, default=False)
+@click.pass_context
+def cli_command_set_source_info_policy(ctx, **kwargs):
+    output_entry(ctx, policies.set_source_info_policy(ctx, **kwargs).to_dict())
+
+
 @click.command(name="list-multifactor-policies")
 @click.option("--org-id", default=None)
 @click.pass_context

{agilicus-1.251.4.dist-info → agilicus-1.253.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agilicus
-Version: 1.251.4
+Version: 1.253.0
 Summary: Agilicus SDK
 Home-page: https://www.agilicus.com/
 License: MIT

{agilicus-1.251.4.dist-info → agilicus-1.253.0.dist-info}/RECORD

@@ -2,7 +2,7 @@ agilicus/.gitignore,sha256=TyHq6BCuVrFiqgUb1QAayLxWcBseKwEOoKJTQb_-iW8,5
 agilicus/.openapi-generator/FILES,sha256=Z-r9T67MUUFuhU8dAaieYz8TdhwDztVdqJEQcs4LDB8,110119
 agilicus/.openapi-generator/VERSION,sha256=LXsIFraL1muXBEww1sZWaewc9ji87Ih3gIa04Z37RYM,14
 agilicus/.openapi-generator-ignore,sha256=pu2PTide7pJtJ-DFLzDy0cTYQJRlrB-8RRH3zGLeUds,1040
-agilicus/__init__.py,sha256=EiVvVQh2UNOkISQKIO2K6s_OXNB0tPFUnpIGfQa5bJQ,5037
+agilicus/__init__.py,sha256=AqC7DwkeKW_DniiaZfh5nbDeDQ4E_lW3frheyfmH8Mc,3207
 agilicus/access.py,sha256=UEHHhE3cCaCjxXQDjhKxQAoUEMWandygN0d-yEIIf8A,5457
 agilicus/admin.py,sha256=oQNL_k0dyI_y4oxWtmHsEnavpesiLP0EL_cKC4ABN_4,9922
 agilicus/agilicus_api/__init__.py,sha256=vBy427pkTqJ603GbAcD8l5EB8Er_jQd3XIRSEU_eiaA,62439
@@ -71,9 +71,9 @@ agilicus/agilicus_api/api/users_api.py,sha256=UVNZwoYz6f400eh-3ePelA0ppFpflrzTxz
 agilicus/agilicus_api/api/users_api_mock.py,sha256=wA_xiqL3Pz3KjljKlsmf5NveLZS1FpbaKJHBp7QvarY,15411
 agilicus/agilicus_api/api/whoami_api.py,sha256=8CgAP5sfBwXuyfcyJRx7_0k7maRvzmos7hqqVWnhLio,7941
 agilicus/agilicus_api/api/whoami_api_mock.py,sha256=rlvZoWnMCqORMZBg7SOv6d3xp52kELdh6wXcCaIZ93w,346
-agilicus/agilicus_api/api_client.py,sha256=WHGVlVPU5h2Z7H3VIP9E1-PRbcOeVWGqcrgJbrDAmNQ,38845
+agilicus/agilicus_api/api_client.py,sha256=KU-40S7-VmA6-MfxIJ20Qpe8xB4jELMYLKDtiy0Wtrw,38845
 agilicus/agilicus_api/apis/__init__.py,sha256=aJZD7x-umdSni6ZBr4XxzpH8pwtU9hA5LlCDxcqa1Q8,2224
-agilicus/agilicus_api/configuration.py,sha256=zqtUfDap0pj3Nn8V_QBhxK25G9O_nprzqQXQ7KwsS6A,18447
+agilicus/agilicus_api/configuration.py,sha256=9ltB_6PTmjPj5jwILWqCCsC1pW12gG6IT5KTeeoSpzk,18447
 agilicus/agilicus_api/docs/APIKey.md,sha256=4cKuz4_l9HcEDnUrLwYbEnn9C2WoDayrjfrY1Ixgaf4,1747
 agilicus/agilicus_api/docs/APIKeyIntrospect.md,sha256=nJ-zkuFm3JMbWFDYYN_vYyQk1snGBtBvIxtCQxamhAU,1019
 agilicus/agilicus_api/docs/APIKeyIntrospectAuthorizationInfo.md,sha256=7RApOOLjvWQs5sw2jb25g7i3Kta1BiEY-s8VRXfppH8,725
@@ -2457,7 +2457,7 @@ agilicus/agilicus_api/test/test_x509_root_certificate.py,sha256=0hnwn-PQn0mBdoeS
 agilicus/agilicus_api/test/test_x509_root_certificate_spec.py,sha256=45tuo6GHC_4V88Zle_dFQ-0bYas412T-JRSjuBqX8Yk,2832
 agilicus/agilicus_api/test/test_x509_root_certificate_status.py,sha256=yY7C4F-oLokoBltUxNSUaNin3Ajjj-IexZfZtqFEmjo,2846
 agilicus/agilicus_api/test/test_xss_settings.py,sha256=YbRSwHp5Gbl1XK-UPKWNsbJ-bqQsFLWMWhFqrm0Ggl4,2746
-agilicus/agilicus_api_README.md,sha256=TEVlJxq4uqu3MXI0gr3tbCg71ec6Krc6YAdOMqxUYtE,158240
+agilicus/agilicus_api_README.md,sha256=KUU_QPiy330-hQZpWhDBlSwfXZL6wFu4t5mj4BWLYJ8,158240
 agilicus/aliases.ini,sha256=MxqiVo2f2xdUDVF1YDkNW36AIqN8hrYjlTVfraEUZXY,455
 agilicus/amq.py,sha256=yxi-YTbJPVl10s78Hlr1dmrQR63iaSIoROGVILzFPmE,1775
 agilicus/apps.py,sha256=8k_tGraPUVKJURksZwKR8eNGW-4CSPv54FYcApQ3Ly8,50857
@@ -2471,6 +2471,7 @@ agilicus/click_extension.py,sha256=uE57TjMqJxO9VFPBJO5-bTo9UrQiHPjIdI_iwKSaw6Q,2
 agilicus/client_secrets.json,sha256=eyPFVzziH48RuwWl66cfi0JE4jpmoec0Vo_F2TzYOeI,414
 agilicus/connectors.py,sha256=RhnLMxQNNEgmkh9lCSQazL0uovt7lFuJJ1Eo1Wv9qqE,41229
 agilicus/context.py,sha256=5nf4_EZ3sEpKvhQAHn4DA0HNMotChtxlcmjbPErbNV4,7953
+agilicus/create.py,sha256=ZBZS0FmGhtBO7HUST5ljri2R786y80cDeg8sd1uJnd0,3004
 agilicus/credentials.py,sha256=F3yPAwoMcziP1OLmzeJ8jIz1W-91-HXu4zcymyMp1E4,4730
 agilicus/credentials_commands/credentials.py,sha256=YduI2SJR8nJgFuAloNowy16_s-rdVLZLDkuQvk_YmAg,3574
 agilicus/credentials_commands/credentials_main.py,sha256=V8Nm-WptM_3BuIcS5AGg62x0ffi_FoiKBTWzBlKooTc,2961
@@ -2520,8 +2521,8 @@ agilicus/output/tests/column_builder_test.py,sha256=fKP4V5sqXtmEIr-Q0gWVSFdBqCUt
 agilicus/pagination/pagination.py,sha256=hC7eLHxXKe4Iv7LdOZK3Dh8R_fKQIr9PPiWKJVyP4Nk,1672
 agilicus/patches.py,sha256=qTqLOgCAcFZPcPOkgfqMnK9bnqTXMvj_0ERsjRFcZug,1384
 agilicus/permissions.py,sha256=uB65yuDFICp1N10m0cdUjgizx9MQzAbLbAsBSTw1Rcc,2117
-agilicus/policy/policies.py,sha256=Hq4t9bhJbHxdMpZUzgA-1WB5dHLyjI0zeWj33hBRy4I,3730
-agilicus/policy/policy_main.py,sha256=Zjc8gir2UbCjVZOG5-Sx8HDPJ5aPlsj6VnkuRc5yXCw,1943
+agilicus/policy/policies.py,sha256=g6nuLIXpeq0fjx6zX_aKqgv8GjFC0lHmEV8t-MF26yo,5015
+agilicus/policy/policy_main.py,sha256=fQ5W3w0WulBzuAzz4Cta5xNMT030z1Wl1Vwfjmsgc8g,2615
 agilicus/policy/templates.py,sha256=EUvAhOhSjDuLAe2UKIFEzBp5pNpOT5y1r7uyV_GG6uI,5638
 agilicus/policy_config/policy_config.py,sha256=fjiyAmZ9TEGNGHQzEtLgLT7ZnPm_KgdG24oQXObsBSM,583
 agilicus/policy_config/policy_config_main.py,sha256=s9wBRtGNY5l4rIES_x2hQ-BTZ5hoIZ7SQZQeFU2so24,767
@@ -2547,8 +2548,8 @@ agilicus/trusted_certs/trusted_certs_main.py,sha256=6dHHWXvNIcUa_nA9ptigL4Vibe4n
 agilicus/users.py,sha256=JT7TIiUOtSFXPOdxmIVFm7ygZTH1FjsIkD9j-vjLuMM,38474
 agilicus/version.py,sha256=G9OFdL1v_4dLDfk6I6taDNypM5bbO-JHAwilsu9LYgg,23
 agilicus/whoami.py,sha256=kqghtWMgZOd2rhKmfguDwCTm6A3gNS8Kj-S2IBxBtl0,206
-agilicus-1.251.4.dist-info/LICENSE.txt,sha256=Zq4tqiCroC2CVrBB_PWjapRdvpae23nljdiaSkOzUho,1061
-agilicus-1.251.4.dist-info/METADATA,sha256=_-hEUFB8gzHxRbPNM-qEv47KO2g0wRR6tIl2ZHWL2g8,3822
-agilicus-1.251.4.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-agilicus-1.251.4.dist-info/entry_points.txt,sha256=a66hGozzLkHu0IewFzIMbSAhMTNTddUaA2T3_16Gb_s,51
-agilicus-1.251.4.dist-info/RECORD,,
+agilicus-1.253.0.dist-info/LICENSE.txt,sha256=Zq4tqiCroC2CVrBB_PWjapRdvpae23nljdiaSkOzUho,1061
+agilicus-1.253.0.dist-info/METADATA,sha256=GYd-hlk1FxaDh05ns_rUuoie9fcIq7ruXAEWkqxZHe8,3822
+agilicus-1.253.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+agilicus-1.253.0.dist-info/entry_points.txt,sha256=a66hGozzLkHu0IewFzIMbSAhMTNTddUaA2T3_16Gb_s,51
+agilicus-1.253.0.dist-info/RECORD,,