#!/usr/bin/env python3

import glob
import shutil
import dateutil.parser
import frontmatter
from datetime import datetime, timedelta
import pytz

def main():
    files = glob.glob('*.md')
    if files:
        files.remove('_index.md')
    for f in files:
        post,front_matter = read_markdown_file(f)
        date = front_matter.get('date')
        if date==None:
            print(f"{f} has no date")
            continue
        print(date)
        new_date_string = convert_date_format(date)
        front_matter['date'] = new_date_string
        save_markdown_file(f, post)

def read_markdown_file(md_file):
    with open(md_file, 'r', encoding='utf-8') as f:
        post = frontmatter.load(f)
        front_matter  = post.metadata #.get('date')
        return post,front_matter  

def convert_date_format(date_obj):
    # 添加时区信息
    # local_tz = pytz.timezone('Asia/Shanghai')  # 假设时区为Asia/Shanghai
    # date_obj_with_tz = local_tz.localize(date_obj)
    # date_obj = datetime.strptime(old_date_obj, "%Y-%m-%d %H:%M:%S")

    # 将 datetime 对象格式化为新的日期字符串
    new_date_string = date_obj.strftime("%Y-%m-%dT%H:%M:%S+08:00")

    return new_date_string

def save_markdown_file(md_file, post):
    with open(md_file, 'w', encoding='utf-8') as file:
        file.write(frontmatter.dumps(post))

if __name__ == '__main__':
    main()