utils.db_connection

Contains the utility functions for opening and closing connections to a Postgres database.

 1"""
 2Contains the utility functions for opening and closing connections to a Postgres database.
 3"""
 4
 5import os
 6
 7import dotenv
 8from pg8000.native import Connection
 9
10
11def create_conn():
12    """
13    This function creates a connection to a PostgreSQL database using credentials from environment variables.
14
15    # Returns:
16        A pg8000 database Connection object.
17    """
18    dotenv.load_dotenv()
19
20    user = os.environ["DBUSER"]
21    database = os.environ["DBNAME"]
22    password = os.environ["DBPASSWORD"]
23    host = os.environ["HOST"]
24
25    return Connection(database=database, user=user, password=password, host=host)
26
27
28def close_conn(conn):
29    """
30    This function closes the connection to a PostgreSQL database.
31
32    # Arguments:
33        conn: a pg8000 Connection object that is to be closed.
34
35    # Returns:
36        None.
37    """
38    conn.close()
def create_conn(): 
12def create_conn():
13    """
14    This function creates a connection to a PostgreSQL database using credentials from environment variables.
15
16    # Returns:
17        A pg8000 database Connection object.
18    """
19    dotenv.load_dotenv()
20
21    user = os.environ["DBUSER"]
22    database = os.environ["DBNAME"]
23    password = os.environ["DBPASSWORD"]
24    host = os.environ["HOST"]
25
26    return Connection(database=database, user=user, password=password, host=host)

This function creates a connection to a PostgreSQL database using credentials from environment variables.

Returns:

A pg8000 database Connection object.
def close_conn(conn): 
29def close_conn(conn):
30    """
31    This function closes the connection to a PostgreSQL database.
32
33    # Arguments:
34        conn: a pg8000 Connection object that is to be closed.
35
36    # Returns:
37        None.
38    """
39    conn.close()

This function closes the connection to a PostgreSQL database.

Arguments:

conn: a pg8000 Connection object that is to be closed.

Returns:

None.