recursive_translator/rectrans.py at main · Seerban/recursive_translator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#!/usr/bin/env python3
import sys
import argparse
import random
from deep_translator import GoogleTranslator

__version__ = "0.1.0"

langs = GoogleTranslator().get_supported_languages()


# Random language from filtered language list.
def rand_lang() -> str:
    return random.choice(langs)


# Translate recursively, switching between randomized languages.
# show_lang shows the languages it chose as it executes.
# if target_lang is given it will loop between default lang and target_lang
def rec_translate(s: str, times: int, show_langs: bool = False, verbose: bool = False, default_lang: str = "en", target_lang: str = '') -> str:
    used_langs : list[str] = [default_lang]
    results : list[str] = [s]

    for i in range(times):
        if target_lang:
            lang = target_lang
        else:
            lang = rand_lang()


        if not target_lang:
            s = GoogleTranslator(source="auto", target=lang).translate(s)
            used_langs.append(lang)
        else:
            s = GoogleTranslator(source=default_lang, target=lang).translate(s)
            s = GoogleTranslator(source=lang, target=default_lang).translate(s)
            used_langs.append(lang)
            used_langs.append(default_lang)

        results.append(s)

    if not target_lang:
        s = GoogleTranslator(source="auto", target=default_lang).translate(s)
        used_langs.append(default_lang)
        results.append(s)

    if show_langs:
        print('->'.join(used_langs))

    if verbose:
        print('\n'.join(results[:-1]))

    return s


def main():
    parser = argparse.ArgumentParser(description="Recursive translation tool")
    parser.add_argument("-n", "--number", type=int, default=3, help="Number of times to translate")
    parser.add_argument("-s", "--show", action="store_true", help="Show which languages are being used.")
    parser.add_argument("-S", "--seed", type=int, default=0, help="Seed for random selection, 0 means randomized.")
    parser.add_argument("-l", "--lang", default="en", help="Default language")
    parser.add_argument("-t", "--target", default="", help="Target language")
    parser.add_argument("-v", "--verbose", action="store_true", help="Show all translations")
    parser.add_argument("-V", "--version", action="store_true", help="rectrans version")
    parser.add_argument("text", nargs="?", help="Text to translate")

    args = parser.parse_args()

    if args.version:
        print(f"rectrans {__version__}")
        return

    if args.text is None:
        print(parser.format_help())
        return

    args.text = args.text.strip()

    if len(args.text) >= 5000:
        print("text character limit is 5000")
        return

    if args.seed: random.seed(args.seed)

    result = rec_translate(
        s=args.text,
        times=args.number,
        show_langs=args.show,
        verbose = args.verbose,
        default_lang=args.lang,
        target_lang=args.target,
    )

    print(result)

if __name__ == "__main__":
    main()