def post(self):
data = request.get_json()
system_id = data.get('system_id')
user_email = data.get('user_email')
if None in (system_id, user_email):
abort(400)
system = System.from_system_id(system_id)
user = User.from_email(user_email)
if None in (system, user):
abort(400)
if current_user == user:
abort(400)
if not Owner.is_owner_of(current_user, system):
abort(401)
sec_obj = SecondaryModel.create(user, system)
sec_obj.put()
secondaries = SecondaryModel.from_system(system)
if sec_obj not in secondaries:
secondaries.append(sec_obj)
ret = self.get_response_map_from_secondaries(secondaries)
return jsonify(ret)
def get(self, system_id):
system = System.from_system_id(system_id)
if system is None or not Owner.is_owner_of(current_user, system):
abort(400)
secondaries = SecondaryModel.from_system(system)
ret = self.get_response_map_from_secondaries(secondaries)
return jsonify(ret)
def delete(self, secondary_id):
secondary = SecondaryModel.from_id(secondary_id)
if secondary is None:
abort(400, "That secondary does not exist.")
system = secondary.system_key.get()
if not Owner.is_owner_of(current_user, system):
abort(401)
secondaries = SecondaryModel.from_system(system)
if secondary in secondaries:
secondaries.remove(secondary)
ret = self.get_response_map_from_secondaries(secondaries)
try:
secondary.key.delete()
except AttributeError:
pass
return jsonify(ret)
def test_from_system(random_secondary):
system = random_secondary.system_key.get()
assert random_secondary in Secondary.from_system(system)