interlanguage.py

#! /usr/bin/env python3

import argparse

from ws.client import API
from ws.interactive import require_login
from ws.interlanguage.Categorization import Categorization
from ws.interlanguage.CategoryGraph import CategoryGraph
from ws.interlanguage.Decategorization import Decategorization
from ws.interlanguage.InterlanguageLinks import InterlanguageLinks

modes = ["update", "orphans", "rename"]
_modes_desc = {
    "update": "fix categorization of i18n pages, init wanted categories and update all interlanguage links",
    "orphans": "list all orphans",
    "rename": "rename non-English pages to match the English title after renaming",
}
modes_description = "The available modes are:"
for m in modes:
    modes_description += f"\n- '{m}': {_modes_desc[m]}"


def main(args: argparse.Namespace, api: API) -> None:
    if args.mode == "update":
        # first fix categorization
        cat = Categorization(api)
        cat.fix_allpages()
        decat = Decategorization(api)
        decat.fix_allpages()
        # init wanted categories
        cg = CategoryGraph(api)
        cg.init_wanted_categories()
        # update intelanguage links
        il = InterlanguageLinks(api)
        il.update_allpages()
    elif args.mode == "orphans":
        il = InterlanguageLinks(api)
        for title in il.find_orphans():
            print(f"* [[{title}]]")
    elif args.mode == "rename":
        il = InterlanguageLinks(api)
        il.rename_non_english()
    else:
        raise Exception(f"Unknown mode: {args.mode}")


if __name__ == "__main__":
    import ws.config

    argparser = ws.config.getArgParser(description="Update interlanguage links", epilog=modes_description)
    API.set_argparser(argparser)
    _group = argparser.add_argument_group("interlanguage")
    _group.add_argument("--mode", choices=modes, default="update", help="operation mode of the script")

    args = ws.config.parse_args(argparser)

    api = API.from_argparser(args)
    require_login(api)

    main(args, api)