autoflake.py
def fix_file(filename, args, standard_out):
    """Run fix_code() on a file."""
    encoding = detect_encoding(filename)
    with open_with_encoding(filename, encoding=encoding) as input_file:
        source = input_file.read()

    original_source = source

    isInitFile = os.path.basename(filename) == '__init__.py'

    if args.ignore_init_module_imports and isInitFile:
        ignore_init_module_imports = True
    else:
        ignore_init_module_imports = False

    filtered_source = fix_code(
        source,
        additional_imports=args.imports.split(',') if args.imports else None,
        expand_star_imports=args.expand_star_imports,
        remove_all_unused_imports=args.remove_all_unused_imports,
        remove_duplicate_keys=args.remove_duplicate_keys,
        remove_unused_variables=args.remove_unused_variables,
        ignore_init_module_imports=ignore_init_module_imports,
    )

    if original_source != filtered_source:
        if args.check:
            standard_out.write('Unused imports/variables detected.')
            sys.exit(1)
        if args.in_place:
            with open_with_encoding(filename, mode='w',
                                    encoding=encoding) as output_file:
                output_file.write(filtered_source)
        else:
            diff = get_diff_text(
                io.StringIO(original_source).readlines(),
                io.StringIO(filtered_source).readlines(),
                filename)
            standard_out.write(''.join(diff))
    else:
        if args.check:
            standard_out.write('No issues detected!')
def fix_file(filename, args, standard_out):
    """Run fix_code() on a file."""
    encoding = detect_encoding(filename)
    with open_with_encoding(filename, encoding=encoding) as input_file:
        source = input_file.read()

    original_source = source

    isInitFile = os.path.basename(filename) == '__init__.py'

    if args.ignore_init_module_imports and isInitFile:
        ignore_init_module_imports = True
    else:
        ignore_init_module_imports = False

    filtered_source = fix_code(
        source,
        additional_imports=args.imports.split(',') if args.imports else None,
        expand_star_imports=args.expand_star_imports,
        remove_all_unused_imports=args.remove_all_unused_imports,
        remove_duplicate_keys=args.remove_duplicate_keys,
        remove_unused_variables=args.remove_unused_variables,
        ignore_init_module_imports=ignore_init_module_imports,
    )

    if original_source != filtered_source:
        if args.check:
            standard_out.write('Unused imports/variables detected.')
            sys.exit(1)
        if args.in_place:
            with open_with_encoding(filename, mode='w',
                                    encoding=encoding) as output_file:
                output_file.write(filtered_source)
        else:
            diff = get_diff_text(
                io.StringIO(original_source).readlines(),
                io.StringIO(filtered_source).readlines(),
                filename)
            standard_out.write(''.join(diff))
    else:
        if args.check:
            standard_out.write('No issues detected!')
def fix_file ( filename , args , standard_out ) : encoding = detect_encoding ( filename ) with open_with_encoding ( filename , encoding = encoding ) as input_file : source = input_file . read ( ) original_source = source isInitFile = os . path . basename ( filename ) == '__init__.py' if args . ignore_init_module_imports and isInitFile : ignore_init_module_imports = True else : ignore_init_module_imports = False filtered_source = fix_code ( source , additional_imports = args . imports . split ( ',' ) if args . imports else None , expand_star_imports = args . expand_star_imports , remove_all_unused_imports = args . remove_all_unused_imports , remove_duplicate_keys = args . remove_duplicate_keys , remove_unused_variables = args . remove_unused_variables , ignore_init_module_imports = ignore_init_module_imports , ) if original_source != filtered_source : if args . check : standard_out . write ( 'Unused imports/variables detected.' ) sys . exit ( 1 ) if args . in_place : with open_with_encoding ( filename , mode = 'w' , encoding = encoding ) as output_file : output_file . write ( filtered_source ) else : diff = get_diff_text ( io . StringIO ( original_source ) . readlines ( ) , io . StringIO ( filtered_source ) . readlines ( ) , filename ) standard_out . write ( '' . join ( diff ) ) else : if args . check : standard_out . write ( 'No issues detected!' )
Run fix_code() on a file.
Run fix_code () on a file .