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
def project_stack_images(project_id, stack_id): first_image = Image.select().join(StackImage).where(StackImage.stack == stack_id).get().key last_underscore = first_image.rfind('_') stack_title = first_image[:last_underscore] project_stack = ProjectStack.get(ProjectStack.project == project_id, ProjectStack.stack == stack_id) project_stack_user_data = project_stack.project_stack_data.where(ProjectStackDatum.user == g.user_id) probe_type = project_stack.project.probe_type probe_settings = project_stack.probe_setting.get() images = (Image.select().join(StackImage).where(StackImage.stack == stack_id) .order_by(Image.name)) project_stacks = Stack.select().join(ProjectStack).where(ProjectStack.project == project_id).order_by(Stack.id) marker_options = MarkerOption.select().join(ProjectMarkerOption).where(ProjectMarkerOption.project == project_id) return jsonify({ 'images': [image.stack_image_serialized for image in images], 'data': [d.serialized for d in project_stack_user_data], 'probeSettings': probe_settings.serialized, 'stackTitle': stack_title, 'probeType': probe_type, 'markerOptions': [marker_option.name for marker_option in marker_options], 'projectStackIds': [stack.id for stack in project_stacks] })