Skip to content

Commit

Permalink
Added cli
Browse files Browse the repository at this point in the history
  • Loading branch information
PJ-Finlay committed Dec 12, 2020
1 parent 05ac968 commit 0281953
Show file tree
Hide file tree
Showing 3 changed files with 47 additions and 1 deletion.
39 changes: 39 additions & 0 deletions argostranslate/cli.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
import argparse
import sys
from argostranslate import package, translate

def main():
# Parse args
parser = argparse.ArgumentParser()
parser.add_argument('text', help='The text to translate')
parser.add_argument('--from-lang',
help='The code for the language to translate from (ISO 639-1)')
parser.add_argument('--to-lang',
help='The code for the language to translate to (ISO 639-1)')
args = parser.parse_args()

# Perform translation
if args.from_lang != None and args.to_lang != None:
installed_languages = translate.load_installed_languages()
from_lang_index = None
for i, lang in enumerate(installed_languages):
if lang.code == args.from_lang:
from_lang_index = i
break
to_lang_index = None
for i, lang in enumerate(installed_languages):
if lang.code == args.to_lang:
to_lang_index = i
break
from_lang = installed_languages[from_lang_index]
to_lang = installed_languages[to_lang_index]
translation = from_lang.get_translation(to_lang)
if translation == None:
sys.exit('No translation installed from {} to {}'.format(
args.from_name, args.to_name))
else:
translation = translate.IdentityTranslation('')

# Print translation
print(translation.translate(args.text))

7 changes: 7 additions & 0 deletions bin/argos-translate-cli
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
#!/usr/bin/env python3

from argostranslate import cli

cli.main()


2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,5 +18,5 @@
packages=find_packages(),
install_requires=required_packages,
include_package_data=True,
scripts=['bin/argos-translate'],
scripts=['bin/argos-translate', 'bin/argos-translate-cli'],
)

0 comments on commit 0281953

Please sign in to comment.