auto_update_publications.py

import datetime
from pathlib import Path

from ruamel.yaml import YAML

from . import add_publications_by_author


def main(save_dir="_posts/papers", site_data_dir="_data/", use_ignore_list=True):
    site_data_dir = Path(site_data_dir)

    yaml = YAML()
    yaml.preserve_quotes = True
    with open(site_data_dir / "authors.yml") as f:
        authors = yaml.load(f)

    for author in authors.values():
        auto_update = author.get("auto_update_publications", False)
        author_id = author.get("semantic_scholar_id", False)

        if auto_update and author_id:
            year = int(datetime.datetime.now().year)
            parsed = {
                "start": year,
                "end": year,
                "author_id": author_id,
            }

            print(f"Updating publications for {author['name']} ({author_id})...")

            add_publications_by_author.main(
                parsed=parsed, save_dir=save_dir, use_ignore_list=use_ignore_list
            )


if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--save_dir",
        help="The directory to save the new files.",
        default="_posts/papers",
    )
    parser.add_argument(
        "--site_data_dir", help="The directory with the site data.", default="_data/"
    )
    parser.add_argument(
        "--use_ignore_list",
        help="Whether to use the ignore list.",
        default="true",
        choices=["true", "false"],
    )
    args = parser.parse_args()

    use_ignore_list = args.use_ignore_list == "true"

    main(**vars(args))