haileyhalimj@gmail.com
Rename src/etl to src/preprocess for better code organization
1131bea
raw
history blame
496 Bytes
from sqlalchemy import create_engine
from dotenv import load_dotenv
import os
load_dotenv()
USER = os.getenv("POSTGRES_USER", "myuser")
PWD = os.getenv("POSTGRES_PASSWORD", "mypass")
DB = os.getenv("POSTGRES_DB", "mydb")
PORT = os.getenv("DB_PORT", "5432")
HOST = "localhost"
engine = create_engine(
f"postgresql+psycopg2://{USER}:{PWD}@{HOST}:{PORT}/{DB}", future=True
)
if __name__ == "__main__":
with engine.begin() as conn:
print(conn.execute("select version();").scalar())