Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- """
- Purpose: Connect to mysql database using config file.
- """
- from __future__ import unicode_literals
- from __future__ import print_function
- import configparser
- import mysql.connector
- class DBConnection(object):
- """
- Connect to mysql using config.ini file.
- """
- config_file = 'config.ini'
- def __init__(self, get_config_parser, env):
- self.env = env
- self.get_config_parser = get_config_parser
- self.host = self.get_config_parser[self.env]['host']
- self.user = self.get_config_parser[self.env]['user']
- self.password = self.get_config_parser[self.env]['password']
- self.database = self.get_config_parser[self.env]['database']
- def show_data(self):
- """
- Connect to db using credentails.
- """
- try:
- conn = mysql.connector.connect(
- host=self.host, user=self.user, password=self.password, database=self.database)
- if conn.is_connected():
- print("Connected to db")
- cur = conn.cursor() # cursor object for executing SQL statements
- cur.execute("select * from exam")
- result = cur.fetchall()
- print(result)
- else:
- print("Connection failed")
- except mysql.connector.Error:
- print("Error in connection")
- finally:
- conn.close()
- print("Connection close")
- def get_config():
- """
- Read file and parse
- """
- db_config = configparser.ConfigParser()
- db_config.read(DBConnection.config_file)
- return db_config
- if __name__ == '__main__':
- DB = DBConnection(get_config(), 'dev')
- DB.show_data()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement