sqlscr-oops.py

# Step 1 : import module
import pymysql as p


class dbfuncs:

    def __init__(self, host, user, password, db, dbtype, query):
        self.host = host
        self.user = user
        self.password = password
        self.db = db
        self.dbtype = dbtype
        self.query = query
        print("Init for ", self.dbtype)

    def checkconnection(self):
        dbc = p.connect(host=self.host, user=self.user, password=self.password, db=self.db)
        print("connected successfully for ", self.dbtype)
        return dbc

    def executequery(self):
        dbc = self.checkconnection()
        mycursor = dbc.cursor()

        op = []
        for q in self.query:
            mycursor.execute(q)
            res = mycursor.fetchall()
            op.append(res)
        return op




santhoshdb = dbfuncs('localhost', 'root', 'root', 'santhosh', 'santhosh database',
                     ['select count(*) from login;', 'select * from login where name = "santhosh";'])
sanjaydb = dbfuncs('localhost', 'root', 'root', 'santhosh', 'sanjay database', ['desc testtable;'])
sysdb = dbfuncs('localhost', 'root', 'root', 'srinivas', 'srinivas database', ['select * from users;', ])

# dbfuncs.checkconnection(santhoshdb)
output = dbfuncs.executequery(sysdb)

for x in output:
    print(x)