stigg-api-client 3.84.1__tar.gz → 5.71.0__tar.gz

{stigg_api_client-3.84.1 → stigg_api_client-5.71.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stigg-api-client
-Version: 3.84.1
+Version: 5.71.0
 Summary: 
 License: STIGG SDK LICENSE
 Author: Stigg

{stigg_api_client-3.84.1 → stigg_api_client-5.71.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "stigg-api-client"
-version = "3.84.1"
+version = "5.71.0"
 description = ""
 authors = ["Stigg <support@stigg.io>"]
 license = "STIGG SDK LICENSE"

{stigg_api_client-3.84.1 → stigg_api_client-5.71.0}/stigg/client.py

@@ -29,18 +29,19 @@ class StiggClient:
 
         self._EDGE_QUERIES: Dict[Operation, Callable[[Dict], Dict]] = {
             Operations.query.get_entitlements: self.get_entitlements,
+            Operations.query.get_entitlements_state: self.get_entitlements_state,
             Operations.query.get_paywall: self.get_paywall,
         }
 
     @retry(**RETRY_KWARGS)
-    def _get_entitlements(self, variables: Dict[str, Dict]) -> Dict:
+    def _get_entitlements(self, variables: Dict[str, Dict], endpoint: str) -> Dict:
         params = variables.get("query")
         if params is None:
             raise ValueError('Variables do not include required key "query".')
 
         customer_id = params.pop("customerId")
 
-        url = f"{self._edge_url}/v1/c/{customer_id}/entitlements.json"
+        url = f"{self._edge_url}/v1/c/{customer_id}/{endpoint}"
 
         data = requests.get(url, headers=self._endpoint.base_headers, params=params, timeout=self._endpoint.timeout)
         data.raise_for_status()
@@ -51,7 +52,13 @@ class StiggClient:
         if self._enable_edge is False:
             return self.request(Operations.query.get_entitlements, variables, raw_response=True)
 
-        return self._get_entitlements(variables)
+        return self._get_entitlements(variables, endpoint="entitlements.json")
+
+    def get_entitlements_state(self, variables: Dict[str, Dict]) -> Dict:
+        if self._enable_edge is False:
+            return self.request(Operations.query.get_entitlements_state, variables, raw_response=True)
+
+        return self._get_entitlements(variables, endpoint="entitlements-state.json")
 
     @retry(**RETRY_KWARGS)
     def _get_paywall(self,variables: Dict[str, Dict]) -> Dict:

{stigg_api_client-3.84.1 → stigg_api_client-5.71.0}/stigg/generated/operations.py

@@ -59,6 +59,7 @@ def fragment_price_fragment():
     _frag_credit_rate = _frag.credit_rate()
     _frag_credit_rate.amount()
     _frag_credit_rate.custom_currency_id()
+    _frag_credit_rate.currency_id()
     _frag.tiers_mode()
     _frag_tiers = _frag.tiers()
     _frag_tiers.__fragment__(fragment_price_tier_fragment())
@@ -193,6 +194,7 @@ def fragment_plan_fragment():
     _frag_default_trial_config_budget = _frag_default_trial_config.budget()
     _frag_default_trial_config_budget.limit()
     _frag_default_trial_config.trial_end_behavior()
+    _frag.aws_marketplace_plan_dimension()
     return _frag
 
 
@@ -832,6 +834,10 @@ def fragment_entitlement_fragment():
     _frag_reset_period_configuration.__fragment__(fragment_reset_period_configuration_fragment())
     _frag_feature = _frag.feature()
     _frag_feature.__fragment__(fragment_feature_fragment())
+    _frag_credit_rate = _frag.credit_rate()
+    _frag_credit_rate.amount()
+    _frag_credit_rate.currency_id()
+    _frag.valid_until()
     return _frag
 
 
@@ -894,6 +900,7 @@ def fragment_product_fragment():
     _frag.display_name()
     _frag.description()
     _frag.additional_meta_data()
+    _frag.aws_marketplace_product_id()
     _frag_product_settings = _frag.product_settings()
     _frag_product_settings_downgrade_plan = _frag_product_settings.downgrade_plan()
     _frag_product_settings_downgrade_plan.ref_id()
@@ -930,6 +937,25 @@ def fragment_package_published_payload():
     return _frag
 
 
+def fragment_credit_balance_updated_payload():
+    _frag = sgqlc.operation.Fragment(_schema.CreditBalanceUpdated, 'CreditBalanceUpdatedPayload')
+    _frag_currency = _frag.currency()
+    _frag_currency.currency_id()
+    _frag_currency.display_name()
+    _frag_currency.symbol()
+    _frag_currency_units = _frag_currency.units()
+    _frag_currency_units.singular()
+    _frag_currency_units.plural()
+    _frag.currency_id()
+    _frag.current_balance()
+    _frag.customer_id()
+    _frag.valid_until()
+    _frag.resource_id()
+    _frag.total_consumed()
+    _frag.total_granted()
+    return _frag
+
+
 def fragment_customer_portal_fragment():
     _frag = sgqlc.operation.Fragment(_schema.CustomerPortal, 'CustomerPortalFragment')
     _frag_subscriptions = _frag.subscriptions()
@@ -1035,7 +1061,7 @@ def fragment_customer_portal_subscription_price_fragment():
     _frag_price.currency()
     _frag_credit_rate = _frag.credit_rate()
     _frag_credit_rate.amount()
-    _frag_credit_rate.custom_currency_id()
+    _frag_credit_rate.currency_id()
     _frag_feature = _frag.feature()
     _frag_feature.id()
     _frag_feature.ref_id()
@@ -1064,7 +1090,7 @@ def fragment_customer_portal_subscription_fragment():
     _frag_pricing_price.currency()
     _frag_pricing_credit_rate = _frag_pricing.credit_rate()
     _frag_pricing_credit_rate.amount()
-    _frag_pricing_credit_rate.custom_currency_id()
+    _frag_pricing_credit_rate.currency_id()
     _frag_pricing_feature = _frag_pricing.feature()
     _frag_pricing_feature.feature_units()
     _frag_pricing_feature.feature_units_plural()
@@ -1227,7 +1253,7 @@ def fragment_mock_paywall_price_fragment():
     _frag_price.currency()
     _frag_credit_rate = _frag.credit_rate()
     _frag_credit_rate.amount()
-    _frag_credit_rate.custom_currency_id()
+    _frag_credit_rate.currency_id()
     _frag.tiers_mode()
     _frag_tiers = _frag.tiers()
     _frag_tiers.__fragment__(fragment_price_tier_fragment())
@@ -1448,6 +1474,30 @@ def fragment_schedule_variables_fragment():
     return _frag
 
 
+def fragment_credit_grant_invoice_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.CreditGrantInvoice, 'CreditGrantInvoiceFragment')
+    _frag.billing_id()
+    _frag.status()
+    _frag.created_at()
+    _frag.due_date()
+    _frag.updated_at()
+    _frag.error_message()
+    _frag.requires_action()
+    _frag.payment_secret()
+    _frag.payment_url()
+    _frag.pdf_url()
+    _frag.billing_reason()
+    _frag.currency()
+    _frag.sub_total()
+    _frag.sub_total_excluding_tax()
+    _frag.total()
+    _frag.total_excluding_tax()
+    _frag.tax()
+    _frag.amount_due()
+    _frag.attempt_count()
+    return _frag
+
+
 def fragment_credit_grant_fragment():
     _frag = sgqlc.operation.Fragment(_schema.CreditGrant, 'CreditGrantFragment')
     _frag.grant_id()
@@ -1468,6 +1518,26 @@ def fragment_credit_grant_fragment():
     _frag.updated_at()
     _frag.resource_id()
     _frag.additional_meta_data()
+    _frag.automatic_recharge_configuration_id()
+    _frag.status()
+    _frag.payment_collection()
+    _frag.invoice_id()
+    _frag_latest_invoice = _frag.latest_invoice()
+    _frag_latest_invoice.__fragment__(fragment_credit_grant_invoice_fragment())
+    return _frag
+
+
+def fragment_credit_ledger_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.CreditLedgerEvent, 'CreditLedgerFragment')
+    _frag.timestamp()
+    _frag.event_type()
+    _frag.customer_id()
+    _frag.resource_id()
+    _frag.event_id()
+    _frag.feature_id()
+    _frag.amount()
+    _frag.credit_grant_id()
+    _frag.credit_currency_id()
     return _frag
 
 
@@ -1475,16 +1545,25 @@ def fragment_credits_balance_summary_fragment():
     _frag = sgqlc.operation.Fragment(_schema.CreditBalanceSummary, 'CreditsBalanceSummaryFragment')
     _frag.customer_id()
     _frag_balances = _frag.balances()
-    _frag_balances_currency = _frag_balances.currency()
-    _frag_balances_currency.currency_id()
-    _frag_balances_currency.display_name()
-    _frag_balances_currency.symbol()
-    _frag_balances_currency_units = _frag_balances_currency.units()
-    _frag_balances_currency_units.singular()
-    _frag_balances_currency_units.plural()
-    _frag_balances.current_balance()
-    _frag_balances.total_consumed()
-    _frag_balances.total_granted()
+    _frag_balances.__fragment__(fragment_credit_balance_fragment())
+    return _frag
+
+
+def fragment_credit_balance_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.CreditBalance, 'CreditBalanceFragment')
+    _frag.customer_id()
+    _frag_currency = _frag.currency()
+    _frag_currency.currency_id()
+    _frag_currency.display_name()
+    _frag_currency.symbol()
+    _frag_currency_units = _frag_currency.units()
+    _frag_currency_units.singular()
+    _frag_currency_units.plural()
+    _frag.current_balance()
+    _frag.total_consumed()
+    _frag.total_granted()
+    _frag.resource_id()
+    _frag.valid_until()
     return _frag
 
 
@@ -1494,14 +1573,59 @@ def fragment_payment_session_fragment():
     return _frag
 
 
+def fragment_slim_custom_currency_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.SlimCustomCurrency, 'SlimCustomCurrencyFragment')
+    _frag.currency_id()
+    _frag.display_name()
+    _frag_units = _frag.units()
+    _frag_units.singular()
+    _frag_units.plural()
+    return _frag
+
+
+def fragment_credit_usage_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.CreditUsage, 'CreditUsageFragment')
+    _frag_series = _frag.series()
+    _frag_series.feature_id()
+    _frag_series.feature_name()
+    _frag_series.total_credits()
+    _frag_series_points = _frag_series.points()
+    _frag_series_points.timestamp()
+    _frag_series_points.value()
+    _frag_currency = _frag.currency()
+    _frag_currency.__fragment__(fragment_slim_custom_currency_fragment())
+    return _frag
+
+
+def fragment_auto_recharge_settings_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.AutoRechargeSettingsDTO, 'AutoRechargeSettingsFragment')
+    _frag.customer_id()
+    _frag.currency_id()
+    _frag.is_enabled()
+    _frag.threshold_type()
+    _frag.threshold_value()
+    _frag.target_balance()
+    _frag.max_spend_limit()
+    _frag.grant_expiration_period()
+    _frag.created_at()
+    _frag.updated_at()
+    return _frag
+
+
 class Fragment:
     addon_dependency_fragment = fragment_addon_dependency_fragment()
     addon_fragment = fragment_addon_fragment()
     apply_subscription_fragment = fragment_apply_subscription_fragment()
+    auto_recharge_settings_fragment = fragment_auto_recharge_settings_fragment()
     checkout_configuration_fragment = fragment_checkout_configuration_fragment()
     checkout_state_fragment = fragment_checkout_state_fragment()
     coupon_fragment = fragment_coupon_fragment()
+    credit_balance_fragment = fragment_credit_balance_fragment()
+    credit_balance_updated_payload = fragment_credit_balance_updated_payload()
     credit_grant_fragment = fragment_credit_grant_fragment()
+    credit_grant_invoice_fragment = fragment_credit_grant_invoice_fragment()
+    credit_ledger_fragment = fragment_credit_ledger_fragment()
+    credit_usage_fragment = fragment_credit_usage_fragment()
     credits_balance_summary_fragment = fragment_credits_balance_summary_fragment()
     customer_fragment = fragment_customer_fragment()
     customer_portal_billing_information_fragment = fragment_customer_portal_billing_information_fragment()
@@ -1549,6 +1673,7 @@ class Fragment:
     report_usage_fragment = fragment_report_usage_fragment()
     reset_period_configuration_fragment = fragment_reset_period_configuration_fragment()
     schedule_variables_fragment = fragment_schedule_variables_fragment()
+    slim_custom_currency_fragment = fragment_slim_custom_currency_fragment()
     slim_customer_fragment = fragment_slim_customer_fragment()
     slim_subscription_fragment = fragment_slim_subscription_fragment()
     slim_subscription_fragment_v2 = fragment_slim_subscription_fragment_v2()
@@ -1797,11 +1922,33 @@ def mutation_create_payment_session():
     return _op
 
 
+def mutation_void_credit_grant():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='VoidCreditGrant', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.VoidCreditGrantInput))))
+    _op_void_credit_grant = _op.void_credit_grant(input=sgqlc.types.Variable('input'))
+    _op_void_credit_grant.__fragment__(fragment_credit_grant_fragment())
+    return _op
+
+
+def mutation_create_coupon():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='CreateCoupon', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.CreateCouponInput))))
+    _op_create_one_coupon = _op.create_one_coupon(input=sgqlc.types.Variable('input'))
+    _op_create_one_coupon.__fragment__(fragment_coupon_fragment())
+    return _op
+
+
+def mutation_save_automatic_recharge_configuration():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='SaveAutomaticRechargeConfiguration', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.SaveAutoRechargeSettingsInput))))
+    _op_save_auto_recharge_settings = _op.save_auto_recharge_settings(input=sgqlc.types.Variable('input'))
+    _op_save_auto_recharge_settings.__fragment__(fragment_auto_recharge_settings_fragment())
+    return _op
+
+
 class Mutation:
     apply_subscription = mutation_apply_subscription()
     archive_customer = mutation_archive_customer()
     cancel_subscription = mutation_cancel_subscription()
     cancel_subscription_updates = mutation_cancel_subscription_updates()
+    create_coupon = mutation_create_coupon()
     create_payment_session = mutation_create_payment_session()
     create_subscription = mutation_create_subscription()
     delegate_subscription_to_customer = mutation_delegate_subscription_to_customer()
@@ -1825,12 +1972,14 @@ class Mutation:
     report_usage_bulk = mutation_report_usage_bulk()
     revoke_promotional_entitlement = mutation_revoke_promotional_entitlement()
     revoke_promotional_entitlements_group = mutation_revoke_promotional_entitlements_group()
+    save_automatic_recharge_configuration = mutation_save_automatic_recharge_configuration()
     transfer_subscription = mutation_transfer_subscription()
     transfer_subscription_to_resource = mutation_transfer_subscription_to_resource()
     unarchive_customer = mutation_unarchive_customer()
     unlink_promotional_entitlements_group = mutation_unlink_promotional_entitlements_group()
     update_customer = mutation_update_customer()
     update_subscription = mutation_update_subscription()
+    void_credit_grant = mutation_void_credit_grant()
 
 
 def query_get_customer_by_id():
@@ -1881,7 +2030,7 @@ def query_get_subscription():
 
 def query_get_coupons():
     _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetCoupons')
-    _op_coupons = _op.coupons(filter={'status': {'eq': 'ACTIVE'}}, paging={'first': 50})
+    _op_coupons = _op.coupons(filter={'status': {'eq': 'ACTIVE'}}, paging={'first': 25})
     _op_coupons_edges = _op_coupons.edges()
     _op_coupons_edges_node = _op_coupons_edges.node()
     _op_coupons_edges_node.__fragment__(fragment_coupon_fragment())
@@ -1905,6 +2054,7 @@ def query_get_entitlements():
 def query_get_entitlements_state():
     _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetEntitlementsState', variables=dict(query=sgqlc.types.Arg(sgqlc.types.non_null(_schema.FetchEntitlementsQuery))))
     _op_entitlements_state = _op.entitlements_state(query=sgqlc.types.Variable('query'))
+    _op_entitlements_state.__typename__()
     _op_entitlements_state_entitlements = _op_entitlements_state.entitlements()
     _op_entitlements_state_entitlements.__fragment__(fragment_entitlement_fragment())
     _op_entitlements_state.access_denied_reason()
@@ -1932,6 +2082,7 @@ def query_get_sdk_configuration():
     _op_sdk_configuration = _op.sdk_configuration()
     _op_sdk_configuration.sentry_dsn()
     _op_sdk_configuration.is_widget_watermark_enabled()
+    _op_sdk_configuration.show_watermark()
     return _op
 
 
@@ -1983,17 +2134,52 @@ def query_get_credit_balance():
 def query_get_credit_grants():
     _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetCreditGrants', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.GetCreditGrantsInput))))
     _op_credit_grants = _op.credit_grants(input=sgqlc.types.Variable('input'))
-    _op_credit_grants.__fragment__(fragment_credit_grant_fragment())
+    _op_credit_grants_page_info = _op_credit_grants.page_info()
+    _op_credit_grants_page_info.__fragment__(fragment_page_info_fragment())
+    _op_credit_grants_edges = _op_credit_grants.edges()
+    _op_credit_grants_edges_node = _op_credit_grants_edges.node()
+    _op_credit_grants_edges_node.__fragment__(fragment_credit_grant_fragment())
+    _op_credit_grants.total_count()
+    return _op
+
+
+def query_get_credit_ledger():
+    _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetCreditLedger', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.CreditLedgerInput))))
+    _op_credits_ledger = _op.credits_ledger(input=sgqlc.types.Variable('input'))
+    _op_credits_ledger_page_info = _op_credits_ledger.page_info()
+    _op_credits_ledger_page_info.__fragment__(fragment_page_info_fragment())
+    _op_credits_ledger_edges = _op_credits_ledger.edges()
+    _op_credits_ledger_edges_node = _op_credits_ledger_edges.node()
+    _op_credits_ledger_edges_node.__fragment__(fragment_credit_ledger_fragment())
+    _op_credits_ledger_edges.cursor()
+    _op_credits_ledger.total_count()
+    return _op
+
+
+def query_get_credit_usage():
+    _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetCreditUsage', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.CreditUsageInput))))
+    _op_credit_usage = _op.credit_usage(input=sgqlc.types.Variable('input'))
+    _op_credit_usage.__fragment__(fragment_credit_usage_fragment())
+    return _op
+
+
+def query_get_automatic_recharge_configuration():
+    _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetAutomaticRechargeConfiguration', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.GetAutoRechargeSettingsInput))))
+    _op_get_auto_recharge_settings = _op.get_auto_recharge_settings(input=sgqlc.types.Variable('input'))
+    _op_get_auto_recharge_settings.__fragment__(fragment_auto_recharge_settings_fragment())
     return _op
 
 
 class Query:
     get_active_subscriptions = query_get_active_subscriptions()
     get_active_subscriptions_list = query_get_active_subscriptions_list()
+    get_automatic_recharge_configuration = query_get_automatic_recharge_configuration()
     get_checkout_state = query_get_checkout_state()
     get_coupons = query_get_coupons()
     get_credit_balance = query_get_credit_balance()
     get_credit_grants = query_get_credit_grants()
+    get_credit_ledger = query_get_credit_ledger()
+    get_credit_usage = query_get_credit_usage()
     get_customer_by_id = query_get_customer_by_id()
     get_customer_portal_by_ref_id = query_get_customer_portal_by_ref_id()
     get_customer_statistics = query_get_customer_statistics()
@@ -2031,7 +2217,15 @@ def subscription_on_package_published():
     return _op
 
 
+def subscription_on_credit_balance_updated():
+    _op = sgqlc.operation.Operation(_schema_root.subscription_type, name='OnCreditBalanceUpdated')
+    _op_credit_balance_updated = _op.credit_balance_updated()
+    _op_credit_balance_updated.__fragment__(fragment_credit_balance_updated_payload())
+    return _op
+
+
 class Subscription:
+    on_credit_balance_updated = subscription_on_credit_balance_updated()
     on_entitlements_updated = subscription_on_entitlements_updated()
     on_package_published = subscription_on_package_published()
     on_usage_updated = subscription_on_usage_updated()