SQL queries within Pandas DataFrames

sql_query.py

import re

import matplotlib.pyplot as plt
import pandas as pd
from sqlalchemy import create_engine


def sql_query(query: str) -> pd.DataFrame:
    """
    Helper function to run SQL queries on in-house Pandas DataFrames
    Parameters
    ----------
    query : str
        SQL query to run
    Returns
    -------
    pd.DataFrame
        DataFrame of results
    """
    REGEX = r"(FROM )(\w+)"
    DF_NAME = re.search(REGEX, query)[2]
    DF = eval(DF_NAME)
    engine = create_engine("sqlite://", echo=False)
    DF.to_sql(DF_NAME, con=engine)
    RESULT = pd.DataFrame(engine.execute(query).fetchall())

    return RESULT