orrisroot
8ce9e09e69
Decouple CLI commands from internal helper logic and consolidate the core file transfer operations in the service layer to improve library portability. - Make MdrsClient subclass MdrsService to inherit resource resolution. - Remove all deprecated helper methods from BaseCommand. - Move core upload and download logic to a new transfer module. - Refactor all CLI commands to route actions through MdrsClient. - Eliminate circular imports between client and CLI command modules.
36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
import getpass
|
|
from argparse import Namespace
|
|
from typing import Any
|
|
|
|
from mdrsclient.api import UsersApi
|
|
from mdrsclient.commands.base import BaseCommand
|
|
from mdrsclient.config import ConfigFile
|
|
from mdrsclient.connection import MDRSConnection
|
|
from mdrsclient.exceptions import MissingConfigurationException
|
|
|
|
|
|
class LoginCommand(BaseCommand):
|
|
@classmethod
|
|
def register(cls, parsers: Any) -> None:
|
|
login_parser = parsers.add_parser("login", help="login to remote host")
|
|
login_parser.add_argument("-u", "--username", help="login username")
|
|
login_parser.add_argument("-p", "--password", help="login password")
|
|
login_parser.add_argument("remote", help="label of remote host")
|
|
login_parser.set_defaults(func=cls.func)
|
|
|
|
@classmethod
|
|
def func(cls, args: Namespace) -> None:
|
|
remote = str(args.remote)
|
|
username = str(args.username) if args.username else input("Username: ").strip()
|
|
password = str(args.password) if args.password else getpass.getpass("Password: ").strip()
|
|
cls.login(remote, username, password)
|
|
|
|
@classmethod
|
|
def login(cls, remote: str, username: str, password: str) -> None:
|
|
from mdrsclient.client import MdrsClient
|
|
|
|
remote_host = MdrsClient.parse_remote_host(remote)
|
|
client = MdrsClient.from_remote(remote_host)
|
|
client.login(username, password)
|
|
print("Login Successful")
|