connect_database.py

from cassandra.auth import PlainTextAuthProvider
from cassandra.cluster import Cluster, ExecutionProfile, EXEC_PROFILE_DEFAULT
from cassandra.policies import WhiteListRoundRobinPolicy, DowngradingConsistencyRetryPolicy
from cassandra.query import tuple_factory
from time import strftime, localtime
from dotenv import load_dotenv
import datetime
import os
import json

load_dotenv()
client_id = os.environ.get("ASTRA_DB_ID")
client_secret = os.environ.get("ASTRA_DB_SECRET")

with open('config/config.json', 'r') as config:
    conf = json.load(config)
    
cloud_config = {'secure_connect_bundle': conf['secure_connect_bundle']}

auth_provider = PlainTextAuthProvider(client_id, client_secret) # authenticate
cluster = Cluster(cloud=cloud_config, auth_provider=auth_provider)
session = cluster.connect('my_moods')  # select keyspace


def update_mood(uuid: int, mood: str, today: str = strftime("%Y-%m-%d %H:%M:%S", localtime())):
    try:
        session.execute(f'SELECT (date, mood) FROM user{uuid}')
    except Exception:
        pass
    
    
    session.execute(f"""
                    CREATE TABLE IF NOT EXISTS user{uuid} (
                        date text,
                        mood text,
                        PRIMARY KEY (date, mood)
                    )
                    """)
    session.execute(f"INSERT INTO user{uuid} (date, mood) VALUES ('{today}', '{mood}')")


def get_moods(uuid: int) -> list:
    try:
        my_moods = session.execute(f'SELECT (date, mood) FROM user{uuid}')
    except Exception:
        return False
    moodlist = [[] for _ in range(7)]
    current_date = datetime.datetime.now().date()   # gets current date time

    for mood in my_moods:
        mood_date = datetime.datetime.strptime(mood.date__mood[0], "%Y-%m-%d %H:%M:%S").date()  # gets date of entry (no time)
        diff_days = (current_date - mood_date).days    # finds time diff
        if diff_days and diff_days <= 7:    # everything that is less than 7 days is appended based on diff
            moodlist[diff_days-1].append(mood.date__mood[1])
    return moodlist