3 * ============LICENSE_START=======================================================
5 * ================================================================================
6 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
7 * ================================================================================
8 * Copyright (C) 2017 Amdocs
9 * =============================================================================
10 * Licensed under the Apache License, Version 2.0 (the "License");
11 * you may not use this file except in compliance with the License.
12 * You may obtain a copy of the License at
14 * http://www.apache.org/licenses/LICENSE-2.0
16 * Unless required by applicable law or agreed to in writing, software
17 * distributed under the License is distributed on an "AS IS" BASIS,
18 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
19 * See the License for the specific language governing permissions and
20 * limitations under the License.
22 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
23 * ============LICENSE_END=========================================================
28 from os import listdir
29 from os.path import isfile, join
33 def __init__(self, myhost, myuser, mypasswd, mydb):
38 self.db = pymysql.connect(host=myhost,
42 self.cur = self.db.cursor()
43 except Exception as e:
47 def Query (self, myquery, val = None):
51 self.cur.execute(myquery, val)
53 self.cur.execute(myquery)
55 results = self.cur.fetchall()
64 def loadPlaybook (sqlintf, value, version, ext = '.yml'):
69 # Test if primary key already defined
70 query = "SELECT name FROM playbook WHERE name='" + value +"'"
71 results = sqlintf.Query (query)
75 query = "INSERT INTO playbook (name) VALUES ('" + value + "')"
76 results = sqlintf.Query (query)
82 file = open(playbook_path + value + ext, 'r')
83 load_file = file.read()
86 sql = "UPDATE playbook SET value=%s, version=%s, type=%s WHERE name=%s"
88 results = sqlintf.Query(sql, (load_file, version, ext, value))
91 # Error loading playbook
95 return errorCode, diag
97 def loadCredentials (sqlintf, hostgroup, hostname, cred):
103 query = "SELECT hostname,hostgroup FROM inventory WHERE hostname='" + hostname +"'"
104 results = sqlintf.Query (query)
106 if hostname in str (results):
108 results_hostgroups = results[0][1]
110 if hostgroup in results_hostgroups.split(','):
111 query = "UPDATE inventory SET hostname='" + hostname + "',credentials='" +\
113 "' WHERE hostname='" + hostname + "'"
116 results_hostgroups = results_hostgroups + ',' + hostgroup
118 query = "UPDATE inventory SET hostname='" + hostname + "',credentials='" +\
119 cred + "',hostgroup='" + results_hostgroups + \
120 "' WHERE hostname='" + hostname + "'"
122 results = sqlintf.Query (query)
126 query = "INSERT INTO inventory (hostgroup, hostname, credentials) VALUES ('" + \
127 hostgroup + "','" + hostname + "','" + cred + "')"
128 results = sqlintf.Query (query)
130 if len (results) > 0:
131 # Error loading playbook
135 return errorCode, diag
138 def readPlaybook (sqlintf, value, version=None):
143 print "***> in AnsibleSql.readPlaybook"
146 query = "SELECT MAX(version) FROM playbook WHERE name like'" + value + "%'"
147 print " Query:", query
148 results = sqlintf.Query (query)
149 version = results[0][0]
151 print " Provided playbook name:", value
152 print " Used version:", version
156 query = "SELECT value,type FROM playbook WHERE name='" + value + "@" + version + "'"
157 results = sqlintf.Query (query)
159 print "Query:", query
160 print "Results:", results
162 if len(results) == 0:
165 if len(results[0]) == 0:
171 return value, version, errorCode, diag
173 def readCredentials (sqlintf, tag):
177 print "***> in AnsibleSql.readCredential"
184 query = "SELECT hostgroup, hostname, credentials FROM inventory WHERE hostgroup LIKE '%" + \
186 query_results = sqlintf.Query (query)
189 for q in query_results:
190 if rec in q[0].split(','):
194 results = (q,) + results
196 if len(results) == 0:
198 query = "SELECT hostgroup, hostname, credentials FROM inventory WHERE hostname='" + \
200 results = sqlintf.Query (query)
202 print " Query:", query
203 print " Results:", len(results), results
205 if len(results) == 0:
209 credentials = 'ansible_connection=ssh ansible_ssh_user=na ansible_ssh_private_key_file=na\n'
210 diag.append([hostgroup, hostname, credentials])
213 for i in range(len (results)):
214 for h in results[i][0].split(','):
216 hostname = results[i][1]
217 credentials = results[i][2]
218 diag.append([hostgroup, hostname, credentials])
220 return errorCode, diag
223 if __name__ == '__main__':
225 ################################################################
227 ################################################################
228 host="localhost" # your host, usually localhost
229 user="mysql_user_id" # your username
230 passwd="password_4_mysql_user_id" # your password
231 db="ansible" # name of the data base
233 playbook_path = "/home/ubuntu/RestServerOpenSource/"
234 inventory = "/home/ubuntu/RestServerOpenSource/Ansible_inventory"
235 ################################################################
237 onlyfiles = [f for f in listdir(playbook_path)
238 if isfile(join(playbook_path, f))]
240 sqlintf = mySql (host, user, passwd, db)
244 print "Loading playbooks"
245 for file in onlyfiles:
248 name = file.split (".yml")[0]
249 print " Loading:", name
250 version = name.split("@")[1]
251 errorCode, diag = loadPlaybook (sqlintf, name, version, '.yml')
253 print " Results: Failed - ", diag
255 print " Results: Success"
257 print "\nLoading inventory"
263 file = open(inventory, 'r')
267 if '[' in line and ']' in line:
268 hostgroup = line.strip().replace('[','').replace(']','')
270 elif hostgroup and len(line.strip())>0:
271 host = line.strip().split(" ")[0]
272 credentials = line.replace(host,"")
273 inv[hostgroup][host] = credentials
277 for hostgroup in inv:
278 print " Loading:", hostgroup
281 for hostname in inv[hostgroup]:
282 cred = inv[hostgroup][hostname]
283 errorCode, diag = loadCredentials (sqlintf, hostgroup, hostname, cred)
285 print " Results: Failed - ", diag
287 print " Results: Success"
289 print "\nReading playbook"
294 print "Cannot connect to MySql:", sqlintf.error
297 name = "ansible_sleep"
298 print "Reading playbook:", name
299 value, version, errorCode, diag = readPlaybook (sqlintf, name)
301 print "Results: Failed - ", diag
303 print "Results: Success"
308 print "\nReading inventory"
312 tag = ["your_inventory_test_group_name"]
313 print "Reading inventory tag:", tag
314 errorCode, diag = readCredentials (sqlintf, tag)
316 print "Results: Failed - ", diag
318 print "Results: Success"