main.py

"""
    Created by howie.hu at 2021-10-30.
    Description: 周刊初始化脚本
    Changelog: all notable changes to this file will be documented
"""
import os
import sys
import time

import yaml

DAILY_REPORT_TMP = """
# 老胡的 ChatGPT 日报信息流({0})

> 老胡的 ChatGPT 日报信息流，记录我当天看到的有价值的信息，欢迎进[微信群](https://images-1252557999.file.myqcloud.com/uPic/ETIbMe.jpg)一起交流。

## 👀 文章

### 文章

`ChatGPT` 总结:

>

## 🎯 项目

## 🤖 说明

老胡的 ChatGPT 日报信息流相关信息：

- Github 地址：[howie6879/gpt123.ai-daily](https://github.com/howie6879/gpt123.ai-daily)，觉得不错麻烦给我一个**Star**，谢谢 ❤️
- 浏览地址：[pt123.ai/daily](https://www.gpt123.ai/daily/)
"""


def read_mkdocs_config(path=""):
    """读取mkdocs配置

    Args:
        path (str): 可选，mkdocs配置路径
    """
    config_path = path or os.path.join(os.path.dirname(__file__), "mkdocs.yml")
    with open(config_path, "r", encoding="utf-8") as f:
        result = f.read()
        return yaml.load(result, Loader=yaml.FullLoader)


def write_mkdocs_config(data: dict, path=""):
    """写入mkdocs配置

    Args:
        data (dict): 配置数据
        path (str): 可选，mkdocs配置路径
    """
    config_path = path or os.path.join(os.path.dirname(__file__), "mkdocs.yml")
    with open(config_path, "w", encoding="utf-8") as f:
        yaml.dump(data, f, allow_unicode=True)


def gen_daily_file():
    """
    生成模版
    """
    root_path = os.path.dirname(__file__)

    date_time = time.strftime("%Y-%m-%d", time.localtime(time.time()))

    today_list = date_time.split("-")
    file_name = f"{today_list[1]}-{today_list[2]}"
    target_path = os.path.join(root_path, f"docs/{today_list[0]}/{file_name}.md")
    with open(target_path, "w", encoding="utf-8") as f:
        f.write(DAILY_REPORT_TMP.format(file_name))


def gen_daily_title(years: int):
    """生成周刊标题

    Args:
        years (int): 年份
    """
    root_path = os.path.dirname(__file__)
    target_path = os.path.join(root_path, f"docs/{years}")
    res_dict = {}
    for file_name in os.listdir(target_path):
        if str(file_name).endswith(".md"):
            # 目标文件
            title = str(file_name).split(".md", maxsplit=1)[0]
            res_dict[title] = {
                "file_name": file_name,
                "title": title,
            }

    res_list = []
    for i in sorted(res_dict):
        file_data_dict: dict() = res_dict[i]
        print(f"- {file_data_dict['title']}: ./{years}/{file_data_dict['file_name']}")
        res_list.append(
            {
                f"{file_data_dict['title']} 日报": f"./{years}/{file_data_dict['file_name']}"
            }
        )

    # 获取yaml配置
    mkdocs_config = read_mkdocs_config()
    # 赋值
    for each in mkdocs_config["nav"]:
        for key, _ in each.items():
            if str(years) in key:
                # 倒序
                each[key] = res_list[::-1]
    # 写入新配置
    write_mkdocs_config(mkdocs_config)


def run():
    """
    启动
    """
    argv: list = sys.argv
    script_name = argv[1]
    if script_name == "gdt":
        gen_daily_title(2023)
    elif script_name == "gdf":
        gen_daily_file()
    else:
        print("错误脚本名称")
        exit()


if __name__ == "__main__":
    run()
    # 生成最新日报
    # gen_daily_title(2023)
    # gen_daily_file()