Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use generate_script method in ATX

Best Python code snippet using ATX

command_controller.py

Source:command_controller.py

...23    def get_global(self, key: str):24        if key in self.globals:25            return self.globals[key]26        return ''27    def generate_script(self, key: str, directory_name: str):28        template_name = self.get_global('template')29        file_name = self.get_global('file_name')30        output_directory = verify_directory(self.get_global('output_directory'))31        if template_name == '':32            template_name = 'standard'33        34        if file_name == '':35            file_name = 'standard_script'36        37        template_name_and_path = 'templates/' + self.templates[template_name][0]38        config_file = 'configuration/' + self.templates[template_name][1]39        40        if not isfile(template_name_and_path) or not isfile(config_file):41            return42        with open(template_name_and_path, 'r') as file_template:43            template = file_template.read()44        with open(config_file, 'r', encoding='utf-8') as f:45            json_data = json.load(f)46        47        placeholder = {}48        placeholder['custom_join'] = ''49        placeholder['custom_condition'] = ''50        try:51            for node in json_data[key]:                    52                placeholder[node] = format_data(node, str(json_data[key][node]))53            54        except KeyError:55            print(f'Key: {key} not found')56            return57        58        if 'pool_language' in json_data[key]:59            language_file = f'configuration/language/{key}.txt'60            if os.path.isfile(language_file):61                with open(language_file) as f:62                    placeholder['pool_language'] = f.read()63        if 'salutation' in json_data[key]:64            salutation_file = f'configuration/salutation/{key}.txt'65            if os.path.isfile(salutation_file):66                with open(salutation_file) as f:67                    placeholder['salutation'] = f.read()68        # Look for custom scripts and apply them if found69        custom_join_file = self.get_global('custom_join')70        if custom_join_file == '':71            custom_join_file = 'custom/custom_join.txt'72        73        if os.path.isfile(custom_join_file):74            with open(custom_join_file) as f:75                placeholder['custom_join'] = f.read()76        custom_condition_file = 'custom/' + self.get_global('custom_condition')77        if custom_condition_file == '' or custom_condition_file == 'custom/':78            custom_condition_file = 'custom/custom_condition.txt'79        80        if os.path.isfile(custom_condition_file):81            with open(custom_condition_file) as f:82                    placeholder['custom_condition'] = f.read()83        generated_script = replace_placeholder(placeholder, template, key)84        85        directory_path = f'{output_directory}\\{directory_name}\\'86        if not os.path.isdir(directory_path):87            os.makedirs(directory_path)88        file_name_and_path = f'{output_directory}\\{directory_name}\\{file_name}_{key}.sql'89        if os.path.isfile(file_name_and_path):90            os.remove(file_name_and_path)91        with open(file_name_and_path, 'w') as f:92            f.write(generated_script)93        print(f'Script generated in {os.path.abspath(directory_path)}')94    def auto_config(self):95        files = os.listdir('custom/auto/')96        for file in files:97            config_file = 'custom/auto/' + file98            with open(config_file, 'r') as f:99                config_file_contents = json.load(f)100            for node in config_file_contents:101                self.set_global('template', node['template'])102                self.set_global('output_directory', node['output_directory'])103                self.set_global('file_name', node['file_name'])104                self.set_global('custom_join', node['custom_join'])105                self.set_global('custom_condition', node['custom_condition'])106                107                print('Generating scripts from auto config ...')108                for market in node['market'].split(','):109                    the_market = market.strip().upper()110                    self.generate_script(key=the_market, directory_name='autoconfig')111                    print(f'{the_market} done ...')112        print('... Complete')113    def meia(self):114        directory = 'MEIA'115        print('Generating scripts for MEIA ...')116        print()117        self.generate_script(key='UAE', directory_name=directory)118        print('UAE done ...')119        self.generate_script(key='KSA', directory_name=directory)120        print('KSA done ...')121        self.generate_script(key='IND', directory_name=directory)122        print('IND done ...')123        print()124        print('... Complete')125    def europe(self):126        directory = 'Europe'127        print('Generating scripts for Europe ...')128        print()129        130        self.generate_script(key='BESC', directory_name=directory)131        print('BESC done ...')132        self.generate_script(key='CH', directory_name=directory)133        print('CH done ...')134        self.generate_script(key='FR', directory_name=directory)135        print('FR done ...')136        self.generate_script(key='GR', directory_name=directory)137        print('GR done ...')138        self.generate_script(key='IB', directory_name=directory)139        print('IB done ...')140        self.generate_script(key='NEU', directory_name=directory)141        print('NEU done ...')142        self.generate_script(key='SEE', directory_name=directory)143        print('SEE done ...')144        self.generate_script(key='TR', directory_name=directory)145        print('TR done ...')146        self.generate_script(key='UK', directory_name=directory)147        print('UK done ...')148        print()149        print('... Complete')150    def besc(self):151        print('Generating script for BESC ...')152        self.generate_script(key='BESC', directory_name='BESC')153        print('... Done')154    def ch(self):155        print('Generating script for CH ...')156        self.generate_script(key='CH', directory_name='CH')157        print('... Done')158    def fr(self):159        print('Generating script for FR ...')160        self.generate_script(key='FR', directory_name='FR')161        print('... Done')162    def gr(self):163        print('Generating script for GR ...')164        self.generate_script(key='GR', directory_name='GR')165        print('... Done')166    def ib(self):167        print('Generating script for IB ...')168        self.generate_script(key='IB', directory_name='IB')169        print('... Done')170    def neu(self):171        print('Generating script for NEU ...')172        self.generate_script(key='NEU', directory_name='NEU')173        print('... Done')174    def see(self):175        print('Generating script for SEE ...')176        self.generate_script(key='SEE', directory_name='SEE')177        print('... Done')178    def tr(self):179        print('Generating script for TR ...')180        self.generate_script(key='TR', directory_name='TR')181        print('... Done')182    def uk(self):183        print('Generating script for UK ...')184        self.generate_script(key='UK', directory_name='UK')185        print('... Done')186    def uae(self):187        print('Generating script for UAE ...')188        self.generate_script(key='UAE', directory_name='UAE')189        print('... Done')190    191    def ksa(self):192        print('Generating script for KSA ...')193        self.generate_script(key='KSA', directory_name='KSA')194        print('... Done')195    def ind(self):196        print('Generating script for IN ...')197        self.generate_script(key='IND', directory_name='IND')198        print('... Done')199    def do_command(self, command):200        command_method = getattr(self, command.method.__name__)201        if command_method:...

mixins.py

Source:mixins.py

...46                print('unable to delete {}'.format(name))47        super(ScriptTearDown, self).tearDown()48class ScriptFactoryMixin(ScriptTearDown, object):49    def setUp(self):50        self.translate_script = factories.generate_script(os.path.join(config.WOOEY_TEST_SCRIPTS, 'translate.py'))51        self.choice_script = factories.generate_script(os.path.join(config.WOOEY_TEST_SCRIPTS, 'choices.py'))52        self.without_args = factories.generate_script(os.path.join(config.WOOEY_TEST_SCRIPTS, 'without_args.py'))53        self.subparser_script = factories.generate_script(os.path.join(config.WOOEY_TEST_SCRIPTS, 'subparser_script.py'))54        self.version1_script = factories.generate_script(55            os.path.join(config.WOOEY_TEST_SCRIPTS, 'versioned_script', 'v1.py'),56            script_name='version_test',57        )58        self.version2_script = factories.generate_script(59            os.path.join(config.WOOEY_TEST_SCRIPTS, 'versioned_script', 'v2.py'),60            script_name='version_test',61        )62        super(ScriptFactoryMixin, self).setUp()63class FileMixin(object):64    def setUp(self):65        self.storage = utils.get_storage(local=not wooey_settings.WOOEY_EPHEMERAL_FILES)66        self.filename_func = lambda x: os.path.join(wooey_settings.WOOEY_SCRIPT_DIR, x)67        super(FileMixin, self).setUp()68    def get_any_file(self):69        script = os.path.join(config.WOOEY_TEST_SCRIPTS, 'command_order.py')...

test_generate_script.py

Source:test_generate_script.py

1import pytest2def test_generate_script_no_blocks(visualizer):3    result = visualizer.generate_script("fakeid", list())4    assert result == False5def test_generate_script_no_blocks_text(visualizer):6    result = visualizer.generate_script("fakeid", list(), text=True)7    assert result == False8def test_generate_script_nonexistent(visualizer, full_sb3):9    result = visualizer.generate_script("fakeid", full_sb3["targets"][0]["blocks"])10    assert result == False11def test_generate_script_nonexistent_text(visualizer, full_sb3):12    result = visualizer.generate_script("fakeid", full_sb3["targets"][0]["blocks"], text=True)13    assert result == False14def test_generate_script_exists(parser, visualizer, full_sb3):15    target, _ = parser.get_target("q3cBGj=%#c^E;eayaZs6", full_sb3)16    result = visualizer.generate_script("q3cBGj=%#c^E;eayaZs6", target["blocks"])17    assert result == {'label': 'play sound (Sound! v) until done'}18def test_generate_script_exists_text(parser, visualizer, full_sb3):19    target, _ = parser.get_target("q3cBGj=%#c^E;eayaZs6", full_sb3)20    result = visualizer.generate_script("q3cBGj=%#c^E;eayaZs6", target["blocks"], text=True)21    22    assert result == "play sound (Sound! v) until done\n\n"23    24def test_generate_script_exists_limit(parser, visualizer, full_sb3):25    target, _ = parser.get_target("t0gCI_RcP^f%JiDTILyU", full_sb3)26    surrounding = parser.get_surrounding_blocks("t0gCI_RcP^f%JiDTILyU", full_sb3)27    result = visualizer.generate_script(surrounding[0], target["blocks"], surrounding)28    29    assert result == {'label': 'define test_function', 'next': {'label': 'ask [Question] and wait', 'next': {'label': 'if <(answer) > [50]> then', 'substack': {'label': 'set [test_variable v] to (pick random (1) to (10))', 'next': {'label': 'play sound (Sound! v) until done'}}}}}30def test_generate_script_starts_with_input(parser, visualizer, full_sb3):31    """Should fail since we are forcing scratch-to-blocks to use an INPUT to start."""32    target, _ = parser.get_target("LFsgr8N^R^kVy-c66YJ8", full_sb3)33    with pytest.raises(Exception):34        surrounding = parser.get_surrounding_blocks("LFsgr8N^R^kVy-c66YJ8", full_sb3, 2)35        result = visualizer.generate_script(surrounding[0], target["blocks"], surrounding, find_block=False)36def test_generate_script_nostart_with_input(parser, visualizer, full_sb3):37    """Should succeed since we're letting scratch-to-blocks find the closest BLOCK it can."""38    target, _ = parser.get_target("LFsgr8N^R^kVy-c66YJ8", full_sb3)39    surrounding = parser.get_surrounding_blocks("LFsgr8N^R^kVy-c66YJ8", full_sb3, 2)...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.