bootstrap.py

import os
import psycopg2
from dotenv import load_dotenv

from src.db.db_utils import insert_user

load_dotenv()

DATABASE_URL = os.getenv("DATABASE_URL")
print(f"{DATABASE_URL=}")

# SQL statements to create tables
# TODO: remove the user_profiles table..
CREATE_TABLES_SQL = """
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";

CREATE TABLE IF NOT EXISTS users (
    user_id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
    first_name TEXT,
    last_name TEXT,
    email TEXT UNIQUE,
    date_of_birth DATE,
    height REAL,
    join_date DATE DEFAULT CURRENT_DATE,
    fitness_level TEXT,
    onboarded BOOLEAN DEFAULT FALSE,
    goal_set BOOLEAN DEFAULT FALSE
);

CREATE TABLE IF NOT EXISTS user_profiles (
    user_profile_id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
    user_id UUID,
    last_updated DATE,
    activity_preferences TEXT,
    workout_location TEXT,
    workout_frequency TEXT,
    workout_duration TEXT,
    workout_constraints TEXT,
    fitness_level TEXT,
    weight DOUBLE PRECISION,
    goal_weight DOUBLE PRECISION,
    FOREIGN KEY (user_id) REFERENCES users(user_id)
);

CREATE TABLE IF NOT EXISTS user_activities (
    user_activity_id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
    user_id UUID,
    activity_name TEXT,
    activity_location TEXT,
    activity_duration TEXT,
    activity_frequency TEXT,
    FOREIGN KEY (user_id) REFERENCES users(user_id)
);

CREATE TABLE IF NOT EXISTS goals (
    goal_id TEXT PRIMARY KEY,
    user_id UUID,
    goal_type TEXT,
    description TEXT,
    target_value TEXT,
    current_value TEXT,
    unit TEXT,
    start_date DATE,
    end_date DATE,
    goal_status TEXT,
    notes TEXT,
    last_updated DATE,
    FOREIGN KEY (user_id) REFERENCES users(user_id)
);
"""


def create_tables():
    """Create tables in the PostgreSQL database."""
    conn = None
    try:
        conn = psycopg2.connect(DATABASE_URL)
        cur = conn.cursor()
        cur.execute(CREATE_TABLES_SQL)
        conn.commit()
        cur.close()
        print("Tables created successfully.")
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    finally:
        if conn is not None:
            conn.close()


if __name__ == "__main__":
    create_tables()

    # Insert a new user
    first_name = input("Enter your first name: ")
    last_name = input("Enter your last name: ")
    email = input("Enter your email: ")
    date_of_birth = input("Enter your date of birth (YYYY-MM-DD): ")
    height_cm = float(input("Enter your height in centimeters: "))

    user_id = insert_user(first_name=first_name, last_name=last_name, email=email, date_of_birth=date_of_birth, height_cm=height_cm)

    print(f"User created with ID: {user_id}")

    # Update the user_id in the agent.yaml file

    import yaml

    yaml_file = "configs/agent.yaml"
    with open(yaml_file, "r") as file:
        data = yaml.safe_load(file)
    data["user"]["user_id"] = user_id
    with open(yaml_file, "w") as file:
        yaml.dump(data, file)

    print(f"Updated {yaml_file} with user_id: {user_id}")