示例#1
0
文件: school.py 项目: devxoul/schoool
def search():
    query = request.values.get('query')
    if not query:
        raise BadRequest('The required parameter "query" is missing.')

    cache_key = 'school-search:' + query
    cached_data = cache.get(cache_key)
    if cached_data:
        return render_json(json.loads(cached_data))

    data = []
    for host in URLS:
        try:
            url = 'http://{}/spr_ccm_cm01_100.do'.format(host)
            params = {
                'kraOrgNm': query,
            }
            json_data = requests.get(url, params=params).json
            dirty = json_data.get('resultSVO').get('orgDVOList')
            data += [{
                'code': school['orgCode'],
                'name': school['kraOrgNm'],
                'address': school['zipAdres'],
                'type': school['schulCrseScCodeNm'],
            } for school in dirty]
        except Exception:
            pass

    cache.set(cache_key, json.dumps(data))
    return render_json(data)
示例#2
0
文件: school.py 项目: devxoul/schoool
def meals(code):
    year = request.values.get('year', date.today().year, type=int)
    month = request.values.get('month', date.today().month, type=int)

    cache_key = 'school-meals:{}:{}-{}'.format(code, year, month)
    cached_data = cache.get(cache_key)
    if cached_data:
        return render_json(json.loads(cached_data))

    if code[0] == 'A':
        host_cache_key = 'school-host:{}'.format(code)
        host = cache.get(host_cache_key)
        if not host:
            url = 'http://www.schoolinfo.go.kr/index.jsp?HG_CD={}'.format(code)
            html = requests.get(url).html
            address = html.find('dt', string='주소')\
                          .find_next_sibling('dd')\
                          .string
            cities = [u'서울', u'부산', u'대구', u'인천', u'광주', u'대전', u'울산',
                      u'세종', u'경기', u'강원', u'충청북도', u'충청남도', u'전라북도',
                      u'전라남도', u'경상북도', u'경상남도', u'제주']
            for i, city in enumerate(cities):
                if address.startswith(city):
                    host = URLS[i]
                    cache.set(host_cache_key, host)
                    break
    else:
        index = ord(code[0]) - 66
        host = URLS[index]

    url = 'http://{}/sts_sci_md00_001.do'.format(host)
    data = {
        'schulCode': code,
        'schulCrseScCode': 4,
        'ay': year,
        'mm': month,
    }

    meals = []

    html = requests.get(url, data=data).html
    if u'자료가 없습니다.' not in html.decode('utf-8'):
        tds = [list(td.stripped_strings) for td in html.find_all('td')]
        tds = filter(None, tds)

        for items in tds:
            meal = {
                'date': '{}-{:02d}-{:02d}'.format(year, month, int(items[0])),
                'lunch': [],
                'dinner': [],
            }

            try:
                dinner_index = items.index(u'[석식]')
            except ValueError:
                dinner_index = len(items) - 1

            meal['lunch'] = items[2:dinner_index]
            meal['dinner'] = items[dinner_index + 1:]
            meals.append(meal)

    cache.set(cache_key, json.dumps(meals))
    return render_json(meals)