Parses Connection Strings like "protocol://[host[:password]]@host[:port]" - supports Public Key passing instead of Password

conn_string_parser.py

def parse_conn_string(conn_string):
'''
  Parses string like:

  smb://username@host:445
  sftp://host:22
  ssh://host:22
  sftp://username#pub_key_path@host:22
  sftp://username:password@host:22
  sftp://username:[email protected]:22  

*Used in satori-remote:
https://github.com/satori-ng/satori-remote/blob/master/satoriremote/__init__.py
'''
    arg_regex = r'(\w*)://((\w*)(([:#])(.+))?@)?([\.\w]*)(\:(\d+))?'
    m = re.match(arg_regex, conn_string)

    try:
        conn_dict={}
        conn_dict['protocol'] = m.group(1)
        conn_dict['username'] = m.group(3)
        conn_dict['auth_type'] = "key" if m.group(5) == '#' else "passwd"
        conn_dict['auth'] = m.group(6)
        conn_dict['host'] =  m.group(7)
        try:
            conn_dict['port'] = int(m.group(9))
        except:       # if :port is not defined
            conn_dict['port'] = None
            pass

        if conn_dict['username'] is None:
            conn_dict['username'] = raw_input("Username:")

        if conn_dict['auth'] is None:
            import getpass
            conn_dict['auth'] = getpass.getpass("Password:")
        return conn_dict
    except AttributeError:
        raise ValueError("'{}' is not a valid remote argument".format(conn_string))

sftp://username:[email protected]:22

{'protocol': 'sftp', 'username': 'username', 'auth_type': 'passwd', 'auth': 'password', 'host': '0.0.0.0', 'port': 22}

sftp://username#path/to/[email protected]:22

{'protocol': 'sftp', 'username': 'username', 'auth_type': 'key', 'auth': 'path/to/key', 'host': '0.0.0.0', 'port': 22}

Test it out on:
Regex101.com.