Пример #1
0
def handle_related_models(object_list, parent, clean, **kwargs):
    """ Create/find and attach all related models for a piece.
    :param object_list: The list of objects to create/find, and which
     field they belong to, in the format [{id: field_name, value: x},]
    :param parent: The object to relate to.
    :param clean: A Cleanup object.
    :kwargs
            creator: the creator of the piece, taken from request.user
            birth_date: The birth-date of a composer object
            death_date: The death-date of a composer object
    """
    for item in object_list:
        field = item.get("id")
        if field == "title":
            parent.title = item.get("value")
            continue
        if field == "composer":
            birth_date = kwargs.pop("birth_date", None)
            death_date = kwargs.pop("death_date", None)
            try:
                composer_list = abstract_model_factory(
                    item.get("value"), "Composer", clean, birth_date=birth_date, death_date=death_date
                )
                composer = composer_list[0]
                parent.composer = composer
                continue
            except:
                clean.cleanup()
                raise
        if field == "collections":
            user = kwargs.pop("user", None)
            try:
                collection_list = abstract_model_factory(item.get("value"), "Collection", clean, creator=user)
                for x in collection_list:
                    x.add(parent)
                continue
            except:
                clean.cleanup()
                raise
        if field == "languages":
            try:
                language_list = abstract_model_factory(item.get("value"), "Language", clean)
                parent.languages.clear()
                for x in language_list:
                    parent.languages.add(x)
                continue
            except:
                clean.cleanup()
                raise
        if field == "locations":
            try:
                location_list = abstract_model_factory(item.get("value"), "Location", clean)
                parent.locations.clear()
                for x in location_list:
                    parent.locations.add(x)
                continue
            except:
                clean.cleanup()
                raise
        if field == "sources":
            try:
                source_list = abstract_model_factory(item.get("value"), "Source", clean)
                parent.sources.clear()
                for x in source_list:
                    parent.sources.add(x)
                continue
            except:
                clean.cleanup()
                raise
        if field == "tags":
            try:
                tag_list = abstract_model_factory(item.get("value"), "Tag", clean)
                parent.tags.clear()
                for x in tag_list:
                    parent.tags.add(x)
                continue
            except:
                clean.cleanup()
                raise
        if field == "genres":
            try:
                genre_list = abstract_model_factory(item.get("value"), "Genre", clean)
                parent.genres.clear()
                for x in genre_list:
                    parent.genres.add(x)
                continue
            except:
                clean.cleanup()
                raise
        if field == "instruments_voices":
            try:
                ins_list = abstract_model_factory(item.get("value"), "InstrumentVoice", clean)
                parent.instruments_voices.clear()
                for x in ins_list:
                    parent.instruments_voices.add(x)
                continue
            except:
                clean.cleanup()
                raise
        if field == "number_of_voices":
            parent.number_of_voices = int(item.get("value"))
        if field == "vocalization":
            parent.vocalization = item.get("value")
        if field == "religiosity":
            parent.religiosity = item.get("value")
        if field == "composition_start_date" and item.get("value"):
            parent.composition_start_date = int(item.get("value"))
        if field == "composition_end_date" and item.get("value"):
            parent.composition_end_date = int(item.get("value"))
        if field == "comment":
            parent.comment = item.get("value")
Пример #2
0
def piece_update(request, *args, **kwargs):
    # Update a piece based on a dict of changes in request.POST['changes']
    patch_data = request.data
    form = validate_dynamic_piece_form(request, PieceForm(patch_data))
    if not form.is_valid():
        # Form errors are rendered for user on the front end. Collection
        # validation errors are ignored, as these cannot be modified from
        # the update page.
        if form.errors.get("collections"):
            del form.errors["collections"]
        if form.errors:
            errors = {f: [e.get_json_data()[0]["message"]] for f, e in form.errors.items()}
            errors = json.dumps({"errors": errors})
            return HttpResponse(content=errors, content_type="application/json", status=status.HTTP_400_BAD_REQUEST)

    clean = Cleanup()
    piece = Piece.objects.get(id=int(kwargs["pk"]))
    change = json.loads(patch_data["changes"])

    """
    Creating new movements must occur before old movements are deleted,
    as the attachment of files depends on the ordering and numbering
    of movements in the piece's present state. See comment in
    handle_dynamic_file_table() for more information.
    """
    handle_dynamic_file_table(request, piece, clean)

    modify_movements = [x for x in change["modify"] if x["type"] == "M"]
    if modify_movements:
        for item in modify_movements:
            mov = Movement.objects.filter(id=item["id"])
            if not mov:
                break
            mov = mov[0]
            if item.get("tags"):
                tag_list = abstract_model_factory(item["tags"], "Tag", clean)
                mov.tags.clear()
                for x in tag_list:
                    mov.tags.add(x)
            if item.get("instruments_voices"):
                ins_list = abstract_model_factory(item["instruments_voices"], "InstrumentVoice", clean)
                mov.instruments_voices.clear()
                for x in ins_list:
                    mov.instruments_voices.add(x)
            if item.get("comment"):
                mov.comment = item["comment"]
            if item.get("number_of_voices"):
                mov.number_of_voices = int(item["number_of_voices"])
            if item.get("vocalization"):
                mov.vocalization = item["vocalization"]
            if item.get("title"):
                mov.title = item["title"]

            mov.save()

    movement_positions = [x for x in change["modify"] if x["type"] == "M" and x.get("position")]
    movement_positions.extend([x for x in change["add"] if x["type"] == "M" and x.get("position")])
    if movement_positions:
        movements = piece.movements.all()
        for item in movement_positions:
            if item.get("id"):
                mov = movements.get(id=item["id"])
                mov.position = item["position"]
                mov.save()
            elif item.get("name"):
                mov = movements.get(title=item["name"])
                mov.position = item["position"]
                mov.save()

    modify_atts = [x for x in change["modify"] if x["type"] == "A"]
    if modify_atts:
        for item in modify_atts:
            att = Attachment.objects.filter(pk=item["id"])
            if not att:
                break
            att = att[0]
            if item.get("parent"):
                if item.get("newParentTitle") == "Attach to Piece":
                    att.movements.clear()
                    att.pieces.clear()
                    att.pieces.add(piece)
                    att.save()
                else:
                    mov = piece.movements.filter(title=item["newParentTitle"])
                    if not mov:
                        break
                    mov = mov[0]
                    att.movements.clear()
                    att.pieces.clear()
                    att.movements.add(mov)
                    att.save()
            if item.get("source"):
                att.source = item.get("source")
                att.save()

    modify_piece = [x for x in change["modify"] if x["type"] == "F"]
    if modify_piece:
        # Can use the same function as the piece-create here.
        handle_related_models(modify_piece, piece, clean)

    delete_attachments = [x for x in change["delete"] if x["type"] == "A"]
    if delete_attachments:
        for item in delete_attachments:
            att = Attachment.objects.filter(pk=item["id"])[0]
            if att:
                att.delete()

    delete_movements = [x for x in change["delete"] if x["type"] == "M"]
    if delete_movements:
        for item in delete_movements:
            mov = Movement.objects.filter(pk=item["id"])[0]
            if mov:
                mov.delete()

    piece.save()
    rebuild_suggester_dicts.delay()
    data = json.dumps({"success": True, "id": piece.id, "url": "/piece/{0}".format(piece.id)})
    return HttpResponse(data, content_type="json")