pack_backlog_data.py

#!/usr/bin/env python3.11
# coding: utf-8
from typing import Dict, List

import yaml

from make_backlog_data import (
    TCacheData,
    cache_to_json,
    export_cache_format,
    performers_target,
    scenes_target,
    script_dir,
    submitted_target,
)


def get_data():
    print('collecting information...')

    scenes: TCacheData = {
        fp.stem: yaml.safe_load(fp.read_bytes())
        for fp in scenes_target.glob('*/*.yml')
    }
    performers: TCacheData = {
        fp.stem: yaml.safe_load(fp.read_bytes())
        for fp in performers_target.glob('*/*.yml')
    }

    submitted: Dict[str, List[str]] = yaml.safe_load(submitted_target.read_bytes())

    return scenes, performers, submitted


def main():
    scenes, performers, submitted = get_data()
    submitted_scenes = dict.fromkeys(submitted['scenes'])

    cache_data = export_cache_format(dict(scenes=scenes, performers=performers), submitted=submitted_scenes)
    (script_dir / 'stashdb_backlog.json').write_bytes(cache_to_json(cache_data, True))

    print('done')


if __name__ == '__main__':
    main()