add create invitation with payments to lightspark sdk

Author: matthappens

lightspark/lightspark_client.py

@@ -15,6 +15,7 @@
     PrivateFormat,
     PublicFormat,
 )
+
 from lightspark.exceptions import LightsparkException
 from lightspark.objects.Account import Account
 from lightspark.objects.Account import from_json as Account_from_json
@@ -79,6 +80,7 @@
 from lightspark.scripts.create_uma_invitation import (
     CREATE_UMA_INVITATION_MUTATION,
     CREATE_UMA_INVITATION_WITH_INCENTIVES_MUTATION,
+    CREATE_UMA_INVITATION_WITH_PAYMENT_MUTATION,
 )
 from lightspark.scripts.create_uma_invoice import CREATE_UMA_INVOICE_MUTATION
 from lightspark.scripts.current_account import CURRENT_ACCOUNT_QUERY
@@ -1015,5 +1017,37 @@ def fail_htlcs(self, invoice_id: str, cancel_invoice: bool = True) -> str:
 
         return json["fail_htlcs"]["invoice"]["id"]
 
+    def create_uma_invitation_with_payment(
+        self,
+        inviter_uma: str,
+        payment_amount: float,
+        payment_currency: Any,
+        expires_at: str,
+    ) -> UmaInvitation:
+        """
+        Creates a new UMA invitation with payment.
+
+        Args:
+            inviter_uma: The UMA of the inviter.
+            payment_amount: The amount to be paid.
+            payment_currency: The currency object (should have code, symbol, name, decimals attributes).
+            expires_at: The expiration datetime as an ISO8601 string.
+        """
+        json = self._requester.execute_graphql(
+            CREATE_UMA_INVITATION_WITH_PAYMENT_MUTATION,
+            {
+                "inviter_uma": inviter_uma,
+                "payment_amount": payment_amount,
+                "payment_currency_code": payment_currency.code,
+                "payment_currency_symbol": payment_currency.symbol,
+                "payment_currency_name": payment_currency.name,
+                "payment_currency_decimals": payment_currency.decimals,
+                "expires_at": expires_at,
+            },
+        )
+        return UmaInvitation_from_json(
+            self._requester, json["create_uma_invitation_with_payment"]["invitation"]
+        )
+
 
 E614_REGEX = re.compile(r"^\+?[1-9]\d{1,14}$")

lightspark/scripts/create_uma_invitation.py

@@ -37,3 +37,33 @@
 
 {INVITATION_FRAGMENT}
 """
+
+CREATE_UMA_INVITATION_WITH_PAYMENT_MUTATION = f"""
+mutation CreateUmaInvitationWithPayment(
+    $inviter_uma: String!
+    $payment_amount: Float!
+    $payment_currency_code: String!
+    $payment_currency_symbol: String!
+    $payment_currency_name: String!
+    $payment_currency_decimals: Int!
+    $expires_at: DateTime!
+) {{
+    create_uma_invitation_with_payment(input: {{
+        inviter_uma: $inviter_uma
+        payment_amount: $payment_amount
+        payment_currency: {{
+          code: $payment_currency_code
+          symbol: $payment_currency_symbol
+          name: $payment_currency_name
+          decimals: $payment_currency_decimals
+        }}
+        expires_at: $expires_at
+    }}) {{
+        invitation {{
+            ...UmaInvitationFragment
+        }}
+    }}
+}}
+
+{INVITATION_FRAGMENT}
+"""