Antoine Legrand
9 years ago
3 changed files with 4 additions and 309 deletions
Split View
Diff Options
-
3.gitmodules
-
1library
-
309library/kube.py
@ -0,0 +1,3 @@ |
|||
[submodule "library"] |
|||
path = library |
|||
url = git@github.com:ansibl8s/k8s-ansible-module.git |
|||
@ -1,309 +0,0 @@ |
|||
#!/usr/bin/python |
|||
# -*- coding: utf-8 -*- |
|||
|
|||
DOCUMENTATION = """ |
|||
--- |
|||
module: kube |
|||
short_description: Manage Kubernetes Cluster |
|||
description: |
|||
- Create, replace, remove, and stop resources within a Kubernetes Cluster |
|||
version_added: "2.0" |
|||
options: |
|||
name: |
|||
required: false |
|||
default: null |
|||
description: |
|||
- The name associated with resource |
|||
filename: |
|||
required: false |
|||
default: null |
|||
description: |
|||
- The path and filename of the resource(s) definition file. |
|||
namespace: |
|||
required: false |
|||
default: null |
|||
description: |
|||
- The namespace associated with the resource(s) |
|||
resource: |
|||
required: false |
|||
default: null |
|||
description: |
|||
- The resource to perform an action on. pods (po), replicationControllers (rc), services (svc) |
|||
label: |
|||
required: false |
|||
default: null |
|||
description: |
|||
- The labels used to filter specific resources. |
|||
server: |
|||
required: false |
|||
default: null |
|||
description: |
|||
- The url for the API server that commands are executed against. |
|||
api_version: |
|||
required: false |
|||
choices: ['v1', 'v1beta3'] |
|||
default: v1 |
|||
description: |
|||
- The API version associated with cluster. |
|||
force: |
|||
required: false |
|||
default: false |
|||
description: |
|||
- A flag to indicate to force delete, replace, or stop. |
|||
all: |
|||
required: false |
|||
default: false |
|||
description: |
|||
- A flag to indicate delete all, stop all, or all namespaces when checking exists. |
|||
log_level: |
|||
required: false |
|||
default: 0 |
|||
description: |
|||
- Indicates the level of verbosity of logging by kubectl. |
|||
state: |
|||
required: false |
|||
choices: ['present', 'absent', 'latest', 'reloaded', 'stopped'] |
|||
default: present |
|||
description: |
|||
- present handles checking existence or creating if definition file provided, |
|||
absent handles deleting resource(s) based on other options, |
|||
latest handles creating ore updating based on existence, |
|||
reloaded handles updating resource(s) definition using definition file, |
|||
stopped handles stopping resource(s) based on other options. |
|||
requirements: |
|||
- kubectl |
|||
author: "Kenny Jones (@kenjones-cisco)" |
|||
""" |
|||
|
|||
EXAMPLES = """ |
|||
- name: test nginx is present |
|||
kube: name=nginx resource=rc state=present |
|||
|
|||
- name: test nginx is stopped |
|||
kube: name=nginx resource=rc state=stopped |
|||
|
|||
- name: test nginx is absent |
|||
kube: name=nginx resource=rc state=absent |
|||
|
|||
- name: test nginx is present |
|||
kube: filename=/tmp/nginx.yml |
|||
""" |
|||
|
|||
|
|||
class KubeManager(object): |
|||
|
|||
def __init__(self, module): |
|||
|
|||
self.module = module |
|||
|
|||
self.base_cmd = [module.get_bin_path('kubectl', True)] |
|||
self.api_version = module.params.get('api_version') |
|||
|
|||
if self.api_version: |
|||
self.base_cmd.append('--api-version=' + self.api_version) |
|||
|
|||
if module.params.get('server'): |
|||
self.base_cmd.append('--server=' + module.params.get('server')) |
|||
|
|||
if module.params.get('log_level'): |
|||
self.base_cmd.append('--v=' + str(module.params.get('log_level'))) |
|||
|
|||
if module.params.get('namespace'): |
|||
self.base_cmd.append('--namespace=' + module.params.get('namespace')) |
|||
|
|||
self.all = module.params.get('all') |
|||
self.force = module.params.get('force') |
|||
self.name = module.params.get('name') |
|||
self.filename = module.params.get('filename') |
|||
self.resource = module.params.get('resource') |
|||
self.label = module.params.get('label') |
|||
|
|||
def _execute(self, cmd): |
|||
args = self.base_cmd + cmd |
|||
try: |
|||
rc, out, err = self.module.run_command(args) |
|||
if rc != 0: |
|||
self.module.fail_json( |
|||
msg='error running kubectl (%s) command (rc=%d): %s' % (' '.join(args), rc, out or err)) |
|||
except Exception as exc: |
|||
self.module.fail_json( |
|||
msg='error running kubectl (%s) command: %s' % (' '.join(args), str(exc))) |
|||
return out.splitlines() |
|||
|
|||
def _execute_nofail(self, cmd): |
|||
args = self.base_cmd + cmd |
|||
rc, out, err = self.module.run_command(args) |
|||
if rc != 0: |
|||
return None |
|||
return out.splitlines() |
|||
|
|||
def create(self, check=True): |
|||
if check and self.exists(): |
|||
return [] |
|||
|
|||
cmd = ['create'] |
|||
|
|||
if not self.filename: |
|||
self.module.fail_json(msg='filename required to create') |
|||
|
|||
cmd.append('--filename=' + self.filename) |
|||
|
|||
return self._execute(cmd) |
|||
|
|||
def replace(self): |
|||
|
|||
if not self.force and not self.exists(): |
|||
return [] |
|||
|
|||
cmd = ['replace'] |
|||
if self.api_version != 'v1': |
|||
cmd = ['update'] |
|||
|
|||
if self.force: |
|||
cmd.append('--force') |
|||
|
|||
if not self.filename: |
|||
self.module.fail_json(msg='filename required to reload') |
|||
|
|||
cmd.append('--filename=' + self.filename) |
|||
|
|||
return self._execute(cmd) |
|||
|
|||
def delete(self): |
|||
|
|||
if not self.force and not self.exists(): |
|||
return [] |
|||
|
|||
cmd = ['delete'] |
|||
|
|||
if self.filename: |
|||
cmd.append('--filename=' + self.filename) |
|||
else: |
|||
if not self.resource: |
|||
self.module.fail_json(msg='resource required to delete without filename') |
|||
|
|||
cmd.append(self.resource) |
|||
|
|||
if self.name: |
|||
cmd.append(self.name) |
|||
|
|||
if self.label: |
|||
cmd.append('--selector=' + self.label) |
|||
|
|||
if self.all: |
|||
cmd.append('--all') |
|||
|
|||
if self.force: |
|||
cmd.append('--ignore-not-found') |
|||
|
|||
return self._execute(cmd) |
|||
|
|||
def exists(self): |
|||
cmd = ['get'] |
|||
|
|||
if not self.resource: |
|||
return False |
|||
|
|||
cmd.append(self.resource) |
|||
|
|||
if self.name: |
|||
cmd.append(self.name) |
|||
|
|||
cmd.append('--no-headers') |
|||
|
|||
if self.label: |
|||
cmd.append('--selector=' + self.label) |
|||
|
|||
if self.all: |
|||
cmd.append('--all-namespaces') |
|||
|
|||
result = self._execute_nofail(cmd) |
|||
if not result: |
|||
return False |
|||
return True |
|||
|
|||
def stop(self): |
|||
|
|||
if not self.force and not self.exists(): |
|||
return [] |
|||
|
|||
cmd = ['stop'] |
|||
|
|||
if self.filename: |
|||
cmd.append('--filename=' + self.filename) |
|||
else: |
|||
if not self.resource: |
|||
self.module.fail_json(msg='resource required to stop without filename') |
|||
|
|||
cmd.append(self.resource) |
|||
|
|||
if self.name: |
|||
cmd.append(self.name) |
|||
|
|||
if self.label: |
|||
cmd.append('--selector=' + self.label) |
|||
|
|||
if self.all: |
|||
cmd.append('--all') |
|||
|
|||
if self.force: |
|||
cmd.append('--ignore-not-found') |
|||
|
|||
return self._execute(cmd) |
|||
|
|||
|
|||
def main(): |
|||
|
|||
module = AnsibleModule( |
|||
argument_spec=dict( |
|||
name=dict(), |
|||
filename=dict(), |
|||
namespace=dict(), |
|||
resource=dict(), |
|||
label=dict(), |
|||
server=dict(), |
|||
api_version=dict(default='v1', choices=['v1', 'v1beta3']), |
|||
force=dict(default=False, type='bool'), |
|||
all=dict(default=False, type='bool'), |
|||
log_level=dict(default=0, type='int'), |
|||
state=dict(default='present', choices=['present', 'absent', 'latest', 'reloaded', 'stopped']), |
|||
) |
|||
) |
|||
|
|||
changed = False |
|||
|
|||
manager = KubeManager(module) |
|||
state = module.params.get('state') |
|||
|
|||
if state == 'present': |
|||
result = manager.create() |
|||
|
|||
elif state == 'absent': |
|||
result = manager.delete() |
|||
|
|||
elif state == 'reloaded': |
|||
result = manager.replace() |
|||
|
|||
elif state == 'stopped': |
|||
result = manager.stop() |
|||
|
|||
elif state == 'latest': |
|||
if manager.exists(): |
|||
manager.force = True |
|||
result = manager.replace() |
|||
else: |
|||
result = manager.create(check=False) |
|||
|
|||
else: |
|||
module.fail_json(msg='Unrecognized state %s.' % state) |
|||
|
|||
if result: |
|||
changed = True |
|||
module.exit_json(changed=changed, |
|||
msg='success: %s' % (' '.join(result)) |
|||
) |
|||
|
|||
|
|||
from ansible.module_utils.basic import * # noqa |
|||
if __name__ == '__main__': |
|||
main() |
|||
Write
Preview
Loading…
Cancel
Save