aboutsummaryrefslogtreecommitdiff
path: root/sql_functions.py
blob: 54d1d05903dd22da44e622c4a8bf7a7b0af58284 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sqlite3 as sql
import pandas as pd


def execute_sql_statement(sql_statement):
    conn = sql.connect("database.db")
    cur = conn.cursor()
    cur.execute(sql_statement)
    rows = cur.fetchall()
    return rows


def run_sql_pandas(sql_statement):
    conn = sql.connect("database.db")
    df = pd.read_sql_query(sql_statement, conn).to_records(index=False)
    return df


def get_list_of_dict(keys, list_of_tuples):
    list_of_dict = [dict(zip(keys, values)) for values in list_of_tuples]
    return list_of_dict