database.py

from pydantic import BaseModel
from friend import Friend
from connect_database import Database
import datetime
from cassandra.query import named_tuple_factory
from datetime import datetime
from datetime import date


class addFriend:
    def add(self, Friend):
        #Name and birthdate are mandatory
        #DB format: riend_information ( id UUID PRIMARY KEY, name text, birthdate text, pronouns text, 
            #allergies text, phone_number text, col1_name text, col1_info text, col2_name text, col2_info text, 
            #col3_name text, col3_info text, col4_name text, col4_info text);

        cql_columns = "id"
        cql_values = "uuid()"
        optional_counter = 1

        parameters = ["name", "birthdate", "allergies", "phone_number", "pronouns", "avatar", "optional1", "optional2", "optional3", "optional4"]
        for key in Friend:
            if key[1]:
                information = key[1]
                if 'optional' in key[0]:
                    info_list = list(information.items())
                    cql_columns = cql_columns + ", col" + str(optional_counter) +"_name, col" + str(optional_counter) +"_info"
                    cql_values = cql_values + ", '" + info_list[0][0] + "', '"+ info_list[0][1] +"'"
                    optional_counter += 1
                else:
                    if key[0] == 'birthdate':
                        dts = key[1]
                        information = dts.strftime('%Y/%m/%d')
                    cql_columns = cql_columns + ", " + key[0]
                    cql_values = cql_values + ", '" + information + "'"

        cql_command = "insert into friend_information (" + cql_columns + ") values ( " + cql_values + ")"
        #print(cql_command)

        newDatabase = Database()
        newDatabase.execute(cql_command)
        newDatabase.close()


#information returned in labeled list format
#use friend_list[i] to get one person's name, id, pronouns
#use friend_list[i].id/name/pronouns for just the specific info 
class getFriend:
    def getAllFriends(self):
        friend_list = []
        cql_command = "select id, name, pronouns from friend_information"

        newDatabase = Database()
        row = newDatabase.executeSelect(cql_command)
        newDatabase.close()
        
        if row:
            for x in row:
                friend_list.append(x)
        else:
            print("An error occurred.")
        
        return friend_list


#Returns all information about one person as a Friend var
class getFriendFromID:
    def returnOne(self, idd):
        cql_command = "select * from friend_information where id = " + idd
        info_list = []

        newDatabase = Database()
        row = newDatabase.executeSelect(cql_command)
        newDatabase.close()

        if row:
            for x in row:
                info_list.append(x)
        else:
            print("An error occurred.")
        
        one = Friend(
            name = info_list[0].name,
            birthdate = datetime.strptime(info_list[0].birthdate, '%Y/%m/%d'),
            allergies = info_list[0].allergies,
            pronouns = info_list[0].pronouns,
            phone_number = info_list[0].phone_number,
            optional1 = {info_list[0].col1_name: info_list[0].col1_info},
            optional2 = {info_list[0].col2_name: info_list[0].col2_info},
            optional3 = {info_list[0].col3_name: info_list[0].col3_info},
            optional4 = {info_list[0].col4_name: info_list[0].col4_info}
        )

        return(one)


#Queries the database for all birthdays, returns 4 of the upcoming ones from today.
#Returns a list of lists in form: [["name", "Month Day"],["name", "Month Day"]] in increasing days away from today
class birthdays:
    def getUpcomingBirthdays(self):
        def myFunc(e):
            return e[2]

        info = []
        today = datetime.today()

        cql_command = "select name, birthdate from friend_information"

        newDatabase = Database()
        row = newDatabase.executeSelect(cql_command)
        newDatabase.close()

        if row:
            for x in row:
                mini = []
                mini.append(x.name)
                dt= datetime.strptime(x.birthdate, '%Y/%m/%d')
                newtime= dt.replace(year = today.year)
                mini.append(newtime)
                d = newtime - today
                mini.append(d.days)
                info.append(mini)
        else:
            print("An error occurred.")
        
        info.sort(key=myFunc)

        final_info = []
        counter = 0
        for z in info:
            if counter >= 4:
                return final_info

            if z[2] > 0:
                lol = []
                lol.append(z[0])
                lol.append(z[1].strftime("%b %d"))
                final_info.append(lol)
                counter +=1
        
        return final_info