mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import os
|
|
import sys
|
|
from deep_translator import GoogleTranslator
|
|
|
|
def main():
|
|
# Get message from environment variable or command line argument
|
|
msg = os.environ.get('COMMIT_MSG', '')
|
|
if len(sys.argv) > 1:
|
|
msg = sys.argv[1]
|
|
|
|
if not msg:
|
|
print("No commit message found.")
|
|
return
|
|
|
|
# Use a single instance of translators to potentially benefit from some caching if it exists
|
|
translator_zh = GoogleTranslator(source='auto', target='zh-CN')
|
|
translator_en = GoogleTranslator(source='auto', target='en')
|
|
|
|
lines = msg.split('\n')
|
|
|
|
for line in lines:
|
|
# Extract leading whitespace (indentation)
|
|
indent = line[:len(line) - len(line.lstrip())]
|
|
content = line.strip()
|
|
|
|
if not content:
|
|
print(line) # Preserve empty lines or lines with only spaces
|
|
continue
|
|
|
|
try:
|
|
# Translate only the content to avoid translator messing with indentation
|
|
zh_trans = translator_zh.translate(content)
|
|
en_trans = translator_en.translate(content)
|
|
|
|
# Re-apply indentation to both translations
|
|
print(f"{indent}{zh_trans}")
|
|
print(f"{indent}{en_trans}")
|
|
|
|
except Exception as e:
|
|
# If translation fails, print original line (which contains indentation)
|
|
print(f"{line}")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|