Changed required argument to optional in stop() start()

This commit is contained in:
Lulzette 2022-04-07 17:05:19 +03:00
parent d9f951f316
commit 1b4f4b8484

View File

@ -2,13 +2,13 @@
import json import json
import sys import sys
import requests import requests
import typer import typer
import os import os
import configparser import configparser
import base64 import base64
from prettytable import PrettyTable from prettytable import PrettyTable
from typing import Optional
app = typer.Typer() app = typer.Typer()
servers_app = typer.Typer() servers_app = typer.Typer()
@ -67,7 +67,10 @@ def get_balance():
@servers_app.command("start") @servers_app.command("start")
def vds_start(vds_id: int = typer.Argument(...)): def vds_start(vds_id: Optional[int] = typer.Argument(None)):
if vds_id is None:
vds_list()
vds_id = input("Enter VDS ID: ")
result = Server.start(vds_id) result = Server.start(vds_id)
if result is None: if result is None:
print(typer.style("Error", fg=typer.colors.RED)) print(typer.style("Error", fg=typer.colors.RED))
@ -77,7 +80,10 @@ def vds_start(vds_id: int = typer.Argument(...)):
@servers_app.command("stop") @servers_app.command("stop")
def vds_stop(vds_id: int = typer.Argument(...)): def vds_stop(vds_id: Optional[int] = typer.Argument(None)):
if vds_id is None:
vds_list()
vds_id = input("Enter VDS ID: ")
result = Server.stop(vds_id) result = Server.stop(vds_id)
if result is None: if result is None:
print(typer.style("Error", fg=typer.colors.RED)) print(typer.style("Error", fg=typer.colors.RED))