autoflake_fix_code_remove_unused_imports_and_variables.py

1# While autoflake is primarily used as a command-line tool, 
2# this is the official way to execute it via Python:
3
4import autoflake
5
6# Example: Programmatically removing unused imports and variables from a string
7code_with_unused_imports = """\
8import os
9import sys
10
11def my_function():
12    x = 10
13    print("Hello")
14"""
15
16# The fix_code function is the core API used by the CLI
17clean_code = autoflake.fix_code(
18    code_with_unused_imports,
19    remove_all_unused_imports=True,
20    remove_unused_variables=True
21)
22
23print(clean_code)