def update_project(project_id):
new_project = request.get_json()
try:
project = Project.get(Project.id == project_id)
except models.DoesNotExist:
return jsonify({ 'message': 'Project not found' }), 400
if new_project['id'] != project.id:
return jsonify({ 'message': 'Cannot update project at this url' }), 400
try:
if new_project['markerOptions']:
marker_options = new_project['markerOptions']
for marker_name in marker_options:
try:
marker_option = MarkerOption.get(MarkerOption.name == marker_name)
ProjectMarkerOption.get(ProjectMarkerOption.project == project.id,
ProjectMarkerOption.marker_option == marker_option.id)
except models.DoesNotExist:
ProjectMarkerOption.create(project = project, marker_option = marker_option)
if new_project['users']:
users = new_project['users']
for user in users:
try:
project_user = ProjectUser.get(ProjectUser.app_user == user['id'],
ProjectUser.project == project)
except models.DoesNotExist:
ProjectUser.create(app_user = user['id'], project = project)
if new_project['stacks']:
stacks = new_project['stacks']
with db.atomic():
for stack in stacks:
try:
project_stack = ProjectStack.get(ProjectStack.stack == stack['id'],
ProjectStack.project == project)
except models.DoesNotExist:
project_stack = ProjectStack.create(project=project, stack = stack['id'])
probe_setting = stack['probeSettings']
probe_position = json.loads(probe_setting['position'])
ProbeSetting.create(x_offset=int(probe_position['xOffset']),
y_offset=int(probe_position['yOffset']),
project_stack=project_stack,
label=probe_position['label']
)
message = "Project updated"
except Exception as ex:
print ex
project = Project.get(Project.id == project_id)
return jsonify({ 'message': 'Project updated', 'project': project.deep_serialized() }), 200
