uncloud-cli/commands/product.py
2019-09-19 15:50:14 +05:30

65 lines
2.4 KiB
Python
Executable file

import click
import requests
from decouple import config
from .helper import OTPCredentials, load_dump_pretty
@click.group()
def product():
pass
@product.command("list")
def list():
r = requests.get("{}/product/list".format(config('UCLOUD_PAY_SERVER')))
print(load_dump_pretty(r.content))
@product.command("add")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--product-name", required=True)
@click.option("--product-description", required=True)
@click.option("--product-type", required=True,
help="Either one-time or recurring",
type=click.Choice(['one-time', 'recurring']))
@click.option("--product-price", required=True, help="Price in CHF cents",
type=int)
@click.option("--product-recurring-period", required=False,
help="Only used for products with recurring costs")
@click.option("--product-minimum-subscription-period", required=False,
help="Minimum period for which a user needs to "
"subscribe the product for")
def add(name, realm, seed, product_name, product_description, product_type,
product_price, product_recurring_period,
product_minimum_subscription_period):
data = {
**OTPCredentials(name, realm, seed).get_json(),
"product_name": product_name,
"product_description": product_description,
"product_type": product_type,
"product_price": product_price,
"product_recurring_period": product_recurring_period,
"product_minimum_subscription_period":
product_minimum_subscription_period
}
r = requests.post("{}/product/add".format(config('UCLOUD_PAY_SERVER')),
json=data)
print(load_dump_pretty(r.content))
@product.command("order")
@click.option("--name", envvar="OTP_NAME", required=True)
@click.option("--realm", envvar="OTP_REALM", required=True)
@click.option("--seed", envvar="OTP_SEED", required=True)
@click.option("--product-id", help="The uuid of the product", required=True)
def order(name, realm, seed, product_id):
data = {
**OTPCredentials(name, realm, seed).get_json(),
"product_id": product_id
}
r = requests.post("{}/product/order".format(config('UCLOUD_PAY_SERVER')),
json=data)
print(load_dump_pretty(r.content))