Method import_statement

src/Parser.py:1322–1404

(self)

        return node

    def import_statement(self) -> Optional[Node]:
        if Def.fun_name != '':
            print_error('import_statement',
                        'Local imports are not allowed', self)

        module = ''
        while not self.no_more_tokens():
            tok = self.curr_token()
            if tok.kind == TokenKind.PERIOD:
                print_error('import_statement',
                            'Invalid period in module name', parser=self)
            self.next_token()

            part = tok.value
            module = os.path.join(module, part)

            if not self.no_more_tokens():
                self.match_token(TokenKind.PERIOD)

                if self.no_more_tokens():
                    print_error('import_statement',
                                'Invalid trailing period in import', parser=self)

        # Multi-file import
        if module.endswith('*'):
            module_roots = []
            module = module[:-1]

            module_source = module
            if not exists(module_source):
                for module_dir in Def.include_list:
                    other_source = os.path.join(module_dir, module_source)
                    if exists(other_source):
                        module_source = other_source

            def is_ml_source(f): return f.lower().endswith('.ml')
            modules = list(filter(is_ml_source, [f for f in listdir(
                module_source) if isfile(os.path.join(module_source, f))]))

            for module in modules:
                module = os.path.join(module_source, module)

                if not exists(module):
                    print_error('import_statement',
                                f'Module \'{module}\' does not exist.', self)

                if module in Def.included:
                    continue
                else:
                    Def.included.add(module)

                try:
                    module_root = Parser().parse(module)
                    module_roots.append(module_root)
                except RecursionError:
                    print_error('import_statement',
                                f'Cannot import module "{module}" (circular import)', self)