Commit 7ac6d7ec authored by dmMaze's avatar dmMaze
Browse files

add debug mode and debug translator

parent f63c779c

launch.py

+7 −0
Original line number Diff line number Diff line
@@ -29,9 +29,11 @@ parser = argparse.ArgumentParser() 
parser.add_argument("--reinstall-torch", action='store_true', help="launch.py argument: install the appropriate version of torch even if you have some version already installed")

parser.add_argument("--proj-dir", default='', type=str, help='Open project directory on startup')

parser.add_argument("--qt-api", default='', choices=QT_APIS, help='Set qt api')

parser.add_argument("--debug", action='store_true')

parser.add_argument("--requirements", default='requirements.txt')

args, _ = parser.parse_known_args()





def is_installed(package):

    try:

        spec = importlib.util.find_spec(package)
@@ -117,6 +119,9 @@ def restart(): 


def main():



    if args.debug:

        os.environ['BALLOONTRANS_DEBUG'] = '1' 



    from utils import appinfo



    commit = commit_hash()
@@ -154,6 +159,8 @@ def main(): 


    from ui import constants as C

    from ui import config as program_config



    C.DEBUG = args.debug

    C.DEFAULT_DISPLAY_LANG = QLocale.system().name()

    C.USE_PYSIDE6 = API == 'pyside6'

    if qtpy.API_NAME[-1] == '6':

modules/translators/hooks.py

+1 −1
Original line number Diff line number Diff line
@@ -6,7 +6,7 @@ CHS2CHT_CONVERTER = None 


from .base import BaseTranslator, TextBlock



def chs2cht(translations: List[str] = None, textblocks: List[TextBlock] = None, translator: BaseTranslator = None,) -> str:

def chs2cht(translations: List[str] = None, textblocks: List[TextBlock] = None, translator: BaseTranslator = None, **kwargs) -> str:

    

    if not translator.cht_require_convert or translator.lang_target != '繁體中文':

        return

modules/translators/trans_debug.py

0 → 100644
+50 −0
Original line number Diff line number Diff line 
import copy

import os



from .base import *







if os.environ.get('BALLOONTRANS_DEBUG', ''):



    @register_translator('Debug Original')

    class TransOriginal(BaseTranslator):



        concate_text = False

        cht_require_convert = True

        params: Dict = {

            'api_key': '',

            'delay': 0.0,

        }



        def _setup_translator(self):

            for k in self.lang_map.keys():

                self.lang_map[k] = 'dummy language'

            

        def _translate(self, src_list: List[str]) -> List[str]:

            return copy.copy(src_list)

        

    def transhook_copy_original(translations: List[str] = None, textblocks: List[TextBlock] = None, translator: BaseTranslator = None, **kwargs):

        if textblocks is not None:

            for ii, _ in enumerate(translations):

                translations[ii] = textblocks[ii].translation



    TransOriginal.register_postprocess_hooks({'copy_original': transhook_copy_original})



    @register_translator('Debug Source')

    class TransSource(BaseTranslator):



        concate_text = False

        cht_require_convert = True

        params: Dict = {

            'api_key': '',

            'delay': 0.0,

        }



        def _setup_translator(self):

            for k in self.lang_map.keys():

                self.lang_map[k] = 'dummy language'

            self.register_preprocess_hooks

            

        def _translate(self, src_list: List[str]) -> List[str]:

            return copy.copy(src_list)
 
 No newline at end of file

ui/constants.py

+3 −1
Original line number Diff line number Diff line
@@ -77,3 +77,5 @@ DEFAULT_DISPLAY_LANG = 'English' 


USE_PYSIDE6 = False

ON_MACOS = sys.platform == 'darwin'



DEBUG = False
 
 No newline at end of file