Användare:Skalbot/update userlang categories.py

# -*- coding: utf-8  -*-

from __future__ import print_function

import time
import re
import pywikibot
import pywikibot.bot
from pywikibot.pagegenerators import PrefixingPageGenerator
from pywikibot.page import Page
from pywikibot.exceptions import NoPage, IsRedirectPage

cat_prefix = u'Kategori:Wiktionary:Användare '
main_cat = u'Kategori:Wiktionary:Användare per språk'
cats_to_process_prefix = cat_prefix + ''

pywikibot.bot.handleArgs()
site = pywikibot.Site('sv', 'wiktionary')

cats = PrefixingPageGenerator(cats_to_process_prefix, content=True)

re_end = '(?!)'

re_lang_title = re.compile(cat_prefix + '+' + re_end)
re_lvl_title = re.compile(cat_prefix + '(+)\-' + re_end)

re_lang_can_autofix = re.compile(
    u'^(Användare i den här kategorin visar att de har kunskap i språket +?.)\n?' +
    # iws
    u'((\n\+:]+\]\])*)\n?' +
    re_end
)
re_lang_fixed = re.compile(
    u'^Användare i den här kategorin visar att de har kunskap i språket +?.\n\n' +
    # cat
    '\\]\n?'
    # iws
    '((\n\+:]+\]\])*)\n?' +
    re_end
)
re_lvl_can_autofix = re.compile(
    u'^(Användare i den här kategorin visar att de kan språket +? på en kunskapsnivå av .)\n?' +
    # iws
    '((\n\+:]+\]\])*)\n?' +
    re_end
)
re_lvl_fixed = re.compile(
    u'^Användare i den här kategorin visar att de ' +
    u'(inte kan språket +?' +
    u'|har språket +? som modersmål' +
    u'|kan språket +? på en kunskapsnivå av ).\n\n'
    # cat
    '\+\]\]\n?'
    # iws
    '((\n\+:]+\]\])*)\n?' +
    re_end
)


def p(str):
    print(str)


def indent(str):
    print('\t' + str.replace('\n', '\n\t'))


for cat in cats:

    title = cat.title()
    text = cat.get()

    # level
    if re.match(re_lvl_title, title):
        if re_lvl_fixed.match(text):
            p('] skipped: already fixed')
        elif not re_lvl_can_autofix.match(text):
            p('] skipped: cannot autofix:')
            indent(text)
            break
        else:
            lang_code = re_lvl_title.match(title).group(1)
            # add cat
            text = re_lvl_can_autofix.sub(
                '\\1\n\n]\n\n\\2', text)
            if title.endswith('N'):
                text = text.replace(
                    u'de kan språket', u'de har språket').replace(
                    u'på en kunskapsnivå av N', u'som modersmål')
            elif title.endswith('0'):
                text = text.replace(
                    u'de kan språket', u'de inte kan språket').replace(
                    u' på en kunskapsnivå av 0', '')

            cat.text = text
            cat.save(
                comment=u'Lägger till kategori',
                botflag=True)
            # p('] (would save):')
            # indent(text)
    # lang
    elif re.match(re_lang_title, title):
        if re_lang_fixed.match(text):
            p('] skipped: already fixed')
        elif not re_lang_can_autofix.match(text):
            p('] skipped: cannot autofix:')
            indent(text)
            break
        else:
            text = re_lang_can_autofix.sub(
                '\\1\n\n]\n\n\\2', text)
            cat.text = text
            cat.save(
                comment=u'Lägger till kategori',
                botflag=True)
            # p('] (would save):')
            # indent(text)
    else:
        known_titles = [
            u'Kategori:Wiktionary:Användare',
            u'Kategori:Wiktionary:Användare per språk',
        ]
        if title in known_titles:
            p('] skipped')
        else:
            p('] skipped: unexpected title')
            break

p('DONE')