• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• margaritashotgun-master
  • README.rst
  • setup.py
  • Vagrantfile
  • margaritashotgun
    • repository.py
    • exceptions.py
    • cli.py
    • .keep
    • client.py
    • remote_host.py
    • ssh_tunnel.py
    • util
      • __init__.py
      • parser.py
    • workers.py
    • memory.py
    • logger.py
    • __init__.py
    • settings.py
    • _version.py
    • auth.py
    • remote_shell.py
  • setup.cfg
  • .travis.yml
  • tests
    • .keep
    • files
      • validate_passing.yml
      • yml_failing.yml
      • encrypted.key.pub
      • encrypted.key
      • validate_failing_host.yml
      • rsa.key
      • validate_failing_aws.yml
      • validate_failing_root.yml
      • validate_failing_hosts.yml
      • yml_passing.yml
      • rsa.key.pub
      • validate_failing_logging.yml
    • __init__.py
    • unit
      • test_memory.py
      • test_exceptions.py
      • test_module.py
      • test_util_parser.py
      • test_remote_shell.py
      • test_client.py
      • test_cli.py
      • test_auth.py
      • __init__.py
  • bin
    • .keep
    • margaritashotgun
  • requirements.txt
  • .gitignore
  • docs
    • reference_guide.rst
    • architecture.rst
    • user_guide.rst
    • Makefile
    • .keep
    • about.rst
    • installing.rst
    • quickstart.rst
    • index.rst
    • conf.py
    • _static
    • development.rst
  • LICENSE.txt
  • conf
    • logging_config.yml.example
    • parallel_config.yml.example
    • s3_config.yml.example
    • file_config.yml.example
    • jump_host.yml.example

from enum import Enum
import paramiko
from paramiko import PasswordRequiredException
from margaritashotgun.exceptions import AuthenticationMissingUsernameError
from margaritashotgun.exceptions import AuthenticationMethodMissingError


class AuthMethods(Enum):
    key = 'key'
    password = 'password'

class Auth():

    def __init__(self, username=None, password=None, key=None):
        """
        :type username: str
        :param username: username for ssh authentication
        :type password: str
        :param password: password for ssh authentication
        :type key: str
        :param key: path to rsa key for ssh authentication
        """
        self.method = None
        self.username = None
        self.password = None
        self.key = None

        if username is None or username == "":
            raise AuthenticationMissingUsernameError()
        else:
            self.username = username

        if key is not None:
            self.key = self.load_key(key, password)
            self.method = AuthMethods.key
        elif password is not None:
            self.password = password
            self.method = AuthMethods.password
        else:
            raise AuthenticationMethodMissingError()

    def load_key(self, key_path, password):
        """
        Creates paramiko rsa key

        :type key_path: str
        :param key_path: path to rsa key
        :type password: str
        :param password: password to try if rsa key is encrypted
        """

        try:
            return paramiko.RSAKey.from_private_key_file(key_path)
        except PasswordRequiredException as ex:
            return paramiko.RSAKey.from_private_key_file(key_path,
                                                         password=password)