From 7c09a7a1b2f5913b7f7ea654cf3c8e8d54e22dc1 Mon Sep 17 00:00:00 2001 From: CloverHearts Date: Fri, 10 Jun 2016 18:57:16 +0900 Subject: [PATCH] fixed pyspark completion. --- spark/src/main/resources/python/zeppelin_pyspark.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/spark/src/main/resources/python/zeppelin_pyspark.py b/spark/src/main/resources/python/zeppelin_pyspark.py index 83ab7b3e77f..0ea547487e0 100644 --- a/spark/src/main/resources/python/zeppelin_pyspark.py +++ b/spark/src/main/resources/python/zeppelin_pyspark.py @@ -118,6 +118,9 @@ def isImportAllPackageUnderSparkSql(self): return self.version >= self.SPARK_1_3_0 class PySparkCompletion: + def __init__(self, interpreterObject): + self.interpreterObject = interpreterObject + def getGlobalCompletion(self): objectDefList = [] try: @@ -159,9 +162,10 @@ def getCompletion(self, text_value): for completionItem in list(objectCompletionList): completionList.add(completionItem) if len(completionList) <= 0: - print("") + self.interpreterObject.setStatementsFinished("", False) else: - print(json.dumps(list(filter(lambda x : not re.match("^__.*", x), list(completionList))))) + result = json.dumps(list(filter(lambda x : not re.match("^__.*", x), list(completionList)))) + self.interpreterObject.setStatementsFinished(result, False) output = Logger() @@ -205,7 +209,7 @@ def getCompletion(self, text_value): sqlc = SQLContext(sc, intp.getSQLContext()) sqlContext = sqlc -completion = PySparkCompletion() +completion = PySparkCompletion(intp) z = PyZeppelinContext(intp.getZeppelinContext()) while True :