diff --git a/main.py b/main.py index 84e051c..c3fcdae 100755 --- a/main.py +++ b/main.py @@ -1,7 +1,8 @@ #!/usr/bin/pyhton3 import os import re -import sys.argv +import sys + from lib.translate import translate @@ -19,9 +20,6 @@ def scan_files(path) -> list: return list_files -# name = "weapongenerator.lua" - - def parsing_file(f): with open(f) as _f: _cached = _f.read() @@ -43,18 +41,22 @@ def pars_file(name: str) -> list: """ list_to_translate: list = [] for _str in parsing_file(name): - string_to_translate = re.findall('= "(.*[^/])(?: /\* (.*)\*/)?"%_[t|T]', _str) + string_to_translate = re.findall('"(.*[^/])(?: /\* (.*)\*/)?"%_[t|T]', _str) list_to_translate.append(string_to_translate) return list_to_translate def all_to_translate(_dir): _files = scan_files(_dir) - print(_files) - # for _file in _files: - # _strings = pars_file(_file) - # print(_strings) + for _file in _files: + _strings = pars_file(_file) + if len(_strings) > 0: print(_file, _strings, len(_strings)) if __name__ == '__main__': - all_to_translate(sys.argv[1]) + if len(sys.argv) > 1: + arg = sys.argv[1] + else: + arg = r'C:\Users\riksl\AppData\Roaming\Avorion\mods\WPE2' + + all_to_translate(arg)