VisitingChefSlackBot/dining.py at main · atauln/VisitingChefSlackBot · 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
from datetime import date, timedelta
import requests

def get_all_dining(tod = date.today()):
    url = f'https://tigercenter.rit.edu/tigerCenterApi/tc/dining-all?date={tod.strftime("%Y-%m-%d")}'
    response = requests.get(
        url
    ).json()
    return response

def find_dining_loc(loc_search: str, tod = date.today()) -> list[dict]:
    locations = []
    foodstuffs = get_all_dining(tod)
    for item in foodstuffs['locations']:
        for menu in item['menus']:
            if loc_search in menu['name']:
                locations.append({
                    'name': menu['name'],
                    'location': item['name']
                })
    return locations

def find_visiting_chefs(tod = date.today()) -> list[dict]:
    locations = []
    foodstuffs = get_all_dining(tod)
    for item in foodstuffs['locations']:
        for menu in item['menus']:
            if menu['category'] == "Visiting Chef":
                locations.append({
                    'name': menu['name'],
                    'location': item['name'],
                    'description': menu['description']
                        if menu['description'] != None else ''
                })
    return locations

def check_vc_last_x_days(x: int) -> list[dict]:
    locations = []
    for i in range(x):
        locations += find_visiting_chefs(date.today() - timedelta(days=i))
    return True if locations else False


def find_visiting_chefs_fmt(tod = date.today()) -> str:
    locations = find_visiting_chefs(tod)
    if not check_vc_last_x_days(3):
        if not check_vc_last_x_days(4):
            return ""
        return "No visiting chefs found. The last visiting chef was 4 days ago, this bot will not post again until a new visiting chef is found."
    if len(locations) == 0:
        return "No visiting chefs found for today."
    base_str = "Here are the visiting chefs for today:\n"
    menu_per_location = {}
    for chef in locations:
        if menu_per_location.get(chef['location']) == None:
            menu_per_location[chef['location']] = [chef]
        else:
            menu_per_location[chef['location']].append(chef)
    for (location, menus) in menu_per_location.items():
        base_str += f"*{location}*\n"
        for menu in sorted(menus, key=lambda x: 'am' not in x['name'].lower()):
            base_str += f" - {menu['name']}\n"
    return base_str

def main():
    print(find_visiting_chefs())

if __name__ == "__main__":
    main()