clean up code after debugging apache/datafusion-python#862

shoots/shoots_server.py

@@ -9,6 +9,7 @@
 import argparse
 import jwt
 import datetime
+
 try:
     from .jwt_server_auth import JWTServerAuthHandler, JWTMiddleware
 except ImportError:
@@ -152,14 +153,17 @@ def do_get(self, context, ticket):
             raise e
 
         except Exception as e:
-            raise flight.FlightServerError(extra_info=str(e))
+            exception = {"type":"DataFusionError",
+                         "message":str(e)}
+            raise flight.FlightServerError(extra_info=json.dumps(exception))
 
     def _get_arrow_table_from_sql(self, name, file_path, sql_query):
         try:
             ctx = SessionContext()
             ctx.register_parquet(name, file_path)
             result = ctx.sql(sql_query)
             table = result.to_arrow_table()
+
             return table
         except Exception as e:
             if "DataFusion error" in str(e):