Main.hs

import Data.Generics.Uniplate.Data (universeBi)
import Data.List (nub, intercalate)
import Language.Python.Common
import Language.Python.Version3 (parseModule)
import System.Environment (getArgs)

fromName :: DottedNameSpan -> String
fromName dn = intercalate "." [s | Ident s _ <- dn]

extract :: StatementSpan -> [String]
extract (Import items _) = [fromName name | ImportItem name _ _ <- items]
extract (FromImport (ImportRelative _ (Just name) _) _ _) = [fromName name]
extract _ = []

main :: IO ()
main = do
  [path] <- getArgs
  source <- readFile path
  let Right (module_, _) = parseModule source path
  putStrLn (unlines (nub (universeBi module_ >>= extract)))

package.yaml

name: hspg

dependencies:
  - base
  - language-python
  - uniplate

executables:
  hspg-exe:
    main: Main.hs