Habib Ur Rehman
Habib Ur Rehman

Reputation: 131

How put db connection in seperate file and use in another file in python

Hi i have problem regarding db connection. i want to put db connection in seprate file and can use in multiple files.

i have tried this

connection.py

import pymysql
import mysql.connector

class Connection:

    def __init__(self):
        conn = mysql.connector.connect(host="localhost", user="root", password="", db="")
        cur = conn.cursor()
        return cur, conn


main.py

import connection

cur, conn = connection.Connection()

Error

cur, conn = connection.Connection()
TypeError: __init__() should return None, not 'tuple'

Upvotes: 4

Views: 3719

Answers (1)

Jaybe Park
Jaybe Park

Reputation: 192

connection.py

import pymysql
import mysql.connector

def get_connection():
    conn = mysql.connector.connect(host="localhost", user="root", password="", db="")
    cur = conn.cursor()
    return cur, conn

main.py

import connection

cur, conn = connection.get_connection()

Upvotes: 7

Related Questions