QuoteBoard/quote_picker.py at main · n8many/QuoteBoard · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import random
from datetime import date, datetime
import pandas as pd

from typing import Optional

def coerce_dates(x):
        try:
            return date.fromisoformat(x)
        except ValueError:
            return None

def date_is_today(x):
        today = date.today()
        try:
            return x.month==today.month and x.day==today.day
        except ValueError:
            return False
        except AttributeError:
            return False


def pick_random_index(quotes: pd.DataFrame):
    if len(quotes) == 0:
        return None
    else:
        return quotes.index[random.randint(0, len(quotes)-1)]


def filter_recent_quotes(quotes: pd.DataFrame, recent_quotes: pd.Series):
    # Remove from list any quotes that have been used recently (defined by anti-repeat depth)
    return quotes.loc[~quotes.index.isin(recent_quotes)]


def filter_quotes_by_name(quotes, names):
    return quotes.loc[quotes['Who'].isin(names)]


def filter_quotes_by_date(quotes, target_date):

    date_quotes = quotes.loc[~quotes['Date'].apply(coerce_dates).isna()].copy()

    return date_quotes.loc[date_quotes['Date'].apply(coerce_dates).apply(date_is_today)]


def pick_quote(quotes: pd.DataFrame, recent_quotes: Optional[list] = None, birthdays: Optional[pd.DataFrame] = None, enable_quoteversary: bool = False):
    # Pick a quote from the list, with options for birthdays and quote anniversaries
    new_quotes = filter_recent_quotes(quotes, recent_quotes)

    if birthdays is not None and (bday_index:=pick_birthday_quote(new_quotes, birthdays)) is not None:
        return bday_index

    if enable_quoteversary and (qv_index:=pick_quoteversary_quote(new_quotes)) is not None:
        return qv_index

    if index := pick_random_index(new_quotes):
        return index
    else:
        # No unused quotes, ignore the recent list
        return pick_random_index(quotes)

def get_current_birthdays(birthdays):
    return birthdays.loc[birthdays['Birthday'].apply(date_is_today)]


def pick_birthday_quote(quotes, birthdays):
    # Pick a quote from the birthday person
    current_birthdays=get_current_birthdays(birthdays)

    if current_birthdays.empty:
        return None
    else:
        return pick_random_index(filter_quotes_by_name(quotes, current_birthdays['Name'].tolist()))


def pick_quoteversary_quote(quotes):
    # Pick a quote that happened on today's date, respects repeats
    today_quotes = filter_quotes_by_date(quotes, date.today())

    if today_quotes.empty:
        return None
    else:
        return pick_random_index(today_quotes)