daily-counter/app/queries/connection.py

from os import getenv

import streamlit as st
from sqlalchemy.sql import text
from streamlit.connections import BaseConnection

def connection() -> BaseConnection:
    _connection = st.connection("sql", url=getenv('DATABASE_URL'))
    with _connection.session as configured_session:
        configured_session.execute(text('PRAGMA foreign_keys=ON'))
    return _connection