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)
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)