Adjust plugin for Ansible version 2

action_plugins/jiffybox.py

 import json
 import requests
 
-from ansible.callbacks import vv
 from ansible.errors import AnsibleError as ae
-from ansible.runner.return_data import ReturnData
-from ansible.utils import parse_kv
-
-class ActionModule(object):
+from ansible.plugins.action import ActionBase
+try:
+    from __main__ import display
+except ImportError:
+    from ansible.utils.display import Display
+    display = Display()
+
+class ActionModule(ActionBase):
     ''' Manage hosts and infrastructure at JiffyBox'''
 
     ### We need to be able to modify the inventory
     BYPASS_HOST_LOOP = True
     TRANSFERS_FILES = False
 
-    def __init__(self, runner):
-        self.runner = runner
+    def __init__(self, task, connection, play_context, loader, templar,
+                 shared_loader_obj):
+        super(ActionModule, self).__init__(task, connection, play_context,
+                                           loader, templar, shared_loader_obj)
         self.api_token = None
         self.url = None
         self.args = {}
+        self.vars = {}
 
         self.devices = []
         self.backups = []
@@ -28,22 +34,24 @@ class ActionModule(object):
         self.contacts = []
         self.documents = []
 
-    def run(self, conn, tmp, module_name, module_args, inject, complex_args=None, **kwargs):
+    def run(self, tmp=None, task_vars=None):
+        ''' handler for file transfer operations '''
+        if task_vars is None:
+            task_vars = dict()
+        self.vars = task_vars
 
-        if self.runner.noop_on_check(inject):
-            return ReturnData(conn=conn, comm_ok=True, result=dict(skipped=True, msg='check mode not supported for this module'))
+        result = super(ActionModule, self).run(tmp, task_vars)
 
-        if complex_args:
-            self.args.update(complex_args)
-        self.args.update(parse_kv(module_args))
+        if self._play_context.check_mode:
+            result['skipped'] = True
+            result['msg'] = 'check mode not supported for this module'
+            return result
 
-        self.api_token = self._get_var('jiffy_api_token', self.args.get('api_token', False))
+        self.api_token = self._task.args.get('api_token', task_vars.get('jiffy_api_token', False))
         if not self.api_token:
             raise ae("'api_token' is a required argument or you define the variable 'jiffy_api_token' in your inventory.")
 
-        action = self.args.get('action', 'list')
-
-        result = {}
+        action = self._task.args.get('action', 'list')
 
         if action == 'list':
             self.list_devices()
@@ -52,16 +60,7 @@ class ActionModule(object):
         else:
             raise ae("'%s' is an unknown action" % action)
 
-        return ReturnData(conn=conn, comm_ok=True, result=result)
-
-    def _get_var(self, key, default):
-        firsthost = self.runner.host_set[0]
-        variables = self.runner.inventory.get_variables(firsthost)
-        if not variables.has_key(key):
-            self.runner.inventory._vars_per_host[firsthost].__setitem__(key, default)
-            result = default
-        else:
-            result = variables.get(key)
+        result['changed'] = False
         return result
 
     def _request(self, path, data = None, method = 'GET'):
@@ -102,17 +101,12 @@ class ActionModule(object):
         return content['result']
 
     def _load_objects(self, type, path):
-        vv("Reading %s from JiffyBox" % type)
+        display.vv("Reading %s from JiffyBox" % type)
         changed = False
-        allgroup = self.runner.inventory.get_group('all')
-        allvariables = allgroup.get_variables()
-        if not allvariables.has_key('_jiffybox_' + type):
+        if not self.vars.has_key('_jiffybox_' + type):
             changed = True
-            objects = self._request(path)
-            allgroup.set_variable('_jiffybox_' + type, objects)
-        else:
-            objects = allvariables.get('_jiffybox_' + type)
-        return changed, objects
+            self.vars.set('_jiffybox_' + type, self._request(path))
+        return changed, self.vars.get('_jiffybox_' + type)
 
     def load_devices(self):
         if len(self.devices) == 0:
@@ -169,7 +163,7 @@ class ActionModule(object):
             id = str(self.devices[device].get('id'))
             name = self.devices[device].get('name')
             f_ips = []
-            if name in self.runner.host_set:
+            if name in self.vars.get('ansible_play_hosts'):
                 for address in self.devices[device].get('ips').get('public'):
                     ip = self.find_ip(address)
                     if ip:
@@ -198,29 +192,29 @@ class ActionModule(object):
         return ''
 
     def move_ip(self):
-        vv("Moving IP ...")
+        display.vv("Moving IP ...")
         target = self.args.get('target', False)
         if not target:
             raise ae("'target' is a required argument.")
         targethost = self.find_host(target)
         if not targethost:
             raise ae("Target host %s unknown in this JiffyBox account." % target)
-        vv("- target host: %s" % target)
+        display.vv("- target host: %s" % target)
 
         ips = self.find_floating_ip()
         for ip in ips:
             if not ip:
                 raise ae('No ip defined for moving.')
-            vv("- floating IP: %s" % ip.get('ip'))
+            display.vv("- floating IP: %s" % ip.get('ip'))
 
             sourcehost = self.find_host_by_ip(ip)
             if not sourcehost:
-                vv('Current host for the floating ip can not be found.')
+                display.vv('Current host for the floating ip can not be found.')
             else:
-                vv("- source host: %s" % sourcehost.get('name'))
+                display.vv("- source host: %s" % sourcehost.get('name'))
 
                 if sourcehost.get('id') == targethost.get('id'):
-                    vv('- cancelled as source and target are the same')
+                    display.vv('- cancelled as source and target are the same')
                 else:
                     path = 'ips/' + str(sourcehost.get('id')) + '/' + str(ip.get('id')) + '/move'
                     self._request(path, {'targetid': targethost.get('id')})