gistfile1.py

__author__ = 'cristian'

from base import *


def get_hostmame(server_name, server):
    """
        get hostname of hosts
        @type param server_name: str
        @type param server: dict
        :return:
    """
    client = SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    client.connect(server["ip"], username=server["user"],  port=server["port"])

    server["hostname"] = exec_command(client, "hostname", server.ip)


def get_ip(server_name, server):
    """
        get private ip of servers
        @type param server_name: str
        @type param server: dict
        :return:
    """
    client = SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    client.connect(server["ip"], username=server["user"],  port=server["port"])

    server["private_ip"] = exec_command(client, "ifconfig eth1|grep inet|head -1|sed 's/\:/ /'|awk '{print $3}'",
                                        server.ip)

execute = [get_hostmame, get_ip]

if __name__ == "__main__":
    run_task(execute)