def show_form(self, mid=None):
if mid:
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
else:
robot = Robot()
form = forms.RobotForm(data=robot.to_dict())
self.render_response('robot/form.html', form=form, mid=mid, robot=robot)
def show_form(self, mid, dsid):
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
if dsid:
dataset = DataSet.get_by_id(int(dsid), parent=robot.key)
else:
dataset = DataSet(parent=robot.key)
form = forms.DataSetForm(data=dataset.to_dict())
self.render_response('robot/datasetform.html', form=form, mid=mid, dsid=dsid,
dataset=dataset)
def test_set_robot_name(self):
self.loginUser()
self.giveTeamAdminAccess()
# There is no Robot models that exists yet for this team
form = self.getTeamInfoForm(1124)
form['robot_name'] = 'Test Robot Name'
response = form.submit().follow()
self.assertEqual(response.status_int, 301)
robot = Robot.get_by_id(Robot.renderKeyName('frc1124', self.now.year))
self.assertIsNotNone(robot)
self.assertEqual(robot.robot_name, 'Test Robot Name')
def process_form(self, mid, dsid):
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
if dsid:
dataset = DataSet.get_by_id(int(dsid), parent=robot.key)
else:
dataset = DataSet(parent=robot.key)
form = forms.DataSetForm(self.request.POST)
if form.validate():
dataset.populate(**form.data)
key = dataset.put()
return self.redirect_to('dataset-edit', mid=mid, dsid=key.id())
self.render_response('robot/datasetform.html', form=form, mid=mid, dsid=dsid,
dataset=dataset)
def test_delete_robot_name(self):
self.loginUser()
self.giveTeamAdminAccess()
Robot(
id=Robot.renderKeyName(self.team.key_name, self.now.year),
team=self.team.key,
year=self.now.year,
robot_name='First Robot Name',
).put()
form = self.getTeamInfoForm(1124)
self.assertEqual(form['robot_name'].value, 'First Robot Name')
form['robot_name'] = ''
response = form.submit().follow()
self.assertEqual(response.status_int, 301)
robot = Robot.get_by_id(Robot.renderKeyName('frc1124', self.now.year))
self.assertIsNone(robot)
def process_form(self, mid=None):
if mid:
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
else:
robot = Robot(parent=self.current_user.key)
form = forms.RobotForm(self.request.POST, obj=robot)
if form.validate():
# TODO better populate object
robot.name = form.data['name']
robot.schedule = form.data['schedule']
robot.rps = form.data['rps']
robot.timeout = form.data['timeout']
robot.urlsource = URLSource.factory(form.urlsource.form.data)
key = robot.put()
return self.redirect_to('robot-view', mid=key.id())
self.render_response('robot/form.html', form=form, mid=mid, robot=robot)
def get(self, robot_id):
robot = Robot.get_by_id(robot_id)
parsed_robot = Robot(robot['name']).json()
return parsed_robot, 200 if parsed_robot else 402
def run(self, mid):
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
taskmanager.enqueue_robot('/task/runrobot', robot)
self.response.write(json.encode({'status': 'ok'}))
def view_crawl(self, mid, cid):
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
crawl = Crawl.get_by_id(int(cid), parent=robot.key)
jobs = crawl.jobs
self.render_response('robot/crawl.html', robot=robot, crawl=crawl, jobs=jobs)
def delete(self, mid):
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
robot.key.delete()
return self.redirect_to('robot-index')
def view(self, mid):
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
crawls = Crawl.query(ancestor=robot.key).order(-Crawl.started_at).fetch()
datasets = DataSet.query(ancestor=robot.key).fetch()
self.render_response('robot/view.html', robot=robot, mid=mid, crawls=crawls,
datasets=datasets, schedules=SCHEDULES)
def delete(self, mid, dsid):
robot = Robot.get_by_id(int(mid), parent=self.current_user.key)
dataset = DataSet.get_by_id(int(dsid), parent=robot.key)
dataset.key.delete()
return self.redirect_to('robot-view', mid=mid)
