|
|
|
|
@ -7,6 +7,14 @@ from ansible.inventory import Inventory
|
|
|
|
|
from ansible.runner import Runner
|
|
|
|
|
from ansible.playbook import PlayBook
|
|
|
|
|
|
|
|
|
|
from ansible import callbacks
|
|
|
|
|
from ansible import utils
|
|
|
|
|
from passlib.hash import sha512_crypt
|
|
|
|
|
|
|
|
|
|
import os.path
|
|
|
|
|
JPERM_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
ANSIBLE_DIR = os.path.join(JPERM_DIR, 'playbooks')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AnsibleError(StandardError):
|
|
|
|
|
"""
|
|
|
|
|
@ -38,21 +46,28 @@ class MyInventory(object):
|
|
|
|
|
def __init__(self, resource):
|
|
|
|
|
"""
|
|
|
|
|
resource :
|
|
|
|
|
必须是一个字典列表，比如
|
|
|
|
|
[{"hostname": "10.10.10.10", "port": "22",
|
|
|
|
|
"username": "test", "password": "mypass"}, ...]
|
|
|
|
|
resource的数据格式是一个列表字典，比如
|
|
|
|
|
{
|
|
|
|
|
"group1": [{"hostname": "10.10.10.10", "port": "22",
|
|
|
|
|
"username": "test", "password": "mypass"}, ...],
|
|
|
|
|
"group2": [{"hostname": "10.10.10.10", "port": "22",
|
|
|
|
|
"username": "test", "password": "mypass"}, ...]
|
|
|
|
|
}
|
|
|
|
|
如果你只传入1个列表，这默认该列表内的所有主机属于my_group组,比如
|
|
|
|
|
[{"hostname": "10.10.10.10", "port": "22",
|
|
|
|
|
"username": "test", "password": "mypass"}, ...]
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
self.resource = resource
|
|
|
|
|
self._gen_inventory()
|
|
|
|
|
|
|
|
|
|
self.inventory = Inventory()
|
|
|
|
|
self.gen_inventory()
|
|
|
|
|
|
|
|
|
|
def _gen_inventory(self):
|
|
|
|
|
def add_group(self, hosts, groupname):
|
|
|
|
|
"""
|
|
|
|
|
add hosts to inventory.
|
|
|
|
|
add hosts to a group
|
|
|
|
|
"""
|
|
|
|
|
my_group = Group(name='my_group')
|
|
|
|
|
|
|
|
|
|
for host in self.resource:
|
|
|
|
|
my_group = Group(name=groupname)
|
|
|
|
|
for host in hosts:
|
|
|
|
|
hostname = host.get("hostname")
|
|
|
|
|
hostport = host.get("hostport")
|
|
|
|
|
username = host.get("username")
|
|
|
|
|
@ -64,11 +79,17 @@ class MyInventory(object):
|
|
|
|
|
my_host.set_variable('ansible_ssh_pass', password)
|
|
|
|
|
my_group.add_host(my_host)
|
|
|
|
|
|
|
|
|
|
my_inventory = Inventory()
|
|
|
|
|
my_inventory.add_group(my_group)
|
|
|
|
|
my_inventory.subset('my_group')
|
|
|
|
|
self.inventory.add_group(my_group)
|
|
|
|
|
|
|
|
|
|
self.inventory = my_inventory
|
|
|
|
|
def gen_inventory(self):
|
|
|
|
|
"""
|
|
|
|
|
add hosts to inventory.
|
|
|
|
|
"""
|
|
|
|
|
if isinstance(self.resource, list):
|
|
|
|
|
self.add_group(self.resource, 'my_group')
|
|
|
|
|
elif isinstance(self.resource, dict):
|
|
|
|
|
for groupname, hosts in self.resource.iteritems():
|
|
|
|
|
self.add_group(hosts, groupname)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Command(MyInventory):
|
|
|
|
|
@ -77,7 +98,6 @@ class Command(MyInventory):
|
|
|
|
|
"""
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
|
super(Command, self).__init__(*args, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def run(self, command, module_name="command", timeout=5, forks=10):
|
|
|
|
|
"""
|
|
|
|
|
@ -99,7 +119,6 @@ class Command(MyInventory):
|
|
|
|
|
self.results = hoc.run()
|
|
|
|
|
return self.stdout
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def raw_results(self):
|
|
|
|
|
"""
|
|
|
|
|
@ -107,7 +126,6 @@ class Command(MyInventory):
|
|
|
|
|
"""
|
|
|
|
|
return self.results
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def exec_time(self):
|
|
|
|
|
"""
|
|
|
|
|
@ -121,7 +139,6 @@ class Command(MyInventory):
|
|
|
|
|
"end" : value.get("end"),
|
|
|
|
|
"delta": value.get("delta"),}
|
|
|
|
|
return result
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def stdout(self):
|
|
|
|
|
@ -133,7 +150,6 @@ class Command(MyInventory):
|
|
|
|
|
for key, value in all.iteritems():
|
|
|
|
|
result[key] = value.get("stdout")
|
|
|
|
|
return result
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def stderr(self):
|
|
|
|
|
@ -163,7 +179,6 @@ class Tasks(Command):
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
|
super(Tasks, self).__init__(*args, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def __run(self, module_args, module_name="command", timeout=5, forks=10):
|
|
|
|
|
"""
|
|
|
|
|
run command from andible ad-hoc.
|
|
|
|
|
@ -180,41 +195,81 @@ class Tasks(Command):
|
|
|
|
|
|
|
|
|
|
self.results = hoc.run()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def push_key(self, user, key_path):
|
|
|
|
|
@property
|
|
|
|
|
def msg(self):
|
|
|
|
|
"""
|
|
|
|
|
push the ssh authorized key to target.
|
|
|
|
|
get the contacted and dark msg
|
|
|
|
|
"""
|
|
|
|
|
module_args = 'user="%s" key="{{ lookup("file", "%s") }}"' % (user, key_path)
|
|
|
|
|
self.__run(module_args, "authorized_key")
|
|
|
|
|
|
|
|
|
|
msg = {}
|
|
|
|
|
for result in ["contacted", "dark"]:
|
|
|
|
|
all = self.results.get(result)
|
|
|
|
|
for key, value in all.iteritems():
|
|
|
|
|
if value.get("msg"):
|
|
|
|
|
msg[key] = value.get("msg")
|
|
|
|
|
return msg
|
|
|
|
|
|
|
|
|
|
return {"status": "ok"} if msg else {"status": "failed","msg": msg}
|
|
|
|
|
def push_key(self, user, key_path):
|
|
|
|
|
"""
|
|
|
|
|
push the ssh authorized key to target.
|
|
|
|
|
"""
|
|
|
|
|
module_args = 'user="%s" key="{{ lookup("file", "%s") }}"' % (user, key_path)
|
|
|
|
|
self.__run(module_args, "authorized_key")
|
|
|
|
|
|
|
|
|
|
return {"status": "failed","msg": self.msg} if self.msg else {"status": "ok"}
|
|
|
|
|
|
|
|
|
|
def add_user(self, user):
|
|
|
|
|
def add_user(self, username, password):
|
|
|
|
|
"""
|
|
|
|
|
add a host user.
|
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
encrypt_pass = sha512_crypt.encrypt(password)
|
|
|
|
|
module_args = 'name=%s shell=/bin/bash password=%s' % (username, encrypt_pass)
|
|
|
|
|
self.__run(module_args, "user")
|
|
|
|
|
|
|
|
|
|
return {"status": "failed","msg": self.msg} if self.msg else {"status": "ok"}
|
|
|
|
|
|
|
|
|
|
def del_user(self, user):
|
|
|
|
|
def del_user(self, username):
|
|
|
|
|
"""
|
|
|
|
|
delete a host user.
|
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
module_args = 'name=%s state=absent remove=yes move_home=yes force=yes' % (username)
|
|
|
|
|
self.__run(module_args, "user")
|
|
|
|
|
|
|
|
|
|
return {"status": "failed","msg": self.msg} if self.msg else {"status": "ok"}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class CustomAggregateStats(callbacks.AggregateStats):
|
|
|
|
|
"""
|
|
|
|
|
Holds stats about per-host activity during playbook runs.
|
|
|
|
|
"""
|
|
|
|
|
def __init__(self):
|
|
|
|
|
super(CustomAggregateStats, self).__init__()
|
|
|
|
|
self.results = []
|
|
|
|
|
|
|
|
|
|
def compute(self, runner_results, setup=False, poll=False,
|
|
|
|
|
ignore_errors=False):
|
|
|
|
|
"""
|
|
|
|
|
Walk through all results and increment stats.
|
|
|
|
|
"""
|
|
|
|
|
super(CustomAggregateStats, self).compute(runner_results, setup, poll,
|
|
|
|
|
ignore_errors)
|
|
|
|
|
|
|
|
|
|
self.results.append(runner_results)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def summarize(self, host):
|
|
|
|
|
"""
|
|
|
|
|
Return information about a particular host
|
|
|
|
|
"""
|
|
|
|
|
summarized_info = super(CustomAggregateStats, self).summarize(host)
|
|
|
|
|
|
|
|
|
|
# Adding the info I need
|
|
|
|
|
summarized_info['result'] = self.results
|
|
|
|
|
|
|
|
|
|
return summarized_info
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MyPlaybook(MyInventory):
|
|
|
|
|
"""
|
|
|
|
|
this is my playbook object for execute playbook.
|
|
|
|
|
@ -223,11 +278,32 @@ class MyPlaybook(MyInventory):
|
|
|
|
|
super(MyPlaybook, self).__init__(*args, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def deploy(self):
|
|
|
|
|
def run(self, playbook_relational_path):
|
|
|
|
|
"""
|
|
|
|
|
run ansible playbook,
|
|
|
|
|
only surport relational path.
|
|
|
|
|
"""
|
|
|
|
|
stats = CustomAggregateStats()
|
|
|
|
|
playbook_cb = callbacks.PlaybookCallbacks(verbose=utils.VERBOSITY)
|
|
|
|
|
runner_cb = callbacks.PlaybookRunnerCallbacks(stats, verbose=utils.VERBOSITY)
|
|
|
|
|
playbook_path = os.path.join(ANSIBLE_DIR, playbook_relational_path)
|
|
|
|
|
|
|
|
|
|
pb = PlayBook(
|
|
|
|
|
playbook = playbook_path,
|
|
|
|
|
stats = stats,
|
|
|
|
|
callbacks = playbook_cb,
|
|
|
|
|
runner_callbacks = runner_cb,
|
|
|
|
|
inventory = self.inventory,
|
|
|
|
|
check=True)
|
|
|
|
|
|
|
|
|
|
self.results = pb.run()
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def raw_results(self):
|
|
|
|
|
"""
|
|
|
|
|
use ansible playbook to deploy a application.
|
|
|
|
|
get the raw results after playbook run.
|
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
return self.results
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class App(MyPlaybook):
|
|
|
|
|
@ -240,15 +316,13 @@ class App(MyPlaybook):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
resource = [ {"hostname": "192.168.10.128", "port": "22", "username": "root", "password": "xxx"}]
|
|
|
|
|
task = Tasks(resource)
|
|
|
|
|
print task.push_key('root', '/root/.ssh/id_rsa.pub')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
resource = {"test": [{"hostname": "192.168.10.128", "port": "22", "username": "root", "password": "xxx"}]}
|
|
|
|
|
playbook = MyPlaybook(resource)
|
|
|
|
|
playbook.run('test.yml')
|
|
|
|
|
print playbook.raw_results
|
|
|
|
|
# print task.add_user('test', 'mypass')
|
|
|
|
|
# print task.del_user('test')
|
|
|
|
|
# print task.push_key('root', '/root/.ssh/id_rsa.pub')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Zi Chuanxiu
9 years ago