Refactored code to be more concise
This commit is contained in:
parent
d79a01cfb1
commit
58aea47921
13 changed files with 64 additions and 46 deletions
|
@ -1,7 +0,0 @@
|
|||
from ...models import Command
|
||||
from .descriptors import mapping
|
||||
|
||||
commands: dict[str, Command] = {}
|
||||
|
||||
for descriptor in mapping:
|
||||
commands.update({ descriptor.cmd: descriptor })
|
4
dockge_cli/client/commands/__init__.py
Normal file
4
dockge_cli/client/commands/__init__.py
Normal file
|
@ -0,0 +1,4 @@
|
|||
from ...models import Command
|
||||
from .mappings import mapping
|
||||
|
||||
commands: dict[str, Command] = { c.cmd: c for c in mapping }
|
|
@ -4,15 +4,18 @@ from getpass import getpass
|
|||
from ...models import Credentials
|
||||
from ...service import storage
|
||||
from ...service.communicate import DockgeConnection
|
||||
from ..utils import stack_formatter, status_formatter, generic_formatter, get_credential_parser
|
||||
from ..utils import stack_formatter, status_formatter, generic_formatter, credential_parser_factory
|
||||
|
||||
class ExecutionCommands():
|
||||
class FunctionBindings():
|
||||
"""
|
||||
Helper class that provides all the static methods in an organized way
|
||||
This is an abstraction layer of the CLI, as those functions only do little preprocessing before calling the actural DockgeConnection
|
||||
"""
|
||||
@staticmethod
|
||||
def __setup():
|
||||
"""
|
||||
Creates a connection and logs into Dockge
|
||||
"""
|
||||
con = DockgeConnection()
|
||||
con.connect_and_login()
|
||||
return con
|
||||
|
@ -37,7 +40,7 @@ class ExecutionCommands():
|
|||
login command binding
|
||||
"""
|
||||
if len(extra_args) > 0:
|
||||
credentials = get_credential_parser().parse_args(extra_args, namespace=Credentials)
|
||||
credentials = credential_parser_factory().parse_args(extra_args, namespace=Credentials)
|
||||
storage.put("username", credentials.username, encoded=True)
|
||||
storage.put("password", credentials.password, encoded=True)
|
||||
return
|
||||
|
@ -64,7 +67,7 @@ class ExecutionCommands():
|
|||
"""
|
||||
list command binding
|
||||
"""
|
||||
con = ExecutionCommands.__setup()
|
||||
con = FunctionBindings.__setup()
|
||||
stack_formatter(con.list_stacks())
|
||||
con.disconnect()
|
||||
|
||||
|
@ -75,7 +78,7 @@ class ExecutionCommands():
|
|||
"""
|
||||
if extra_args is None:
|
||||
raise ValueError
|
||||
con = ExecutionCommands.__setup()
|
||||
con = FunctionBindings.__setup()
|
||||
status_formatter(con.list_stack(extra_args[0]))
|
||||
con.disconnect()
|
||||
|
||||
|
@ -86,7 +89,7 @@ class ExecutionCommands():
|
|||
"""
|
||||
if extra_args is None:
|
||||
raise ValueError
|
||||
con = ExecutionCommands.__setup()
|
||||
con = FunctionBindings.__setup()
|
||||
generic_formatter(con.restart(extra_args[0]))
|
||||
con.disconnect()
|
||||
|
||||
|
@ -97,7 +100,7 @@ class ExecutionCommands():
|
|||
"""
|
||||
if extra_args is None:
|
||||
raise ValueError
|
||||
con = ExecutionCommands.__setup()
|
||||
con = FunctionBindings.__setup()
|
||||
generic_formatter(con.update(extra_args[0]))
|
||||
con.disconnect()
|
||||
|
||||
|
@ -108,7 +111,7 @@ class ExecutionCommands():
|
|||
"""
|
||||
if extra_args is None:
|
||||
raise ValueError
|
||||
con = ExecutionCommands.__setup()
|
||||
con = FunctionBindings.__setup()
|
||||
generic_formatter(con.stop(extra_args[0]))
|
||||
con.disconnect()
|
||||
|
||||
|
@ -119,7 +122,7 @@ class ExecutionCommands():
|
|||
"""
|
||||
if extra_args is None:
|
||||
raise ValueError
|
||||
con = ExecutionCommands.__setup()
|
||||
con = FunctionBindings.__setup()
|
||||
generic_formatter(con.start(extra_args[0]))
|
||||
con.disconnect()
|
||||
|
||||
|
@ -130,7 +133,7 @@ class ExecutionCommands():
|
|||
"""
|
||||
if extra_args is None:
|
||||
raise ValueError
|
||||
con = ExecutionCommands.__setup()
|
||||
con = FunctionBindings.__setup()
|
||||
generic_formatter(con.down(extra_args[0]))
|
||||
con.disconnect()
|
||||
|
|
@ -1,6 +1,6 @@
|
|||
from typing import List
|
||||
from ...models import Command
|
||||
from .bindings import ExecutionCommands
|
||||
from .functions import FunctionBindings
|
||||
|
||||
mapping: List[Command] = [
|
||||
Command(
|
||||
|
@ -8,83 +8,83 @@ mapping: List[Command] = [
|
|||
description="Sets and gets the URI of the dockge instance. Remove any unnecessary subdomains/protocols from the URI",
|
||||
args=1,
|
||||
optional=True,
|
||||
bind=ExecutionCommands.host
|
||||
func=FunctionBindings.host
|
||||
),
|
||||
Command(
|
||||
cmd="login",
|
||||
description="Logs into a given dockge account, either with an interactive dialogue or by passing --user and --password",
|
||||
args=2,
|
||||
optional=True,
|
||||
bind=ExecutionCommands.login
|
||||
func=FunctionBindings.login
|
||||
),
|
||||
Command(
|
||||
cmd="logout",
|
||||
description="Removes the credentials from the local storage.",
|
||||
args=0,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.logout
|
||||
func=FunctionBindings.logout
|
||||
),
|
||||
Command(
|
||||
cmd="list",
|
||||
description="Lists all available stacks with their status",
|
||||
args=0,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.list
|
||||
func=FunctionBindings.list
|
||||
),
|
||||
Command(
|
||||
cmd="status",
|
||||
description="Returns the status of one stack",
|
||||
args=1,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.status
|
||||
func=FunctionBindings.status
|
||||
),
|
||||
Command(
|
||||
cmd="restart",
|
||||
description="Restarts a given stack",
|
||||
args=1,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.restart
|
||||
func=FunctionBindings.restart
|
||||
),
|
||||
Command(
|
||||
cmd="start",
|
||||
description="Starts a given stack",
|
||||
args=1,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.start
|
||||
func=FunctionBindings.start
|
||||
),
|
||||
Command(
|
||||
cmd="stop",
|
||||
description="Stops a given stack",
|
||||
args=1,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.stop
|
||||
func=FunctionBindings.stop
|
||||
),
|
||||
Command(
|
||||
cmd="down",
|
||||
description="Stop & Downs a given stack",
|
||||
args=1,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.down
|
||||
func=FunctionBindings.down
|
||||
),
|
||||
Command(
|
||||
cmd="update",
|
||||
description="Updates a stack",
|
||||
args=1,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.update
|
||||
func=FunctionBindings.update
|
||||
),
|
||||
Command(
|
||||
cmd="exit",
|
||||
description="Exits the CLI - this will reset all settings, including credentials and host",
|
||||
args=0,
|
||||
optional=False,
|
||||
bind=ExecutionCommands.exit
|
||||
func=FunctionBindings.exit
|
||||
),
|
||||
Command(
|
||||
cmd="help",
|
||||
description="Displays helping hints for commands",
|
||||
args=1,
|
||||
optional=True,
|
||||
bind=ExecutionCommands.help
|
||||
func=FunctionBindings.help
|
||||
)
|
||||
]
|
|
@ -3,7 +3,7 @@ import sys
|
|||
|
||||
from .. import __version__
|
||||
from ..models import Arguments
|
||||
from .commandprovider.factory import commands
|
||||
from .commands import commands
|
||||
|
||||
def parse_arguments():
|
||||
"""
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
from .commandprovider.factory import commands
|
||||
from .commands import commands
|
||||
|
||||
def display_help(extra_args):
|
||||
"""
|
||||
|
@ -32,4 +32,4 @@ def run(command, args):
|
|||
display_help(args)
|
||||
return
|
||||
|
||||
c.bind(args)
|
||||
c.func(args)
|
||||
|
|
|
@ -2,7 +2,7 @@ import argparse
|
|||
from tabulate import tabulate
|
||||
from ..models import StackStatus
|
||||
|
||||
def get_credential_parser():
|
||||
def credential_parser_factory():
|
||||
"""
|
||||
Creates a new parser for login credentials
|
||||
"""
|
||||
|
@ -23,7 +23,7 @@ def stack_formatter(stacks):
|
|||
|
||||
table, headers = [], ["Stackname", "Status"]
|
||||
for key, val in stacks["stackList"].items():
|
||||
table.append([key, StackStatus(val['status']).name])
|
||||
table.append([key, StackStatus(val['status']).name.lower()])
|
||||
|
||||
print(tabulate(table, headers=headers, tablefmt="github"), "\n")
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue