yun主题相册生成器

# tree.py
source_path = r"C:\Users\SKNP\Documents\GitHub"
cdn_url = "https://cdn.jsdelivr.net/gh/Sknp1006/cdn@master"

# gen_albums.py
albums_path = r"C:\Users\SKNP\Documents\GitHub\hexo_source\SKNPLZ\source\albums"
tree_path = r"C:\Users\SKNP\Documents\GitHub\cdn"

from pathlib import Path
from setting import cdn_url, source_path

exclude_dir = ['.git', '.idea', '.txt']
exclude_file = ['txt', 'py', 'gitignore']  # 以"."开头的文件去掉"."



def is_not_txt(file):
    sp = file.split(".")
    if sp[-1] not in exclude_file:
        return 1


def write(file_list, path, type):
    name = '&'.join(str(file_list[0]).split('\\')[-3:-1])
    if type == 'albums':
        print(name)
        with open('{}.txt'.format(name), mode='w') as f:
            for img in file_list:
                caption = str(img).split("\\")[-1]
                src = "/".join([cdn_url, str(path).replace('\\', '/'), caption])
                f.writelines("  - caption: " + caption + '\n')
                f.writelines("    src: " + src + '\n')
    elif type == 'ordinary':
        with open('{}.txt'.format(name), mode='w') as f:
            for img in file_list:
                caption = str(img).split("\\")[-1]
                src = "/".join([cdn_url, str(path).replace('\\', '/'), caption])
                f.writelines(src + '\n')
    elif type == 'likes':
        pass


def tree(dir_name):
    if Path(dir_name).is_dir():
        path = Path(dir_name)
    else:
        path = Path("\\".join([source_path, dir_name]))
        if not path.exists():
            raise Exception('"{}" not exist'.format(path))

    path_list = [dir for dir in path.iterdir() if dir.is_dir()]
    file_list = [dir for dir in path.iterdir() if dir.is_file() and is_not_txt(str(dir))]
    if path_list:
        for path in path_list:
            if path.name not in exclude_dir:
                # print("path:", path)
                # print("dir_name:", dir_name)
                tree(path)

    if file_list:
        if 'albums' in str(file_list[0]).split('\\'):
            write(file_list, path, 'albums')
        else:
            write(file_list, path, 'ordinary')

import glob
import time
import random
from pathlib import Path
from tree import tree
from setting import cdn_path, albums_path


def create_albums(albums_dict, album, title=None, passwd=None,
                  cover="https://cdn.jsdelivr.net/gh/Sknp1006/cdn/img/avatar/none.jpg", desc=None):  # 创建新相册
    print("creating {}".format(album))
    md = "\\".join([albums_path, "{}.md".format(album)])  # 新建的md文件路径
    with open(md, 'w+', encoding='utf-8') as f:
        f.write('---\n')
        if title:
            f.write('title: {}\n'.format(title))
        else:
            f.write('title: {}\n'.format(album))
        f.write('date: {}\n'.format(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))))
        f.write('layout: gallery\n')
        if passwd:
            f.write('password: {}\n'.format(passwd))  # 设置密码
        f.write('photos:\n')
        f.writelines(albums_dict[album])
        f.write('---\n')
    with open("\\".join([albums_path, 'index.md']), 'r', encoding='utf-8') as f:  # 读取原来的index.md
        txt = f.readlines()[:-1]
    with open("\\".join([albums_path, 'index.md']), 'w', encoding='utf-8') as f:  # 在index.md添加新album
        f.writelines(txt)
        f.write('  - caption: {}\n'.format(title))
        f.write('    url: /albums/{}.html\n'.format(album))
        f.write('    cover: {}\n'.format(cover))
        f.write('    desc: {}\n'.format(desc))
        f.write('---\n')


def update_albums(albums_dict):  # 更新相册
    for album in albums_dict:
        md = "\\".join([albums_path, "{}.md".format(album)])  # 对应album的md文件
        if not Path(md).is_file():  # 找不到，则新建一个相册
            cover = random.choice(albums_dict[album][1::2]).strip()[5:]  # 随机选取封面
            create_albums(albums_dict, album, cover=cover)
        with open(md, 'w+', encoding='utf-8') as f:
            f.write('---\n')
            f.write('title: {}\n'.format(album))
            f.write('date: {}\n'.format(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))))
            f.write('layout: gallery\n')
            f.write('photos:\n')
            f.writelines(albums_dict[album])
            f.write('---\n')
        print("{} updated done!".format(album))


def read_txt(tree_path):
    albums_dict = {}
    txt_list = glob.glob(tree_path + "\\" + "*.txt")
    albums_txt = [txt for txt in txt_list if txt.split("\\")[-1].split("&")[0] == "albums"]
    for file in albums_txt:
        with open(file, 'r') as f:
            albums_dict[file.split("&")[-1].split(".")[0]] = f.readlines()
    return albums_dict


def update_tree(dir_name):
    tree(dir_name)
    return "update done"


if __name__ == '__main__':
    update_tree('img/albums')
    dict = read_txt(cdn_path)
    update_albums(dict)