zulip/tools/check-capitalization

74 lines
2.6 KiB
Python
Executable File

#!/usr/bin/env python3
# check for the venv
from lib import sanity_check
sanity_check.check_venv(__file__)
import argparse
import json
import os
import re
import subprocess
import sys
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from scripts.lib.zulip_tools import WARNING, FAIL, ENDC
from tools.lib.capitalization import check_capitalization
DJANGO_PO_REGEX = re.compile('msgid "(.*?)"')
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--show-ignored',
action='store_true', dest='show_ignored', default=False,
help='Show strings that passed the check because they '
'contained ignored phrases.')
parser.add_argument('--no-generate',
action='store_true', dest='no_generate', default=False,
help="Don't run makemessages command.")
args = parser.parse_args()
if not args.no_generate:
subprocess.call(['./manage.py', 'makemessages', '--locale', 'en'],
stderr=subprocess.STDOUT)
with open('static/locale/en/translations.json') as f:
data = json.load(f)
frontend = check_capitalization(list(data.keys()))
frontend_errors, frontend_ignored, banned_errors_front = frontend
with open('static/locale/en/LC_MESSAGES/django.po') as f:
rows = [r for r in DJANGO_PO_REGEX.findall(f.read()) if r]
backend = check_capitalization(rows)
backend_errors, backend_ignored, banned_errors_back = backend
if frontend_errors:
print(WARNING + "Strings not properly capitalized in frontend:" + ENDC)
print('\n'.join(frontend_errors))
if backend_errors:
print(WARNING + "Strings not properly capitalized in backend:" + ENDC)
print('\n'.join(backend_errors))
if banned_errors_front:
print(WARNING + "Found banned words in frontend strings" + ENDC)
print('\n'.join(banned_errors_front))
if banned_errors_back:
print(WARNING + "Found banned words in backend strings" + ENDC)
print('\n'.join(banned_errors_back))
if args.show_ignored:
print(WARNING + "Strings which were ignored: " + ENDC)
print('\n'.join(frontend_ignored + backend_ignored))
if frontend_errors or backend_errors:
# Point the user to the documentation on what the policy is.
docs_url = "https://zulip.readthedocs.io/en/latest/translating/translating.html#capitalization"
print(WARNING + "See " + docs_url + ENDC)
print(FAIL + "Failed!" + ENDC)
sys.exit(1)
else:
sys.exit(0)