import subprocess

from ansible.plugins.connection.ssh import Connection as ConnectionSSH
from ansible.errors import AnsibleError
from socket import create_connection
from time import sleep

try:
    from __main__ import display
except ImportError:
    from ansible.utils.display import Display
    display = Display()

class Connection(ConnectionSSH):

    def __init__(self, *args, **kwargs):

        super(Connection, self).__init__(*args, **kwargs)
        display.vvv("ssh_fwknop connection plugin is used for this host", host=self.host)

    def set_host_overrides(self, host, hostvars=None):

        p = subprocess.Popen(['/usr/bin/fwknop', '-n', '%s' % (host)], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        stdout, stderr = p.communicate()
        status_code = p.wait()
        if status_code != 0:
            raise AnsibleError("fwknop error:\n%s" % stderr)