Adding custom imports on Ansible

Change-Id: I7b2575d4252f040162ac94518bb3844a9b6fd54c
GoogleCloudPlatform · Feb 16, 2018 · 1a53c43 · 1a53c43
1 parent e985d5a
commit 1a53c43
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/products/dns/ansible.yaml b/products/dns/ansible.yaml
@@ -30,6 +30,10 @@ objects: !ruby/object:Api::Resource::HashArray
     skip: true
   ResourceRecordSet:
     version_added: '2.6'
+    imports:
+      - copy
+      - datetime
+      - time
     create: |
       change = create_change(None, updated_record(module), module)
       change_id = int(change['id'])

diff --git a/products/dns/helpers/ansible_provider_resource_set.py.erb b/products/dns/helpers/ansible_provider_resource_set.py.erb
@@ -73,7 +73,6 @@ def update_soa(module):
     original_soa = prefetch_soa_resource(module)
 
     # Create a clone of the SOA record so we can update it
-    import copy
     updated_soa = copy.deepcopy(original_soa)
 
     soa_parts = updated_soa['rrdatas'][0].split(' ')
@@ -112,7 +111,6 @@ def wait_for_change_to_complete(change_id, module):
     while status == 'pending':
         status = get_change_status(change_id, module)
         if status != 'done':
-            import time
             time.sleep(0.5)
 
 
@@ -123,7 +121,6 @@ def get_change_status(change_id, module):
 
 
 def new_change_request():
-    import datetime
     return {
         'kind': 'dns#change',
         'additions': [],

diff --git a/templates/ansible/resource.erb b/templates/ansible/resource.erb
@@ -77,6 +77,11 @@ RETURN = '''
 
 from ansible.module_utils.gcp_utils import navigate_hash, GcpSession, GcpModule, GcpRequestException
 import json
+<% if config['imports'] && !config['imports'].empty? -%>
+<% config['imports'].each do |import| -%>
+<%= lines("import #{import}") -%>
+<% end # config['imports'].each -%>
+<% end # config['imports'] -%>
 
 ################################################################################
 # Main