[SPARK-302] Local privilege escalation (apache#332)

Author: avovchenko

python/pyspark/accumulators.py

@@ -232,7 +232,6 @@ class _UpdateRequestHandler(SocketServer.StreamRequestHandler):
     def handle(self):
         from pyspark.accumulators import _accumulatorRegistry
         auth_token = self.server.auth_token
-
         def poll(func):
             while not self.server.server_shutdown:
                 # Poll every 1 second for new data -- don't block in case of shutdown.
@@ -259,16 +258,14 @@ def authenticate_and_accum_updates():
                 # we've authenticated, we can break out of the first loop now
                 return True
             else:
-                raise Exception(
-                    "The value of the provided token to the AccumulatorServer is not correct.")
+                raise Exception("The value of the provided token to the AccumulatorServer is not correct.")
 
         if auth_token is not None:
             # first we keep polling till we've received the authentication token
             poll(authenticate_and_accum_updates)
         # now we've authenticated if needed, don't need to check for the token anymore
         poll(accum_updates)
 
-
 class AccumulatorServer(SocketServer.TCPServer):
 
     def __init__(self, server_address, RequestHandlerClass, auth_token):
@@ -286,7 +283,6 @@ def shutdown(self):
         SocketServer.TCPServer.shutdown(self)
         self.server_close()
 
-
 def _start_update_server(auth_token):
     """Start a TCP server to receive accumulator updates in a daemon thread, and returns it"""
     server = AccumulatorServer(("localhost", 0), _UpdateRequestHandler, auth_token)