mirror of https://github.com/k3s-io/k3s
97 lines
3.4 KiB
Python
97 lines
3.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
# Copyright 2017 The Kubernetes Authors.
|
||
|
|
#
|
||
|
|
# 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.
|
||
|
|
|
||
|
|
"""Extract strings from command files and externalize into translation files.
|
||
|
|
Expects to be run from the root directory of the repository.
|
||
|
|
|
||
|
|
Usage:
|
||
|
|
extract.py pkg/kubectl/cmd/apply.go
|
||
|
|
|
||
|
|
"""
|
||
|
|
import fileinput
|
||
|
|
import sys
|
||
|
|
import re
|
||
|
|
|
||
|
|
class MatchHandler(object):
|
||
|
|
""" Simple holder for a regular expression and a function
|
||
|
|
to run if that regular expression matches a line.
|
||
|
|
The function should expect (re.match, file, linenumber) as parameters
|
||
|
|
"""
|
||
|
|
def __init__(self, regex, replace_fn):
|
||
|
|
self.regex = re.compile(regex)
|
||
|
|
self.replace_fn = replace_fn
|
||
|
|
|
||
|
|
# global holding all strings discovered
|
||
|
|
STRINGS = []
|
||
|
|
|
||
|
|
def short_replace(match, file, line_number):
|
||
|
|
"""Replace a Short: ... cobra command description with an internationalization
|
||
|
|
"""
|
||
|
|
sys.stdout.write('{}i18n.T({}),\n'.format(match.group(1), match.group(2)))
|
||
|
|
STRINGS.append((match.group(2), file, line_number))
|
||
|
|
|
||
|
|
SHORT_MATCH = MatchHandler(r'(\s+Short:\s+)("[^"]+"),', short_replace)
|
||
|
|
|
||
|
|
def import_replace(match, file, line_number):
|
||
|
|
"""Add an extra import for the i18n library.
|
||
|
|
Doesn't try to be smart and detect if it's already present, assumes a
|
||
|
|
gofmt round wil fix things.
|
||
|
|
"""
|
||
|
|
sys.stdout.write('{}\n"k8s.io/kubernetes/pkg/util/i18n"\n'.format(match.group(1)))
|
||
|
|
|
||
|
|
IMPORT_MATCH = MatchHandler('(.*"k8s.io/kubernetes/pkg/kubectl/cmd/util")', import_replace)
|
||
|
|
|
||
|
|
def replace(filename, matchers):
|
||
|
|
"""Given a file and a set of matchers, run those matchers
|
||
|
|
across the file and replace it with the results.
|
||
|
|
"""
|
||
|
|
# Run all the matchers
|
||
|
|
line_number = 0
|
||
|
|
for line in fileinput.input(filename, inplace=True):
|
||
|
|
line_number += 1
|
||
|
|
matched = False
|
||
|
|
for matcher in matchers:
|
||
|
|
match = matcher.regex.match(line)
|
||
|
|
if match:
|
||
|
|
matcher.replace_fn(match, filename, line_number)
|
||
|
|
matched = True
|
||
|
|
break
|
||
|
|
if not matched:
|
||
|
|
sys.stdout.write(line)
|
||
|
|
sys.stdout.flush()
|
||
|
|
|
||
|
|
# gofmt the file again
|
||
|
|
from subprocess import call
|
||
|
|
call(["gofmt", "-s", "-w", filename])
|
||
|
|
|
||
|
|
# update the translation files
|
||
|
|
translation_files = [
|
||
|
|
"translations/kubectl/default/LC_MESSAGES/k8s.po",
|
||
|
|
"translations/kubectl/en_US/LC_MESSAGES/k8s.po",
|
||
|
|
]
|
||
|
|
|
||
|
|
for translation_filename in translation_files:
|
||
|
|
with open(translation_filename, "a") as tfile:
|
||
|
|
for translation_string in STRINGS:
|
||
|
|
msg_string = translation_string[0]
|
||
|
|
tfile.write('\n')
|
||
|
|
tfile.write('# https://github.com/kubernetes/kubernetes/blob/master/{}#L{}\n'.format(translation_string[1], translation_string[2]))
|
||
|
|
tfile.write('msgctxt {}\n'.format(msg_string))
|
||
|
|
tfile.write('msgid {}\n'.format(msg_string))
|
||
|
|
tfile.write('msgstr {}\n'.format(msg_string))
|
||
|
|
|
||
|
|
replace(sys.argv[1], [SHORT_MATCH, IMPORT_MATCH])
|