131 lines
5.1 KiB
Python
131 lines
5.1 KiB
Python
# The accounting application of the Mia project.
|
|
# by imacat <imacat@mail.imacat.idv.tw>, 2020/9/1
|
|
|
|
# Copyright (c) 2020 imacat.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
"""The command to populate the database with the accounts.
|
|
|
|
"""
|
|
import os
|
|
import re
|
|
from pathlib import Path
|
|
|
|
from django.conf import settings
|
|
from django.core.management import BaseCommand, CommandParser, CommandError, \
|
|
call_command
|
|
from django.utils import timezone
|
|
from opencc import OpenCC
|
|
|
|
|
|
class Command(BaseCommand):
|
|
"""Updates the revision date, converts the Traditional Chinese
|
|
translation into Simplified Chinese, and then calls the
|
|
compilemessages command.
|
|
"""
|
|
help = ("Updates the revision date, converts the Traditional Chinese"
|
|
" translation into Simplified Chinese, and then calls the"
|
|
" compilemessages command.")
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self._cc: OpenCC = OpenCC("tw2sp")
|
|
self._now: str = timezone.localtime().strftime("%Y-%m-%d %H:%M%z")
|
|
|
|
def add_arguments(self, parser):
|
|
"""Adds command line arguments to the parser.
|
|
|
|
Args:
|
|
parser (CommandParser): The command line argument parser.
|
|
"""
|
|
parser.add_argument("app_dir", nargs="+",
|
|
help=("One or more application directories that"
|
|
" contains their locale subdirectories"))
|
|
parser.add_argument("--domain", "-d", action="append",
|
|
choices=["django", "djangojs"], required=True,
|
|
help="The domain, either django or djangojs")
|
|
|
|
def handle(self, *args, **options):
|
|
"""Runs the command.
|
|
|
|
Args:
|
|
*args (list[str]): The command line arguments.
|
|
**options (dict[str,str]): The command line switches.
|
|
"""
|
|
locale_dirs = [os.path.join(settings.BASE_DIR, x, "locale")
|
|
for x in options["app_dir"]]
|
|
missing = [x for x in locale_dirs if not os.path.isdir(x)]
|
|
if len(missing) > 0:
|
|
error = "Directories not exist: " + ", ".join(missing)
|
|
raise CommandError(error, returncode=1)
|
|
domains = [x for x in ["django", "djangojs"] if x in options["domain"]]
|
|
for locale_dir in locale_dirs:
|
|
for domain in domains:
|
|
self._handle_po(locale_dir, domain)
|
|
call_command("compilemessages")
|
|
|
|
def _handle_po(self, locale_dir: str, domain: str) -> None:
|
|
"""Updates a PO file in a specific directory
|
|
|
|
Args:
|
|
locale_dir: the locale directory that contains the PO file
|
|
domain: The domain, either django or djangojs.
|
|
"""
|
|
zh_hant = os.path.join(
|
|
locale_dir, "zh_Hant", "LC_MESSAGES", F"{domain}.po")
|
|
zh_hans = os.path.join(
|
|
locale_dir, "zh_Hans", "LC_MESSAGES", F"{domain}.po")
|
|
self._update_rev_date(zh_hant)
|
|
self._convert_chinese(zh_hant, zh_hans)
|
|
|
|
def _update_rev_date(self, file: str) -> None:
|
|
"""Updates the revision date of the PO file.
|
|
|
|
Args:
|
|
file: the PO file as its full path.
|
|
"""
|
|
size = Path(file).stat().st_size
|
|
with open(file, "r+") as f:
|
|
content = f.read(size)
|
|
content = re.sub("\n\"PO-Revision-Date: [^\n]*\"\n",
|
|
F"\n\"PO-Revision-Date: {self._now}\\\\n\"\n",
|
|
content)
|
|
f.seek(0)
|
|
f.write(content)
|
|
|
|
def _convert_chinese(self, zh_hant: str, zh_hans: str) -> None:
|
|
"""Creates the Simplified Chinese PO file from the Traditional
|
|
Chinese PO file.
|
|
|
|
Args:
|
|
zh_hant: the Traditional Chinese PO file as its full path.
|
|
zh_hans: the Simplified Chinese PO file as its full path.
|
|
"""
|
|
size = Path(zh_hant).stat().st_size
|
|
with open(zh_hant, "r") as f:
|
|
content = f.read(size)
|
|
content = self._cc.convert(content)
|
|
content = re.sub("^# Traditional Chinese PO file for the ",
|
|
"# Simplified Chinese PO file for the ", content)
|
|
content = re.sub("\n\"PO-Revision-Date: [^\n]*\"\n",
|
|
F"\n\"PO-Revision-Date: {self._now}\\\\n\"\n",
|
|
content)
|
|
content = re.sub("\n\"Language-Team: Traditional Chinese",
|
|
"\n\"Language-Team: Simplified Chinese", content)
|
|
content = re.sub("\n\"Language: [^\n]*\"\n",
|
|
"\n\"Language: Simplified Chinese\\\\n\"\n",
|
|
content)
|
|
with open(zh_hans, "w") as f:
|
|
f.write(content)
|