Some small fixes to the subscription-types CLI

pull/56/head
Neeraj Kashyap 2021-08-03 09:23:18 -07:00
rodzic bf9524cd16
commit d301e36ad3
1 zmienionych plików z 12 dodań i 15 usunięć

Wyświetl plik

@ -2,10 +2,6 @@
Moonstream CLI Moonstream CLI
""" """
import argparse import argparse
from distutils.util import strtobool
import json
from typing import List
import uuid
from bugout.data import BugoutResource, BugoutResources from bugout.data import BugoutResource, BugoutResources
from bugout.exceptions import BugoutResponseException from bugout.exceptions import BugoutResponseException
@ -33,7 +29,6 @@ def add_subscription_handler(args: argparse.Namespace) -> None:
params = {"type": "subscription_type"} params = {"type": "subscription_type"}
try: try:
# resolve index # resolve index
try: try:
resources: BugoutResources = bc.list_resources( resources: BugoutResources = bc.list_resources(
@ -57,6 +52,7 @@ def add_subscription_handler(args: argparse.Namespace) -> None:
print("Unexpected Exception on request to brood") print("Unexpected Exception on request to brood")
subscription_data = { subscription_data = {
"type": "subscription_type",
"id": str(new_subscription_id), "id": str(new_subscription_id),
"name": args.name, "name": args.name,
"description": args.description, "description": args.description,
@ -64,17 +60,11 @@ def add_subscription_handler(args: argparse.Namespace) -> None:
"active": args.active, "active": args.active,
} }
# Add subscriptions
resource_data = {"type": "subscription_type"}
resource_data.update(subscription_data)
try: try:
bc.create_resource(
resource: BugoutResource = bc.create_resource(
token=MOONSTREAM_ADMIN_ACCESS_TOKEN, token=MOONSTREAM_ADMIN_ACCESS_TOKEN,
application_id=MOONSTREAM_APPLICATION_ID, application_id=MOONSTREAM_APPLICATION_ID,
resource_data=resource_data, resource_data=subscription_data,
) )
except BugoutResponseException as e: except BugoutResponseException as e:
print(f"status_code={e.status_code}, detail={e.detail}") print(f"status_code={e.status_code}, detail={e.detail}")
@ -95,7 +85,7 @@ def main() -> None:
subcommands = parser.add_subparsers(description="Moonstream commands") subcommands = parser.add_subparsers(description="Moonstream commands")
parser_subscription = subcommands.add_parser( parser_subscription = subcommands.add_parser(
"subscription", description="Moonstream subscription" "subscription-type", description="Manage Moonstream subscription types"
) )
parser_subscription.set_defaults(func=lambda _: parser_subscription.print_help()) parser_subscription.set_defaults(func=lambda _: parser_subscription.print_help())
subcommands_subscription = parser_subscription.add_subparsers( subcommands_subscription = parser_subscription.add_subparsers(
@ -125,7 +115,7 @@ def main() -> None:
"--subscription_plan_id", "--subscription_plan_id",
required=False, required=False,
type=str, type=str,
help="Stripe sibscription id", help="Stripe subscription id",
) )
parser_subscription_create.add_argument( parser_subscription_create.add_argument(
"--active", "--active",
@ -133,3 +123,10 @@ def main() -> None:
help="Set this flag to create a verified user", help="Set this flag to create a verified user",
) )
parser_subscription_create.set_defaults(func=add_subscription_handler) parser_subscription_create.set_defaults(func=add_subscription_handler)
args = parser.parse_args()
args.func(args)
if __name__ == "__main__":
main()