| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| import os |
| from .utilities import bash |
| from .cloudException import CloudRuntimeException |
| import mysql.connector |
| |
| class Database(object): |
| """Database connection""" |
| def __init__(self, username, password=None, host='localhost', port='3306', db="cloud"): |
| self.host = host |
| self.username = username |
| self.password = password |
| self.port = port |
| self.db = db |
| |
| def connect(self): |
| return mysql.connector.connect(host=self.host, |
| user=self.username, |
| password=self.password, |
| database=self.db) |
| |
| def execute(self, statement): |
| txn = None |
| try: |
| txn = self.connect() |
| cursor = txn.cursor() |
| cursor.execute(statement) |
| cursor.close() |
| txn.commit() |
| if txn is not None: |
| try: |
| txn.close() |
| except: |
| pass |
| except: |
| raise CloudRuntimeException("Failed to execute: %s " % statement) |
| finally: |
| if txn is not None: |
| try: |
| txn.close() |
| except: |
| pass |
| |
| def testConnection(self): |
| try: |
| conn = self.connect() |
| conn.ping() |
| conn.close() |
| return True |
| except: |
| raise CloudRuntimeException("Failed to Connect to DB") |
| |
| def executeFromFile(self, file): |
| if not os.path.exists(file): |
| return False |
| |
| cmdLine = "mysql --host=" + self.host + " --port=" + str(self.port) + " --user=" + self.username |
| if self.password is not None: |
| cmdLine += " --password=" + self.password |
| |
| cmdLine += " < " + file |
| |
| try: |
| bash(cmdLine) |
| except: |
| raise CloudRuntimeException("Failed to execute " + cmdLine) |