Bikarhêner:Balyozxane/skrîpt/py/category creator run.py

"""
Bi alîkariya [[Bikarhêner:Balyozxane/skrîpt/py/category creator.py]] kategoriyên ku tên xwestin çêdike. Wisa:

python pwb.py category_creator_run -lang:ku -family:wikipedia -wantedcategories
"""

import pywikibot
from pywikibot.bot import SingleSiteBot, ConfigParserBot, AutomaticTWSummaryBot
from pywikibot import pagegenerators
from pywikibot.exceptions import LockedPageError
import re
from category_creator import CategoryCreator  # Import the CategoryCreator class
bedeng = False

def print_sirove(sirove):
    if not bedeng:
        return pywikibot.output(sirove)

class AppendTextBot(
    SingleSiteBot,
    ConfigParserBot,
    AutomaticTWSummaryBot,
):
    summary_key = 'basic-changing'
    use_redirects = False
    update_options = {
        'summary': None,
        'text': '',
        'top': False,
    }

    def __init__(self, *args, category_creator=None, **kwargs):
        self.category_creator = category_creator
        super().__init__(*args, **kwargs)

    def treat_page(self) -> None:
        page = self.current_page

        if page.exists():
            pywikibot.output(f"Skipping {page.title()} already exists.")
            return

        self.category_creator.create_ku_category(page.title(with_ns=False))
        
def main(*args: str) -> None:
    local_args = pywikibot.handle_args(args)
    gen_factory = pagegenerators.GeneratorFactory()
    local_args = gen_factory.handle_args(local_args)

    options = {'text': ''}

    for arg in local_args:
        option, _, value = arg.partition(':')
        if option in ('summary', 'text'):
            if not value:
                pywikibot.input(f'Please enter a value for {option}')
            options[option] = value
        else:
            options[option] = True
    category_creator = CategoryCreator()
    gen = gen_factory.getCombinedGenerator(preload=True)

    if not pywikibot.bot.suggest_help(missing_generator=not gen):
        bot = AppendTextBot(generator=gen, category_creator=category_creator, **options)
        bot.run()

if __name__ == '__main__':
    main()