scripts/update_sledrun_with_wrsledrunjson.py

   1 #!/usr/bin/python
   2 import argparse
   3 from typing import List
   4
   5 from wrpylib.mwapi import WikiSite
   6
   7
   8 def query_revisions(site: WikiSite, title_list: List[str], extra_rv_prop: List[str]) -> List[dict]:
   9     rv_prop = ['timestamp', 'ids'] + extra_rv_prop
  10     pages = next(site.query(prop='revisions', titles=title_list, rvslots='*', rvprop=rv_prop))['pages']
  11     pages = sorted(pages, key=lambda p: title_list.index(p['title']))
  12     assert len(title_list) == len(pages)
  13     return pages
  14
  15
  16 def update_sledrun_wikitext_with_wrsledrunjson_tag(ini_files: List[str]):
  17     site = WikiSite(ini_files)
  18     cm_limit = 5  # 'max'
  19     for category_members in site.query(list='categorymembers', cmtitle='Kategorie:Rodelbahn', cmlimit=cm_limit):
  20         title_list = [page["title"] for page in category_members['categorymembers']]
  21         pages = query_revisions(site, title_list, ['content'])
  22         for page in pages:
  23             new_text = '<wrsledrunjson />\n\n[[Kategorie:Rodelbahn]]'
  24
  25             previous_text = page['revisions'][0]['slots']['main']['content'].strip()
  26             if new_text == previous_text:
  27                 continue
  28
  29             site(
  30                 'edit',
  31                 pageid=page['pageid'],
  32                 text=new_text,
  33                 summary='Rodelbahnbeschreibung wird jetzt direkt aus den JSON Daten generiert.',
  34                 bot=1,
  35                 baserevid=page['revisions'][0]['revid'],
  36                 nocreate=1,
  37                 token=site.token(),
  38             )
  39
  40
  41 def main():
  42     parser = argparse.ArgumentParser(description='Replace sledrun wikitext with <wrsledrunjson /> tag.')
  43     parser.add_argument('inifile', nargs='+', help='inifile.ini, see: https://www.winterrodeln.org/trac/wiki/ConfigIni')
  44     args = parser.parse_args()
  45     update_sledrun_wikitext_with_wrsledrunjson_tag(args.inifile)
  46
  47
  48 if __name__ == '__main__':
  49     main()