Python Tree.add_photo方法代码示例

# 需要导入模块: from treemap.models import Tree [as 别名]
# 或者: from treemap.models.Tree import add_photo [as 别名]
    def test_treephoto_overrides_tree_and_plot(self):
        tree = Tree(diameter=10, plot=self.plot, instance=self.instance)
        tree.save_with_user(self.user)
        tree.add_photo(self.image, self.other)

        self.clear_and_set_and_reload()
        self.assertEqual(self.plot.updated_by_id, self.other.pk)

# 需要导入模块: from treemap.models import Tree [as 别名]
# 或者: from treemap.models.Tree import add_photo [as 别名]
def add_tree_photo_helper(request, instance, feature_id, tree_id=None):
    plot = get_map_feature_or_404(feature_id, instance, 'Plot')
    tree_ids = [t.pk for t in plot.tree_set.all()]

    if tree_id and int(tree_id) in tree_ids:
        tree = Tree.objects.get(pk=tree_id)
    elif tree_id is None:
        # See if a tree already exists on this plot
        tree = plot.current_tree()

        if tree is None:
            # A tree doesn't exist, create a new tree create a
            # new tree, and attach it to this plot
            tree = Tree(plot=plot, instance=instance)

            # TODO: it is possible that a user has the ability to
            # 'create tree photos' but not trees. In this case we
            # raise an authorization exception here.
            # It is, however, possible to have both a pending
            # tree and a pending tree photo
            # This will be added later, when auth/admin work
            # correctly with this system
            tree.save_with_user(request.user)

    else:
        # Tree id is invalid or not in this plot
        raise Http404('Tree id %s not found on plot %s'
                      % (tree_id, feature_id))

    #TODO: Auth Error
    data = get_image_from_request(request)
    treephoto = tree.add_photo(data, request.user)

    return treephoto, tree

# 需要导入模块: from treemap.models import Tree [as 别名]
# 或者: from treemap.models.Tree import add_photo [as 别名]
    def test_treephoto_overrides_tree_and_plot_updated(self):
        tree = Tree(diameter=10, plot=self.plot, instance=self.instance)
        tree.save_with_user(self.user)
        tree.add_photo(self.image, self.user)

        tree_audit = self.max_audit_for_model_type('Tree')
        treephoto_audit = self.max_audit_for_model_type(['MapFeaturePhoto',
                                                         'TreePhoto'])
        plot_audit = self.max_audit_for_model_type('Plot')
        # Backdate the audits so photo it is definitely the newsest
        plot_audit.created = treephoto_audit.created - timedelta(days=2)
        plot_audit.save()
        tree_audit.created = treephoto_audit.created - timedelta(days=1)
        tree_audit.save()

        self.clear_and_set_and_reload()
        self.assertEqual(self.plot.updated_at, treephoto_audit.created)

# 需要导入模块: from treemap.models import Tree [as 别名]
# 或者: from treemap.models.Tree import add_photo [as 别名]
def add_tree_photo(request, instance, plot_id, tree_id=None):
    plot = get_object_or_404(Plot, pk=plot_id, instance=instance)
    tree_ids = [t.pk for t in plot.tree_set.all()]

    if tree_id and int(tree_id) in tree_ids:
        tree = Tree.objects.get(pk=tree_id)
    elif tree_id is None:
        # See if a tree already exists on this plot
        tree = plot.current_tree()

        if tree is None:
            # A tree doesn't exist, create a new tree create a
            # new tree, and attach it to this plot
            tree = Tree(plot=plot, instance=instance)

            # TODO: it is possible that a user has the ability to
            # 'create tree photos' but not trees. In this case we
            # raise an authorization exception here.
            # It is, however, possible to have both a pending
            # tree and a pending tree photo
            # This will be added later, when auth/admin work
            # correctly with this system
            tree.save_with_user(request.user)

    else:
        # Tree id is invalid or not in this plot
        raise Http404("Tree id %s not found on plot %s" % (tree_id, plot_id))

    # TODO: Validation Error
    # TODO: Auth Error
    if "file" in request.FILES:
        # TODO: Check size before reading
        data = request.FILES["file"].file
    else:
        data = request.body

    photo = tree.add_photo(data, request.user)

    return photo