Function main

src/codetext/__main__.py:35–89

()

def main():
    opt = get_args()
    
    # check args
    if opt.json:
        if not opt.output_file: 
            raise ValueError("Missing --output_file")
    if opt.language:
        if opt.language not in PL_MATCHING.keys():
            raise ValueError(
                "{language} not supported. Currently support {sp_language}"
                .format(language=opt.language, 
                        sp_language=list(PL_MATCHING.keys())))
    
    # check path
    for path in opt.paths:
        assert os.path.exists(path) == True, "paths is not valid"
        
        if os.path.isdir(path):
            files = [os.path.join(path, f) for f in os.listdir(path) \
                    if os.path.isfile(os.path.join(path, f))]
        elif os.path.isfile(path):
            files = [path]
            
        if opt.language:
            for file in files[:]:
                filename, file_extension = os.path.splitext(file)
                if file_extension not in PL_MATCHING[opt.language]:
                    files.remove(file)

    output_metadata = {}
    for file in files:
        filename, file_extension = os.path.splitext(file)
        
        if opt.language == None:
            for lang, ext_list in PL_MATCHING.items():
                if file_extension in ext_list:
                    language = lang
                    break
        else:
            language = opt.language

        output = parse_file(file, language=language)
        print_result(
            output, 
            file_name=str(filename).split(os.sep)[-1]+file_extension
        )
        output_metadata[file] = output
    
    if opt.json:
        save_path = opt.output_file
        with open(save_path, 'w') as output_file:
            json.dump(output_metadata, output_file, sort_keys=True, indent=4)
            print(50*'=')
            print("Save report to {path}".format(path=save_path))


if __name__ == '__main__':