language-python-test-0.5.0: src/ParsePretty.hs

import Language.Python.Common
import Language.Python.Version2 as V2
import Language.Python.Version3 as V3
import System.Exit
import System.Environment

data PythonVersion = Two | Three
   deriving (Eq, Show)

type Parser = String -> String -> Either ParseError (ModuleSpan, [Token])

main :: IO ()
main = do
   args <- getArgs
   case args of
      (versionStr:inFile:_rest) ->
         case parseVersion versionStr of
            Nothing -> do
               putStrLn $ "Unknown Python version: " ++ versionStr
               exitFailure
            Just version -> do
               contents <- readFile inFile
               let parser = pickParser version
               case parseAndPretty parser inFile contents of
                  Left error -> putStrLn $ prettyText error
                  Right ast -> putStrLn $ prettyText ast 
      _other -> putStrLn "Incorrect command line. Expected: <2|3|n> inputFileName"

pickParser :: PythonVersion -> Parser
pickParser Two = V2.parseModule
pickParser Three = V3.parseModule

parseAndPretty :: Parser -> FilePath -> String -> Either ParseError ModuleSpan
parseAndPretty parser fileName contents =
   case parser contents fileName of
      Left e -> Left e
      Right (ast, _comments) -> Right ast

parseVersion :: String -> Maybe PythonVersion
parseVersion "2" = Just Two
parseVersion "3" = Just Three
parseVersion _other = Nothing