viam-sdk 0.41.1__py3-none-linux_armv6l.whl → 0.66.0__py3-none-linux_armv6l.whl

viam/app/billing_client.py

@@ -5,6 +5,8 @@ from grpclib.client import Channel, Stream
 from viam import logging
 from viam.proto.app.billing import (
     BillingServiceStub,
+    CreateInvoiceAndChargeImmediatelyRequest,
+    CreateInvoiceAndChargeImmediatelyResponse,
     GetCurrentMonthUsageRequest,
     GetCurrentMonthUsageResponse,
     GetInvoicePdfRequest,
@@ -40,11 +42,9 @@ class BillingClient:
 
         async def main():
             # Make a ViamClient
-            viam_client = await connect()
-            # Instantiate a BillingClient to run billing client API methods on
-            billing_client = viam_client.billing_client
-
-            viam_client.close()
+            async with await connect() as viam_client:
+                # Instantiate a BillingClient to run billing client API methods on
+                billing_client = viam_client.billing_client
 
         if __name__ == '__main__':
             asyncio.run(main())
@@ -141,3 +141,45 @@ class BillingClient:
         """
         request = GetOrgBillingInformationRequest(org_id=org_id)
         return await self._billing_client.GetOrgBillingInformation(request, metadata=self._metadata, timeout=timeout)
+
+    async def create_invoice_and_charge_immediately(
+        self,
+        org_id_to_charge: str,
+        amount: float,
+        description: Optional[str] = None,
+        org_id_for_branding: Optional[str] = None,
+        disable_email: bool = False,
+    ) -> CreateInvoiceAndChargeImmediatelyResponse:
+        """Create a flat fee invoice and charge the organization on the spot. The caller must be an owner of the organization being charged.
+        This function returns the invoice id once the payment intent is successfully sent for processing. Callers may poll the invoice for
+        its status using the `get_invoices_summary` function and the returned invoice id. The status will be "payment_processing" if the
+        payment is being processed, "paid" if it succeeds, or "outstanding" if it fails.
+
+        ::
+
+            invoice_id = await billing_client.create_invoice_and_charge_immediately(
+                "<ORG-ID-TO-CHARGE>",
+                <AMOUNT>,
+                <DESCRIPTION>,
+                "<ORG-ID-FOR-BRANDING>",
+                False,
+            )
+
+        Args:
+            org_id_to_charge (str): the organization to charge
+            amount (float): the amount to charge in dollars
+            description (str): a short description of the charge to display on the invoice PDF (must be 100 characters or less)
+            org_id_for_branding (str): the organization whose branding to use in the invoice confirmation email
+            disable_email (bool): whether or not to disable sending an email confirmation for the invoice
+
+        Returns:
+            viam.proto.app.billing.CreateInvoiceAndChargeImmediatelyResponse: the invoice id
+        """
+        request = CreateInvoiceAndChargeImmediatelyRequest(
+            org_id_to_charge=org_id_to_charge,
+            amount=amount,
+            description=description,
+            org_id_for_branding=org_id_for_branding,
+            disable_email=disable_email,
+        )
+        return await self._billing_client.CreateInvoiceAndChargeImmediately(request, metadata=self._metadata)