From 4556d28f739be3cd1a1bd6711d31fc588fc02083 Mon Sep 17 00:00:00 2001 From: Abstra Bot Date: Thu, 11 May 2023 18:49:16 +0000 Subject: [PATCH] Update abstra-cli --- abstra_cli/apis/__init__.py | 1 + abstra_cli/apis/billing.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+) create mode 100644 abstra_cli/apis/billing.py diff --git a/abstra_cli/apis/__init__.py b/abstra_cli/apis/__init__.py index eaef4ae..c4f5cae 100644 --- a/abstra_cli/apis/__init__.py +++ b/abstra_cli/apis/__init__.py @@ -10,3 +10,4 @@ from .subdomains import * from .memberships import * from .workspaces import * +from .billing import * diff --git a/abstra_cli/apis/billing.py b/abstra_cli/apis/billing.py new file mode 100644 index 0000000..0f632a3 --- /dev/null +++ b/abstra_cli/apis/billing.py @@ -0,0 +1,17 @@ +import abstra_cli.credentials as credentials +import abstra_cli.apis.main as api_main +import requests + +billing_url = "https://billing.abstra.cloud" + + +def get_account_info(): + headers = credentials.get_auth_headers() + _, workspace_id, _ = api_main.get_auth_info() + response = requests.get( + f"{billing_url}/workspaces/{workspace_id}/account", headers=headers + ) + response_json = response.json() + if response_json is None: + return None + return response_json