Gecko [ theprintave.ca ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2024-02-16 18:48
backends	[ DIR ]	drwxr-xr-x	2024-02-16 18:53
testing	[ DIR ]	drwxr-xr-x	2024-02-16 18:53
util	[ DIR ]	drwxr-xr-x	2024-02-16 18:53
__init__.py	271 B	-rw-r--r--	2022-01-02 00:29
__main__.py	71 B	-rw-r--r--	2022-01-02 00:29
backend.py	6.23 KB	-rw-r--r--	2022-01-02 00:29
cli.py	3.83 KB	-rw-r--r--	2022-01-02 00:29
core.py	5.09 KB	-rw-r--r--	2022-01-02 00:29
credentials.py	1.56 KB	-rw-r--r--	2022-01-02 00:29
devpi_client.py	199 B	-rw-r--r--	2022-01-02 00:29
errors.py	1.4 KB	-rw-r--r--	2022-01-02 00:29
http.py	1.2 KB	-rw-r--r--	2022-01-02 00:29

Rename

#!/usr/bin/env python
"""Simple command line interface to get/set password from a keyring"""

import getpass
import argparse
import sys

from . import core
from . import backend
from . import set_keyring, get_password, set_password, delete_password

class CommandLineTool:
    def __init__(self):
        self.parser = argparse.ArgumentParser()
        self.parser.add_argument(
            "-p",
            "--keyring-path",
            dest="keyring_path",
            default=None,
            help="Path to the keyring backend",
        )
        self.parser.add_argument(
            "-b",
            "--keyring-backend",
            dest="keyring_backend",
            default=None,
            help="Name of the keyring backend",
        )
        self.parser.add_argument(
            "--list-backends",
            action="store_true",
            help="List keyring backends and exit",
        )
        self.parser.add_argument(
            "--disable", action="store_true", help="Disable keyring and exit"
        )
        self.parser.add_argument(
            'operation',
            help="get|set|del",
            nargs="?",
        )
        self.parser.add_argument(
            'service',
            nargs="?",
        )
        self.parser.add_argument(
            'username',
            nargs="?",
        )

def run(self, argv):
        args = self.parser.parse_args(argv)
        vars(self).update(vars(args))

if args.list_backends:
            for k in backend.get_all_keyring():
                print(k)
            return

if args.disable:
            core.disable()
            return

self._check_args()
        self._load_spec_backend()
        method = getattr(self, f'do_{self.operation}', self.invalid_op)
        return method()

def _check_args(self):
        if self.operation:
            if self.service is None or self.username is None:
                self.parser.error(f"{self.operation} requires service and username")

def do_get(self):
        password = get_password(self.service, self.username)
        if password is None:
            raise SystemExit(1)
        print(password)

def do_set(self):
        password = self.input_password(
            f"Password for '{self.username}' in '{self.service}': "
        )
        set_password(self.service, self.username, password)

def do_del(self):
        delete_password(self.service, self.username)

def invalid_op(self):
        self.parser.error("Specify operation 'get', 'del', or 'set'.")

def _load_spec_backend(self):
        if self.keyring_backend is None:
            return

try:
            if self.keyring_path:
                sys.path.insert(0, self.keyring_path)
            set_keyring(core.load_keyring(self.keyring_backend))
        except (Exception,) as exc:
            # Tons of things can go wrong here:
            #   ImportError when using "fjkljfljkl"
            #   AttributeError when using "os.path.bar"
            #   TypeError when using "__builtins__.str"
            # So, we play on the safe side, and catch everything.
            self.parser.error(f"Unable to load specified keyring: {exc}")

def input_password(self, prompt):
        """Retrieve password from input."""
        return self.pass_from_pipe() or getpass.getpass(prompt)

@classmethod
    def pass_from_pipe(cls):
        """Return password from pipe if not on TTY, else False."""
        is_pipe = not sys.stdin.isatty()
        return is_pipe and cls.strip_last_newline(sys.stdin.read())

@staticmethod
    def strip_last_newline(str):
        """Strip one last newline, if present."""
        return str[: -str.endswith('\n')]

def main(argv=None):
    """Main command line interface."""

if argv is None:
        argv = sys.argv[1:]

cli = CommandLineTool()
    return cli.run(argv)

if __name__ == '__main__':
    sys.exit(main())