def update_calendar():
year = int(request.form['year'])
month = int(request.form['month'])
holiday = [str(x[0].day) for x in jpholiday.month_holidays(year, month)]
ym = f'{year:04d}{month:02d}'
query = f'''
select menu.date, "meal".name, "menu".s_stock,"menu".m_stock ,"menu".l_stock from ( select * from "menu" where date between {ym}00 and {ym}32)
as menu inner join "meal" on menu.meal_id = "meal".id;
'''
df = pd.read_sql(query, db_engine)
menus = []
for index, row in df.iterrows():
day = str(int(str(row['date'])[-2:]))
menu = row['name']
if '丼' in menu:
type = 'green'
else:
type = 'red'
menus.append({"day": day, "title": menu,
"s_stock": row["s_stock"], "m_stock": row["m_stock"], "l_stock": row["l_stock"], "type": type})
dict = {
"year": year,
"month": month,
"event": menus,
"holiday": holiday
}
return json.dumps(dict, ensure_ascii=False)
def test_count_month(self):
"""
2021年月祝日数
"""
self.assertEqual(len(jpholiday.month_holidays(2021, 1)), 2)
self.assertEqual(len(jpholiday.month_holidays(2021, 2)), 2)
self.assertEqual(len(jpholiday.month_holidays(2021, 3)), 1)
self.assertEqual(len(jpholiday.month_holidays(2021, 4)), 1)
self.assertEqual(len(jpholiday.month_holidays(2021, 5)), 3)
self.assertEqual(len(jpholiday.month_holidays(2021, 6)), 0)
self.assertEqual(len(jpholiday.month_holidays(2021, 7)), 2)
self.assertEqual(len(jpholiday.month_holidays(2021, 8)), 2)
self.assertEqual(len(jpholiday.month_holidays(2021, 9)), 2)
self.assertEqual(len(jpholiday.month_holidays(2021, 10)), 0)
self.assertEqual(len(jpholiday.month_holidays(2021, 11)), 2)
self.assertEqual(len(jpholiday.month_holidays(2021, 12)), 0)
def test_count_month(self):
"""
2017年月祝日数
"""
self.assertEqual(len(jpholiday.month_holidays(2017, 1)), 3)
self.assertEqual(len(jpholiday.month_holidays(2017, 2)), 1)
self.assertEqual(len(jpholiday.month_holidays(2017, 3)), 1)
self.assertEqual(len(jpholiday.month_holidays(2017, 4)), 1)
self.assertEqual(len(jpholiday.month_holidays(2017, 5)), 3)
self.assertEqual(len(jpholiday.month_holidays(2017, 6)), 0)
self.assertEqual(len(jpholiday.month_holidays(2017, 7)), 1)
self.assertEqual(len(jpholiday.month_holidays(2017, 8)), 1)
self.assertEqual(len(jpholiday.month_holidays(2017, 9)), 2)
self.assertEqual(len(jpholiday.month_holidays(2017, 10)), 1)
self.assertEqual(len(jpholiday.month_holidays(2017, 11)), 2)
self.assertEqual(len(jpholiday.month_holidays(2017, 12)), 1)
def test_count_month(self):
"""
1999年月祝日数
"""
self.assertEqual(len(jpholiday.month_holidays(1999, 1)), 2)
self.assertEqual(len(jpholiday.month_holidays(1999, 2)), 1)
self.assertEqual(len(jpholiday.month_holidays(1999, 3)), 2)
self.assertEqual(len(jpholiday.month_holidays(1999, 4)), 1)
self.assertEqual(len(jpholiday.month_holidays(1999, 5)), 3)
self.assertEqual(len(jpholiday.month_holidays(1999, 6)), 0)
self.assertEqual(len(jpholiday.month_holidays(1999, 7)), 1)
self.assertEqual(len(jpholiday.month_holidays(1999, 8)), 0)
self.assertEqual(len(jpholiday.month_holidays(1999, 9)), 2)
self.assertEqual(len(jpholiday.month_holidays(1999, 10)), 2)
self.assertEqual(len(jpholiday.month_holidays(1999, 11)), 2)
self.assertEqual(len(jpholiday.month_holidays(1999, 12)), 1)
def test_count_month(self):
"""
2014年月祝日数
"""
self.assertEqual(len(jpholiday.month_holidays(2014, 1)), 2)
self.assertEqual(len(jpholiday.month_holidays(2014, 2)), 1)
self.assertEqual(len(jpholiday.month_holidays(2014, 3)), 1)
self.assertEqual(len(jpholiday.month_holidays(2014, 4)), 1)
self.assertEqual(len(jpholiday.month_holidays(2014, 5)), 4)
self.assertEqual(len(jpholiday.month_holidays(2014, 6)), 0)
self.assertEqual(len(jpholiday.month_holidays(2014, 7)), 1)
self.assertEqual(len(jpholiday.month_holidays(2014, 8)), 0)
self.assertEqual(len(jpholiday.month_holidays(2014, 9)), 2)
self.assertEqual(len(jpholiday.month_holidays(2014, 10)), 1)
self.assertEqual(len(jpholiday.month_holidays(2014, 11)), 3)
self.assertEqual(len(jpholiday.month_holidays(2014, 12)), 1)
def test_count_month(self):
"""
2003年月祝日数
"""
self.assertEqual(len(jpholiday.month_holidays(2003, 1)), 2)
self.assertEqual(len(jpholiday.month_holidays(2003, 2)), 1)
self.assertEqual(len(jpholiday.month_holidays(2003, 3)), 1)
self.assertEqual(len(jpholiday.month_holidays(2003, 4)), 1)
self.assertEqual(len(jpholiday.month_holidays(2003, 5)), 2)
self.assertEqual(len(jpholiday.month_holidays(2003, 6)), 0)
self.assertEqual(len(jpholiday.month_holidays(2003, 7)), 1)
self.assertEqual(len(jpholiday.month_holidays(2003, 8)), 0)
self.assertEqual(len(jpholiday.month_holidays(2003, 9)), 2)
self.assertEqual(len(jpholiday.month_holidays(2003, 10)), 1)
self.assertEqual(len(jpholiday.month_holidays(2003, 11)), 3)
self.assertEqual(len(jpholiday.month_holidays(2003, 12)), 1)
def is_holiday(d):
holidays = [d[0] for d in jpholiday.month_holidays(CURRENT_YEAR, CURRENT_MONTH)]
return d in holidays
def test_count_month(self):
"""
2020年月祝日数
"""
self.assertEqual(len(jpholiday.month_holidays(2020, 1)), 2)
self.assertEqual(len(jpholiday.month_holidays(2020, 2)), 3)
self.assertEqual(len(jpholiday.month_holidays(2020, 3)), 1)
self.assertEqual(len(jpholiday.month_holidays(2020, 4)), 1)
self.assertEqual(len(jpholiday.month_holidays(2020, 5)), 4)
self.assertEqual(len(jpholiday.month_holidays(2020, 6)), 0)
self.assertEqual(len(jpholiday.month_holidays(2020, 7)), 2)
self.assertEqual(len(jpholiday.month_holidays(2020, 8)), 1)
self.assertEqual(len(jpholiday.month_holidays(2020, 9)), 2)
self.assertEqual(len(jpholiday.month_holidays(2020, 10)), 0)
self.assertEqual(len(jpholiday.month_holidays(2020, 11)), 2)
self.assertEqual(len(jpholiday.month_holidays(2020, 12)), 0)
def test_count_month(self):
"""
2002年月祝日数
"""
self.assertEqual(len(jpholiday.month_holidays(2002, 1)), 2)
self.assertEqual(len(jpholiday.month_holidays(2002, 2)), 1)
self.assertEqual(len(jpholiday.month_holidays(2002, 3)), 1)
self.assertEqual(len(jpholiday.month_holidays(2002, 4)), 1)
self.assertEqual(len(jpholiday.month_holidays(2002, 5)), 4)
self.assertEqual(len(jpholiday.month_holidays(2002, 6)), 0)
self.assertEqual(len(jpholiday.month_holidays(2002, 7)), 1)
self.assertEqual(len(jpholiday.month_holidays(2002, 8)), 0)
self.assertEqual(len(jpholiday.month_holidays(2002, 9)), 3)
self.assertEqual(len(jpholiday.month_holidays(2002, 10)), 1)
self.assertEqual(len(jpholiday.month_holidays(2002, 11)), 3)
self.assertEqual(len(jpholiday.month_holidays(2002, 12)), 1)
