diff options
Diffstat (limited to 'sjdbmk/daily.py')
| -rw-r--r-- | sjdbmk/daily.py | 38 |
1 files changed, 26 insertions, 12 deletions
diff --git a/sjdbmk/daily.py b/sjdbmk/daily.py index 4c7ad69..6820648 100644 --- a/sjdbmk/daily.py +++ b/sjdbmk/daily.py @@ -57,7 +57,9 @@ def main() -> None: # TODO: Verify validity of date # TODO: Verify consistency of date elsewhere ) - parser.add_argument("--config", default="config.ini", help="path to the configuration file") + parser.add_argument( + "--config", default="config.ini", help="path to the configuration file" + ) args = parser.parse_args() if args.date: @@ -113,7 +115,9 @@ def generate( logger.warning('Cycle day not found, using "SA"') for days_since_beginning in range(0, 5): - week_start_date = datetime_target - datetime.timedelta(days=days_since_beginning) + week_start_date = datetime_target - datetime.timedelta( + days=days_since_beginning + ) try: with open( "week-%s.json" % week_start_date.strftime("%Y%m%d"), @@ -126,7 +130,9 @@ def generate( else: break else: - raise FileNotFoundError("Cannot find a week-{date}.json file without five prior days") + raise FileNotFoundError( + "Cannot find a week-{date}.json file without five prior days" + ) try: aod = week_data["aods"][days_since_beginning] @@ -145,15 +151,15 @@ def generate( except KeyError: breakfast_tomorrow = None try: - snack_morning = week_data["snacks"][0][days_since_beginning] + snack_morning = week_data["snacks"]["Morning"][days_since_beginning] except (KeyError, IndexError): snack_morning = None try: - snack_afternoon = week_data["snacks"][1][days_since_beginning] + snack_afternoon = week_data["snacks"]["Afternoon"][days_since_beginning] except (KeyError, IndexError): snack_afternoon = None try: - snack_evening = week_data["snacks"][2][days_since_beginning] + snack_evening = week_data["snacks"]["Evening"][days_since_beginning] except (KeyError, IndexError): snack_evening = None @@ -180,9 +186,13 @@ def generate( inspiration_image_fn = inspjq["file"] if inspiration_image_fn: logger.info("Inspiration has attachment %s" % inspiration_image_fn) - inspiration_image_mime, inspiration_image_extra_encoding = mimetypes.guess_type(inspiration_image_fn) + inspiration_image_mime, inspiration_image_extra_encoding = ( + mimetypes.guess_type(inspiration_image_fn) + ) assert not inspiration_image_extra_encoding - with open("inspattach-%s" % os.path.basename(inspiration_image_fn), "rb") as ifd: + with open( + "inspattach-%s" % os.path.basename(inspiration_image_fn), "rb" + ) as ifd: inspiration_image_data = base64.b64encode(ifd.read()).decode("ascii") else: inspiration_image_data = None @@ -237,9 +247,11 @@ def generate( "today_dinner": dinner_today, "next_breakfast": breakfast_tomorrow, "the_week_ahead_url": the_week_ahead_url, - "snack_morning": snack_morning, - "snack_afternoon": snack_afternoon, - "snack_evening": snack_evening, + "today_snack": { + "Morning": snack_morning, + "Afternoon": snack_afternoon, + "Evening": snack_evening, + }, "inspiration_type": inspiration_type, "inspiration_shared_by": inspiration_shared_by, "inspiration_origin": inspiration_origin, @@ -251,7 +263,9 @@ def generate( "in_the_news_html_en": in_the_news_html_en, "in_the_news_html_zh": in_the_news_html_zh, } - with open("day-%s.json" % datetime_target.strftime("%Y%m%d"), "w", encoding="utf-8") as fd: + with open( + "day-%s.json" % datetime_target.strftime("%Y%m%d"), "w", encoding="utf-8" + ) as fd: json.dump(data, fd, ensure_ascii=False, indent="\t") logger.info( "Data dumped to " + "day-%s.json" % datetime_target.strftime("%Y%m%d"), |