diff options
Diffstat (limited to 'sjdbmk/daily.py')
| -rw-r--r-- | sjdbmk/daily.py | 45 |
1 files changed, 31 insertions, 14 deletions
diff --git a/sjdbmk/daily.py b/sjdbmk/daily.py index ce21bce..f141743 100644 --- a/sjdbmk/daily.py +++ b/sjdbmk/daily.py @@ -43,8 +43,26 @@ DAYNAMES = [ "Sunday", "Monday", ] -DAYNAMES_CHINESE = ["周一", "周二", "周三", "周四", "周五", "周六", "周日", "周一"] -DAYNAMES_SHORT = ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun", "Mon"] +DAYNAMES_CHINESE = [ + "周一", + "周二", + "周三", + "周四", + "周五", + "周六", + "周日", + "周一", +] +DAYNAMES_SHORT = [ + "Mon", + "Tue", + "Wed", + "Thu", + "Fri", + "Sat", + "Sun", + "Mon", +] def main() -> None: @@ -58,7 +76,9 @@ def main() -> None: # TODO: Verify consistency of date elsewhere ) parser.add_argument( - "--config", default="config.ini", help="path to the configuration file" + "--config", + default="config.ini", + help="path to the configuration file", ) args = parser.parse_args() @@ -114,9 +134,7 @@ 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"), @@ -129,9 +147,7 @@ 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] @@ -182,12 +198,11 @@ 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" + "inspattach-%s" % os.path.basename(inspiration_image_fn), + "rb", ) as ifd: inspiration_image_data = base64.b64encode(ifd.read()).decode("ascii") else: @@ -258,7 +273,9 @@ def generate( "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" + "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( |