69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
import kovachevbot
|
|
import pywikibot
|
|
import mwparserfromhell
|
|
import json
|
|
import sys
|
|
|
|
with open("verbal_nouns.json") as f:
|
|
VERBAL_NOUNS = json.load(f)
|
|
|
|
GRAVE = chr(0x300)
|
|
ACUTE = chr(0x301)
|
|
|
|
# Consume an iterator. For doing side effects using a generator expression in one line.
|
|
def do(iter):
|
|
for _ in iter: pass
|
|
|
|
def get_verbal_nouns(verb: str) -> str:
|
|
return VERBAL_NOUNS[verb]
|
|
|
|
def fix_infl_template(template: mwparserfromhell.nodes.Template):
|
|
def has_inflection(synonyms: list[str]) -> bool:
|
|
return any(s in template.params for s in synonyms)
|
|
|
|
def remove_positional_by_value(val):
|
|
try:
|
|
val = [p for p in template.params if p == val][0]
|
|
template.remove(val)
|
|
except:
|
|
pass
|
|
|
|
indef = ["indefinite", "indef", "indf", "indefinite state"]
|
|
singular = ["singular", "s", "sg"]
|
|
is_indef = has_inflection(indef)
|
|
is_singular = has_inflection(singular)
|
|
|
|
if "vnoun" in template.params or "verbal noun" in template.params:
|
|
if is_indef and is_singular:
|
|
do(remove_positional_by_value(i) for i in indef)
|
|
do(remove_positional_by_value(s) for s in singular)
|
|
else:
|
|
verb = str(template.get(2)).replace(GRAVE, "").replace(ACUTE, "")
|
|
try:
|
|
verbal_noun_list = get_verbal_nouns(verb)
|
|
if len(verbal_noun_list) > 1:
|
|
raise ValueError()
|
|
remove_positional_by_value("vnoun")
|
|
remove_positional_by_value("verbal noun")
|
|
template.params[1] = verbal_noun_list[0]
|
|
except KeyError:
|
|
print(f"Verb {verb} has no verbal noun defined in the lookup table", file=sys.stderr)
|
|
except ValueError:
|
|
print(f"Verb {verb} has more than one possible verbal noun: {verbal_noun_list}", file=sys.stderr)
|
|
|
|
for page in kovachevbot.iterate_safe(kovachevbot.iterate_category("Bulgarian non-lemma forms")):
|
|
page: pywikibot.Page
|
|
parsed = mwparserfromhell.parse(page.text)
|
|
|
|
bulgarian = parsed.get_sections([2], "Bulgarian")[0]
|
|
for template in bulgarian.filter_templates():
|
|
template: mwparserfromhell.nodes.Template
|
|
|
|
if template.name == "infl of" or template.name == "inflection of":
|
|
fix_infl_template(template)
|
|
|
|
out = str(parsed)
|
|
if out != page.text:
|
|
page.text = out
|
|
page.save("Update verbal noun forms' {{infl of}} to point to the verbal noun / turn 'indefinite singular verbal noun' to 'verbal noun'")
|
|
|