70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
from concurrent.futures import ThreadPoolExecutor, as_completed
|
|
import sys
|
|
import traceback
|
|
from colorama import Fore
|
|
import os
|
|
|
|
from rpgmakermvmz import handleMVMZ
|
|
from rpgmakerace import handleACE
|
|
|
|
THREADS = 20
|
|
|
|
# Info Message
|
|
print(Fore.LIGHTYELLOW_EX + "Do not close while translation is in progress. If a file fails or gets stuck, \
|
|
Translated lines will remain translated so you don't have to worry about being charged \
|
|
twice. You can simply copy the file generated in /translations back over to /files and \
|
|
start the script again. It will skip over any translated text." + Fore.RESET, end='\n\n')
|
|
|
|
def main():
|
|
estimate = ''
|
|
while estimate == '':
|
|
estimate = input('Select Translation or Cost Estimation:\n\n1. Translate\n2. Estimate\n')
|
|
match estimate:
|
|
case '1': estimate = False
|
|
case '2': estimate = True
|
|
case _: estimate = ''
|
|
|
|
version = input('Select the RPGMaker Version:\n\n1. MV/MZ\n2. ACE\n')
|
|
|
|
totalCost = 0
|
|
match version:
|
|
case '1':
|
|
# Open File (Threads)
|
|
with ThreadPoolExecutor(max_workers=THREADS) as executor:
|
|
futures = [executor.submit(handleMVMZ, filename, estimate) \
|
|
for filename in os.listdir("files") if filename.endswith('json')]
|
|
|
|
for future in as_completed(futures):
|
|
try:
|
|
totalCost = future.result()
|
|
except Exception as e:
|
|
tracebackLineNo = str(traceback.extract_tb(sys.exc_info()[2])[-1].lineno)
|
|
print(Fore.RED + str(e) + '|' + tracebackLineNo + Fore.RESET)
|
|
|
|
case '2':
|
|
# Open File (Threads)
|
|
with ThreadPoolExecutor(max_workers=THREADS) as executor:
|
|
for filename in os.listdir("files"):
|
|
if filename.endswith('json'):
|
|
future = executor.submit(handleACE, filename, estimate)
|
|
|
|
try:
|
|
totalCost = future.result()
|
|
except Exception as e:
|
|
print(Fore.RED + str(e))
|
|
|
|
if estimate == False:
|
|
# This is to encourage people to grab what's in /translated instead
|
|
deleteFolderFiles('files')
|
|
|
|
# Prevent immediately closing of CLI
|
|
print(totalCost)
|
|
input('Done! Press Enter to close.')
|
|
|
|
def deleteFolderFiles(folderPath):
|
|
for filename in os.listdir(folderPath):
|
|
file_path = os.path.join(folderPath, filename)
|
|
if file_path.endswith('.json'):
|
|
os.remove(file_path)
|
|
|
|
main()
|