unimport_library_find_and_remove_unused_python_imports.py

1from unimport.session import Session
2
3source = """
4import os
5import sys
6
7print(sys.executable)
8"""
9
10session = Session()
11# To find unused imports
12unused_imports = session.list_unused_imports(source=source)
13for imp in unused_imports:
14    print(f"Unused import: {imp['name']} at line {imp['lineno']}")
15
16# To remove unused imports and get the refactored code
17refactored_source = session.refactor(source=source)
18print(refactored_source)