def cmd_extract(args):
pkginfo = load_pkginfo(args)
for cident, cdefn in pkginfo['components'].items():
if args.component is not None and cident not in args.component:
continue
if isinstance(cdefn, six.string_types):
cmod = cdefn
else:
cmod = cdefn['module']
module = import_module(cmod)
modpath = module.__path__[0]
dist = get_distribution(args.package)
meta = dict(message_from_string(dist.get_metadata('PKG-INFO')))
catalog = Catalog(project=args.package,
version=dist.version,
copyright_holder=meta.get('Author'),
msgid_bugs_address=meta.get('Author-email'),
fuzzy=False,
charset='utf-8')
method_map, options_map = get_mappings()
def log_callback(filename, method, options):
if method != 'ignore':
filepath = os.path.normpath(os.path.join(modpath, filename))
logger.debug('Extracting messages from %s', filepath)
extracted = extract_from_dir(modpath,
method_map,
options_map,
callback=log_callback)
for filename, lineno, message, comments, context in extracted:
catalog.add(message,
None, [(filename, lineno)],
auto_comments=comments,
context=context)
logger.info("%d messages extracted from component [%s]", len(catalog),
cident)
locale_path = Path(module.__path__[0]) / 'locale'
if locale_path.is_dir():
outfn = str(locale_path / '.pot')
else:
outfn = resource_filename(args.package, 'locale/%s.pot' % cident)
with io.open(outfn, 'wb') as outfd:
write_po(outfd, catalog, ignore_obsolete=True)
def cmd_compile(args):
def _compile_file(path, component, locale):
with io.open(str(path), 'r') as po:
catalog = read_po(po, locale=locale, domain=component)
logger.info("Compiling component [%s] locale [%s] (%d messages)...",
component, locale, len(catalog))
with io.open(str(path.with_suffix('.mo')), 'wb') as mo:
write_mo(mo, catalog)
with io.open(str(path.with_suffix('.jed')), 'w') as jed:
write_jed(jed, catalog)
components = list(load_components(args))
modern_file_layout = list()
for comp_id, comd_mod in components:
locale_path = Path(import_module(comd_mod).__path__[0]) / 'locale'
if not locale_path.is_dir() or len(list(
locale_path.glob('*.po'))) == 0:
continue
modern_file_layout.append(comp_id)
for po_path in locale_path.glob('*.po'):
locale = po_path.with_suffix('').name
_compile_file(po_path, comp_id, locale)
locpath = resource_filename(args.package, 'locale')
pofiles = []
for root, dirnames, filenames in os.walk(locpath):
for filename in fnmatch.filter(filenames, '*.po'):
pofiles.append(os.path.join(root, filename)[len(locpath) + 1:])
comp_ids = [cid for cid, _ in components]
for pofile in pofiles:
locale = pofile.split(os.sep, 1)[0]
component = os.path.split(pofile)[1][:-3]
if component not in comp_ids:
continue
if component in modern_file_layout:
logger.warning(
"Component [%s] was already compliled from new style "
"locale file layout! Skipping it.", component)
continue
_compile_file(Path(os.path.join(locpath, pofile)), component, locale)
def cmd_init(args):
root = resource_filename(args.package, 'locale')
for component, compmod in load_components(args):
mod = import_module(compmod)
locale_path = Path(mod.__path__[0]) / 'locale'
if locale_path.is_dir():
pot_file = locale_path / '.pot'
po_file = locale_path / ('%s.po' % args.locale)
if not pot_file.is_file():
logger.error("POT-file for component [%s] not found in [%s]",
component, str(pot_file))
continue
else:
pot_file = Path(root) / ('%s.pot' % component)
po_file = Path(root) / args.locale / 'LC_MESSAGES' / ('%s.po' %
component)
if not pot_file.is_file():
logger.error("POT-file for component [%s] not found in [%s]",
component, str(pot_file))
continue
if po_file.is_file() and not args.force:
logger.error(
"Component [%s] target file exists! Skipping. Use --force to overwrite.",
component)
continue
with io.open(str(pot_file), 'r') as infd:
catalog = read_po(infd, locale=args.locale)
catalog.locale = Locale.parse(args.locale)
catalog.revision_date = datetime.now(LOCALTZ)
with io.open(str(po_file), 'wb') as outfd:
write_po(outfd, catalog)
def cmd_update(args):
def _update_file(po_path, pot_path, component, locale):
logger.info("Updating component [%s] locale [%s]...", component,
locale)
with io.open(str(po_path), 'r') as po_fd, io.open(str(pot_path),
'r') as pot_fd:
po = read_po(po_fd, locale=locale)
pot = read_po(pot_fd)
po.update(pot, True)
with io.open(str(po_path), 'wb') as fd:
write_po(fd, po)
components = list(load_components(args))
modern_file_layout = list()
for comp_id, comp_mod in components:
locale_path = Path(import_module(comp_mod).__path__[0]) / 'locale'
if not locale_path.is_dir() or len(list(
locale_path.glob('*.po'))) == 0:
continue
modern_file_layout.append(comp_id)
pot_path = locale_path / '.pot'
if not pot_path.is_file():
logger.error("POT-file for component [%s] not found in [%s]",
comp_id, str(pot_path))
continue
for po_path in locale_path.glob('*.po'):
locale = po_path.with_suffix('').name
_update_file(po_path, pot_path, comp_id, locale)
root = resource_filename(args.package, 'locale')
pofiles = []
for dirname, dirnames, filenames in os.walk(root):
for filename in fnmatch.filter(filenames, '*.po'):
relative = os.path.relpath(os.path.join(dirname, filename), root)
pofiles.append(relative)
comp_ids = [cid for cid, _ in components]
for pofile in pofiles:
locale = pofile.split(os.sep)[0]
component = os.path.split(pofile)[1].split('.', 1)[0]
if component not in comp_ids:
continue
if component in modern_file_layout:
logger.warning(
"Component [%s] was already updated from new style "
"locale file layout! Skipping it.", component)
continue
logger.info("Updating component '%s' locale '%s'...", component,
locale) # NOQA
with open(os.path.join(root, pofile), 'r') as fd:
catalog = read_po(fd, locale=locale, charset='utf-8')
potfile = os.path.join(root, '%s.pot' % component)
if not os.path.isfile(potfile):
logger.warn("Template for %s:%s doesn't exists! Skipping.", locale,
component) # NOQA
with codecs.open(potfile, 'r', 'utf-8') as fd:
template = read_po(fd)
catalog.update(template, True)
with open(os.path.join(root, pofile), 'wb') as fd:
write_po(fd, catalog)