Make the CT Lua script endpoint decode base64

appengine_scripts/skia-tree-status/skia_telemetry.py

 # Copyright (c) 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Skia Telemetry pages."""
 
 
+import base64
 import datetime
 import json
 import urllib2
 
 from google.appengine.ext import db
 
 from base_page import BasePage
 import utils
 
 
@@ skipping 539 matching lines @@
     admin_tasks = AdminTasks.get_all_admin_tasks_of_user(self.user.email())
     template_values['admin_tasks'] = admin_tasks
     template_values['pageset_types'] = PAGESET_TYPES
     chromium_builds = ChromiumBuilds.get_all_completed_chromium_builds()
     template_values['chromium_builds'] = chromium_builds
     template_values['oldest_pending_task_key'] = get_oldest_pending_task_key()
 
     self.DisplayTemplate('admin_tasks.html', template_values)
 
 
-def _AddLuaTask(request, username):
+def _AddLuaTask(request, username, lua_script, lua_aggregator):
   requested_time = datetime.datetime.now()
-  lua_script = db.Text(request.get('lua_script'))
-  lua_aggregator = db.Text(request.get('lua_aggregator'))
+  lua_script = db.Text(lua_script)
+  lua_aggregator = db.Text(lua_aggregator)
   description = request.get('description')
   pagesets_type, chromium_rev, skia_rev = request.get(
       'pagesets_type_and_chromium_build').split('-')
   if not description:
     description = 'None'
 
   LuaTasks(
       username=username,
       lua_script=lua_script,
       lua_aggregator=lua_aggregator,
       pagesets_type=pagesets_type,
       chromium_rev=chromium_rev,
       skia_rev=skia_rev,
       requested_time=requested_time,
       description=description).put()
 
 
 class AddLuaTask(BasePage):
   """Adds the specified lua task."""
   @utils.admin_only
   def post(self):
-    _AddLuaTask(self.request, self.request.get('username'))
+    username = self.request.get('username')
+    lua_script = urllib2.unquote(base64.b64decode(
+        self.request.get('lua_script')))
+    lua_aggregator = urllib2.unquote(base64.b64decode(
+        self.request.get('lua_aggregator')))
+    _AddLuaTask(self.request, username, lua_script, lua_aggregator)
 
 
 class LuaScriptPage(BasePage):
   """Displays the lua script page."""
 
   @utils.require_user
   def get(self):
     return self._handle()
 
   @utils.require_user
   def post(self):
     # Check if this is a delete lua task request.
     delete_key = self.request.get('delete')
     if delete_key:
       LuaTasks.delete_lua_task(delete_key)
       self.redirect('lua_script')
       return
 
     # It is an add lua task request.
-    _AddLuaTask(self.request, self.user.email())
+    lua_script = self.request.get('lua_script')
+    lua_aggregator = self.request.get('lua_aggregator')
+    _AddLuaTask(self.request, self.user.email(), lua_script, lua_aggregator)
     self.redirect('lua_script')
 
   def _handle(self):
     """Sets the information to be displayed on the main page."""
     template_values = self.InitializeTemplate(TEMPLATES_TITLE)
 
     add_telemetry_info_to_template(template_values, self.user.email(),
                                    self.is_admin)
 
     lua_tasks = LuaTasks.get_all_lua_tasks_of_user(self.user.email())
@@ skipping 635 matching lines @@
         completed_time=datetime.datetime.now()).put()
 
   if db.GqlQuery('SELECT __key__ FROM AdminTasks').get() is None:
     AdminTasks(
         username='Admin',
         task_name='Initial Table Creation',
         pagesets_type='Test type',
         requested_time=datetime.datetime.now(),
         completed_time=datetime.datetime.now()).put()