55 lines
2 KiB
Python
Executable file
55 lines
2 KiB
Python
Executable file
import click
|
|
import requests
|
|
|
|
from decouple import config
|
|
from .helper import OTPCredentials, load_dump_pretty
|
|
|
|
|
|
def product_command(command, otp):
|
|
data = {**otp.get_json(), "action": command}
|
|
r = requests.post(f"{config('UCLOUD_PAY_SERVER')}/product/action",
|
|
json=data)
|
|
return r
|
|
|
|
|
|
@click.group()
|
|
def product():
|
|
pass
|
|
|
|
|
|
@product.command("list")
|
|
def list():
|
|
r = requests.get(f"{config('UCLOUD_PAY_SERVER')}/product/list")
|
|
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-duration", required=False,
|
|
help="Only used for products with recurring costs",
|
|
type=int)
|
|
@click.option("--product-recurring-duration-units", required=False, type=int,
|
|
help="Unit of the recurring costs", default=0)
|
|
def add(name, realm, seed, product_name, product_description, product_type,
|
|
product_price, product_recurring_duration,
|
|
product_recurring_duration_units):
|
|
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_duration": product_recurring_duration,
|
|
"product_recurring_duration_units": product_recurring_duration_units
|
|
}
|
|
r = requests.post(f"{config('UCLOUD_API_SERVER')}/product/add", json=data)
|
|
print(load_dump_pretty(r.content))
|