Johnson/johnson.py at main · BackGwa/Johnson · 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
#!/usr/bin/env python3
#-*- coding:utf-8 -*-

from datetime import datetime
import requests
import json

today = datetime.today()


# [함수] : 필요한 정보만 반환하기
def meal_zone(JSON, usetime, meal_value):

    mealzone = meal_value
    data = json.loads(JSON.text)
    result = ''

    if(usetime):
        hour = today.hour
        isbf = False if ((data['menu'][0]['breakfast']) == []) else True
        islc = False if ((data['menu'][0]['lunch']) == []) else True
        isdr = False if ((data['menu'][0]['dinner']) == []) else True

        if(isbf and islc and isdr):
            if(hour >= 0 and hour < 8):
                mealzone = 'breakfast'
            elif(hour >= 8 and hour < 14):
                mealzone = 'lunch'
            else:
                mealzone = 'dinner'

        elif(isbf and islc):
            mealzone = 'breakfast' if (hour >= 0 and hour < 8) else 'lunch'
        elif(islc and isdr):
            mealzone = 'lunch' if (hour >= 0 and hour < 14) else 'dinner'
        else:
            mealzone = 'lunch' if((islc) and ((not isbf) and (not isdr))) else 'None'

        result = ([mealzone] + data['menu'][0][f'{mealzone}']) if (mealzone != 'None') else ['급식 정보가 존재하지 않습니다.']

    else:
        result = [mealzone] + data['menu'][0][f'{mealzone}'] if((data['menu'][0][mealzone]) != []) else ['급식 정보가 존재하지 않습니다.']

    return result


# [함수] : 알레르기 정보 변환
def aleg_info(mealinfo):
    return ((str(mealinfo['name']) + str(mealinfo['allergy'])) if ('{' in str(mealinfo)) else mealinfo)


# [함수] : 급식 정보 가져오기
def get(school_type, school_code, date = [today.year, today.month, today.day], aleg = False, usetime = True, zonevalue = 'NoneValue'):

    allergy = 'formed' if (aleg == True) else 'hidden'

    API = f'https://schoolmenukr.ml/api/{school_type}/{school_code}?year={date[0]}&month={date[1]}&date={date[2]}&allergy={allergy}'
    response = requests.get(API)

    result = meal_zone(response, usetime, zonevalue)
    aleg_result = []

    if(aleg):
        for aleg_value in result:
            aleg_result += [aleg_info(aleg_value)]
        return aleg_result
    else:
        return result