changed main

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)