[
Function – Get Database Connection¶
Let us develop function to get Database Connection so that we can reuse in subsequent topics to perform CRUD Operations.
get_connectionwill take required paramters to establish a connection to the database and return connection object.- We can invoke that function and create required connection objects such as
sms_connectionso that we can perform database operations. sms_connectionis the connection object for our hypothetical sms database.- We can trigger this notebook from other notebooks by using
%run. - The function as well as objects such as
sms_connectioncan be used in the other notebooks from which this notebook is triggered. - In the similar lines we can have connection object
retail_connectionfor retail database.
In [1]:
import psycopg2
def get_connection(host, port, database, user, password):
connection = None
try:
connection = psycopg2.connect(
host=host,
port=port,
database=database,
user=user,
password=password
)
except Exception as e:
raise(e)
return connection
In [2]:
host = 'pg.itversity.com'
port = '5432'
database = 'itversity_sms_db'
user = 'itversity_sms_user'
password = 'itversity'
sms_connection = get_connection(
host=host,
port=port,
database=database,
user=user,
password=password
)
In [4]:
host = 'pg.itversity.com'
port = '5432'
database = 'itversity_sms_db'
user = 'itversity_sms_user'
password = 'itversity'
retail_connection = get_connection(
host=host,
port=port,
database=database,
user=user,
password=password
)
]