awesome-fastapi-projects/scripts/unique_imports.py

import re
import sys

filename_in = sys.argv[1]
filename_out = sys.argv[2]
file_in = open(filename_in, "r")
lines = file_in.readlines()
file_out = open(filename_out, "w")

imports = set()

for line in lines:
    match1 = re.search(r"(from *(?!\.)(.+?)(?= |\.))", line)
    match2 = re.search(r"(: *(import) (.+))", line)
    if match1 is not None:
        imports.add(match1.group(2))
    if match2 is not None:
        imports.add(match2.group(3))


for imp in sorted(list(imports)):
    file_out.write(f"{imp}\n")