stigg-api-client 3.13.0__py3-none-any.whl → 5.109.3__py3-none-any.whl

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/generated/operations.py

@@ -56,6 +56,10 @@ def fragment_price_fragment():
     _frag_price = _frag.price()
     _frag_price.amount()
     _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())
@@ -112,6 +116,7 @@ def fragment_package_entitlement_fragment():
     _frag.is_custom()
     _frag.display_name_override()
     _frag.enum_values()
+    _frag.is_granted()
     _frag_feature = _frag.feature()
     _frag_feature.feature_type()
     _frag_feature.meter_type()
@@ -189,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
 
 
@@ -251,12 +257,17 @@ def fragment_slim_subscription_fragment_v2():
     _frag.status()
     _frag.pricing_type()
     _frag.start_date()
+    _frag.end_date()
     _frag.cancellation_date()
+    _frag.effective_end_date()
     _frag.current_billing_period_end()
+    _frag.additional_meta_data()
     _frag_customer = _frag.customer()
     _frag_customer.customer_id()
+    _frag_customer.email()
     _frag_paying_customer = _frag.paying_customer()
     _frag_paying_customer.customer_id()
+    _frag_paying_customer.email()
     _frag_resource = _frag.resource()
     _frag_resource.resource_id()
     _frag_plan = _frag.plan()
@@ -266,6 +277,13 @@ def fragment_slim_subscription_fragment_v2():
     _frag_addons.quantity()
     _frag_addons_addon = _frag_addons.addon()
     _frag_addons_addon.ref_id(__alias__='addon_id')
+    _frag_prices = _frag.prices()
+    _frag_prices.billing_model()
+    _frag_prices_price = _frag_prices.price()
+    _frag_prices_price.billing_period()
+    _frag_prices_price_price = _frag_prices_price.price()
+    _frag_prices_price_price.amount()
+    _frag_prices_price_price.currency()
     _frag_trial_configuration = _frag.trial_configuration()
     _frag_trial_configuration.trial_end_behavior()
     _frag.trial_end_date()
@@ -349,6 +367,7 @@ def fragment_subscription_invoice_fragment():
     _frag.billing_id()
     _frag.status()
     _frag.created_at()
+    _frag.due_date()
     _frag.updated_at()
     _frag.error_message()
     _frag.requires_action()
@@ -827,6 +846,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
 
 
@@ -889,6 +912,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()
@@ -925,6 +949,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()
@@ -945,6 +988,20 @@ def fragment_customer_portal_fragment():
     return _frag
 
 
+def fragment_stripe_checkout_credentials_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.StripeCheckoutCredentials, 'StripeCheckoutCredentialsFragment')
+    _frag.account_id()
+    _frag.setup_secret()
+    _frag.public_key()
+    return _frag
+
+
+def fragment_zuora_checkout_credentials_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.ZuoraCheckoutCredentials, 'ZuoraCheckoutCredentialsFragment')
+    _frag.publishable_key()
+    return _frag
+
+
 def fragment_checkout_state_fragment():
     _frag = sgqlc.operation.Fragment(_schema.CheckoutState, 'CheckoutStateFragment')
     _frag_configuration = _frag.configuration()
@@ -960,6 +1017,11 @@ def fragment_checkout_state_fragment():
     _frag_plan.__fragment__(fragment_plan_fragment())
     _frag_billing_integration = _frag.billing_integration()
     _frag_billing_integration.billing_identifier()
+    _frag_billing_integration_billing_credentials = _frag_billing_integration.billing_credentials()
+    _frag_billing_integration_billing_credentials__as__StripeCheckoutCredentials = _frag_billing_integration_billing_credentials.__as__(_schema.StripeCheckoutCredentials)
+    _frag_billing_integration_billing_credentials__as__StripeCheckoutCredentials.__fragment__(fragment_stripe_checkout_credentials_fragment())
+    _frag_billing_integration_billing_credentials__as__ZuoraCheckoutCredentials = _frag_billing_integration_billing_credentials.__as__(_schema.ZuoraCheckoutCredentials)
+    _frag_billing_integration_billing_credentials__as__ZuoraCheckoutCredentials.__fragment__(fragment_zuora_checkout_credentials_fragment())
     _frag_billing_integration_credentials = _frag_billing_integration.credentials()
     _frag_billing_integration_credentials.account_id()
     _frag_billing_integration_credentials.public_key()
@@ -1009,6 +1071,9 @@ def fragment_customer_portal_subscription_price_fragment():
     _frag_price = _frag.price()
     _frag_price.amount()
     _frag_price.currency()
+    _frag_credit_rate = _frag.credit_rate()
+    _frag_credit_rate.amount()
+    _frag_credit_rate.currency_id()
     _frag_feature = _frag.feature()
     _frag_feature.id()
     _frag_feature.ref_id()
@@ -1035,6 +1100,9 @@ def fragment_customer_portal_subscription_fragment():
     _frag_pricing_price = _frag_pricing.price()
     _frag_pricing_price.amount()
     _frag_pricing_price.currency()
+    _frag_pricing_credit_rate = _frag_pricing.credit_rate()
+    _frag_pricing_credit_rate.amount()
+    _frag_pricing_credit_rate.currency_id()
     _frag_pricing_feature = _frag_pricing.feature()
     _frag_pricing_feature.feature_units()
     _frag_pricing_feature.feature_units_plural()
@@ -1171,6 +1239,7 @@ def fragment_mock_paywall_package_entitlement_fragment():
     _frag.hidden_from_widgets()
     _frag.display_name_override()
     _frag.enum_values()
+    _frag.is_granted()
     _frag_feature = _frag.feature()
     _frag_feature.feature_type()
     _frag_feature.meter_type()
@@ -1194,6 +1263,9 @@ def fragment_mock_paywall_price_fragment():
     _frag_price = _frag.price()
     _frag_price.amount()
     _frag_price.currency()
+    _frag_credit_rate = _frag.credit_rate()
+    _frag_credit_rate.amount()
+    _frag_credit_rate.currency_id()
     _frag.tiers_mode()
     _frag_tiers = _frag.tiers()
     _frag_tiers.__fragment__(fragment_price_tier_fragment())
@@ -1414,13 +1486,160 @@ 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()
+    _frag.amount()
+    _frag.comment()
+    _frag.currency_id()
+    _frag.customer_id()
+    _frag.display_name()
+    _frag.effective_at()
+    _frag.expire_at()
+    _frag.grant_type()
+    _frag.priority()
+    _frag.consumed_amount()
+    _frag_cost = _frag.cost()
+    _frag_cost.amount()
+    _frag_cost.currency()
+    _frag.created_at()
+    _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
+
+
+def fragment_credits_balance_summary_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.CreditBalanceSummary, 'CreditsBalanceSummaryFragment')
+    _frag.customer_id()
+    _frag_balances = _frag.balances()
+    _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
+
+
+def fragment_payment_session_fragment():
+    _frag = sgqlc.operation.Fragment(_schema.PaymentSession, 'PaymentSessionFragment')
+    _frag.token()
+    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.current_monthly_spend()
+    _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()
     customer_portal_configuration_fragment = fragment_customer_portal_configuration_fragment()
@@ -1451,6 +1670,7 @@ class Fragment:
     package_entitlement_fragment = fragment_package_entitlement_fragment()
     package_published_payload = fragment_package_published_payload()
     page_info_fragment = fragment_page_info_fragment()
+    payment_session_fragment = fragment_payment_session_fragment()
     paywall_calculated_price_points_fragment = fragment_paywall_calculated_price_points_fragment()
     paywall_configuration_fragment = fragment_paywall_configuration_fragment()
     paywall_currency_fragment = fragment_paywall_currency_fragment()
@@ -1466,9 +1686,11 @@ 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()
+    stripe_checkout_credentials_fragment = fragment_stripe_checkout_credentials_fragment()
     subscription_fragment = fragment_subscription_fragment()
     subscription_future_update_data = fragment_subscription_future_update_data()
     subscription_invoice_fragment = fragment_subscription_invoice_fragment()
@@ -1484,6 +1706,7 @@ class Fragment:
     usage_history_fragment = fragment_usage_history_fragment()
     usage_history_v2_fragment = fragment_usage_history_v2_fragment()
     usage_updated_fragment = fragment_usage_updated_fragment()
+    zuora_checkout_credentials_fragment = fragment_zuora_checkout_credentials_fragment()
 
 
 def mutation_provision_customer():
@@ -1677,17 +1900,77 @@ def mutation_transfer_subscription_to_resource():
     return _op
 
 
+def mutation_grant_credits():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='GrantCredits', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.CreditGrantInput))))
+    _op_create_credit_grant = _op.create_credit_grant(input=sgqlc.types.Variable('input'))
+    _op_create_credit_grant.__fragment__(fragment_credit_grant_fragment())
+    return _op
+
+
+def mutation_grant_promotional_entitlements_group():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='GrantPromotionalEntitlementsGroup', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.GrantPromotionalEntitlementsGroupInput))))
+    _op_grant_promotional_entitlements_group = _op.grant_promotional_entitlements_group(input=sgqlc.types.Variable('input'))
+    _op_grant_promotional_entitlements_group.__fragment__(fragment_promotional_entitlement_fragment())
+    return _op
+
+
+def mutation_revoke_promotional_entitlements_group():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='RevokePromotionalEntitlementsGroup', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.RevokePromotionalEntitlementsGroupInput))))
+    _op_revoke_promotional_entitlements_group = _op.revoke_promotional_entitlements_group(input=sgqlc.types.Variable('input'))
+    _op_revoke_promotional_entitlements_group.__fragment__(fragment_promotional_entitlement_fragment())
+    return _op
+
+
+def mutation_unlink_promotional_entitlements_group():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='UnlinkPromotionalEntitlementsGroup', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.UnlinkPromotionalEntitlementsGroupInput))))
+    _op_unlink_promotional_entitlements_group = _op.unlink_promotional_entitlements_group(input=sgqlc.types.Variable('input'))
+    _op_unlink_promotional_entitlements_group.__fragment__(fragment_promotional_entitlement_fragment())
+    return _op
+
+
+def mutation_create_payment_session():
+    _op = sgqlc.operation.Operation(_schema_root.mutation_type, name='CreatePaymentSession', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.PaymentSessionInput))))
+    _op_create_payment_session = _op.create_payment_session(input=sgqlc.types.Variable('input'))
+    _op_create_payment_session.__fragment__(fragment_payment_session_fragment())
+    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()
     detach_customer_payment_method = mutation_detach_customer_payment_method()
     estimate_subscription = mutation_estimate_subscription()
     estimate_subscription_update = mutation_estimate_subscription_update()
+    grant_credits = mutation_grant_credits()
     grant_promotional_entitlements = mutation_grant_promotional_entitlements()
+    grant_promotional_entitlements_group = mutation_grant_promotional_entitlements_group()
     import_customer = mutation_import_customer()
     import_customer_bulk = mutation_import_customer_bulk()
     import_subscriptions_bulk = mutation_import_subscriptions_bulk()
@@ -1701,11 +1984,15 @@ class Mutation:
     report_usage = mutation_report_usage()
     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():
@@ -1756,7 +2043,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())
@@ -1777,6 +2064,16 @@ def query_get_entitlements():
     return _op
 
 
+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()
+    return _op
+
+
 def query_get_entitlement():
     _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetEntitlement', variables=dict(query=sgqlc.types.Arg(sgqlc.types.non_null(_schema.FetchEntitlementQuery))))
     _op_entitlement = _op.entitlement(query=sgqlc.types.Variable('query'))
@@ -1798,6 +2095,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
 
 
@@ -1839,16 +2137,68 @@ def query_get_usage_history_v2():
     return _op
 
 
+def query_get_credit_balance():
+    _op = sgqlc.operation.Operation(_schema_root.query_type, name='GetCreditBalance', variables=dict(input=sgqlc.types.Arg(sgqlc.types.non_null(_schema.CreditBalanceSummaryInput))))
+    _op_credit_balance_summary = _op.credit_balance_summary(input=sgqlc.types.Variable('input'))
+    _op_credit_balance_summary.__fragment__(fragment_credits_balance_summary_fragment())
+    return _op
+
+
+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_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()
     get_entitlement = query_get_entitlement()
     get_entitlements = query_get_entitlements()
+    get_entitlements_state = query_get_entitlements_state()
     get_mock_paywall = query_get_mock_paywall()
     get_paywall = query_get_paywall()
     get_products = query_get_products()
@@ -1880,7 +2230,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()