kopia lustrzana https://github.com/bugout-dev/moonstream
Some small fixes to the subscription-types CLI
rodzic
bf9524cd16
commit
d301e36ad3
|
@ -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()
|
||||||
|
|
Ładowanie…
Reference in New Issue