import os
import json

def get_all_dir(dir):
    '''
    :brief: 获取所有的文件夹名
    :param dir: string 父文件夹路径
    :return: string[] 子文件夹名
    '''
    for _, dirs, _ in os.walk(dir):
        return dirs

def get_all_workshopid(dir):
    '''
    :brief:获取所有项目的workshopid
    :param dir: string 父文件夹路径
    :return: string[] id列表
    '''
    data = []
    dirs = get_all_dir(dir)
    for item in dirs:
        data.extend(get_all_dir(dir + r'\\' + item))
    return data

def create_config(items, sample):
    '''
    :brief:根据示例生成新的项
    :param items: string[] workshopid数组
    :param sample: dict 
    :return: dict[] 
    '''
    data = []
    for item in items:
        new_item = sample.copy()
        old = new_item['workshopid']
        for (k, v) in new_item.items():
            if k == 'title' or k == 'workshopid':
                new_item[k] = item
            elif isinstance(v, str):
                new_item[k] = v.replace(old, item)
        data.append(new_item)
    return data

# steam 的安装路径
steam_dir = r'D:\Steam\\'
# steam下载的文件路径
workshop_dir = steam_dir + r'steamapps\workshop\content'
# wallpager 的配置文件路径
wallpager_config_file = steam_dir + r'steamapps\common\wallpaper_engine\bin\workshopcache.json'

id_items = get_all_workshopid(workshop_dir)

print('总文件个数：%d' %(len(id_items)))

with open(wallpager_config_file, 'r+', encoding='utf-8') as fs:
    content = fs.read()
    res = json.loads(content)
    if len(res['wallpapers']) < 1:
        print('请先订阅一个作为示例')
        exit(0)
    sample = res['wallpapers'][0]
    exist_id = []
    for item in res['wallpapers']:
        exist_id.append(item['workshopid'])
    data = list(set(id_items).difference(set(exist_id)))
    if len(data) < 1:
        print('没有缓存的文件')
        exit(0)
    new_items = create_config(data, sample)
    res['wallpapers'].extend(new_items)
    content = json.dumps(res, indent=4, ensure_ascii=False)
    fs.seek(0)
    fs.write(content)
    print('成功生成缓存配置文件，可以进行取消订阅')