Gustavo Picon is sharing code with you

Bitbucket is a code hosting site. Unlimited public and private repositories. Free for small teams.

Don't show this again

tabo / django-treebeard

Efficient tree implementations for Django 1.0+ :: https://tabo.pe/projects/django-treebeard/

Clone this repository (size: 602.0 KB): HTTPS / SSH 
hg clone https://bitbucket.org/tabo/django-treebeard
hg clone ssh://hg@bitbucket.org/tabo/django-treebeard
Branch
1.6.X
  • 0f2a87d336ee
  • 145 loc
  • 4.7 KB
  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

import datetime
import random

from django import forms
from django.db import transaction
from django.http import HttpResponseRedirect
from django.db import connection
from django.shortcuts import render_to_response, get_object_or_404
from django.core.cache import cache

from tbexample.models import MP_Post, AL_Post, NS_Post
from tbexample.forms import CommentForm


def treetype2model(treetype):
    """ Returns the tree model for a given tree type from the url
    """
    return {'mp': MP_Post,
            'al': AL_Post,
            'ns': NS_Post}[treetype]


@transaction.commit_on_success
def load_random_data(request, treetype):
    """ Makes lots of dummy posts.
    """
    tbmodel = treetype2model(treetype)
    data = {'sql_queries': connection.queries,
            'treetype': treetype}

    key = 'treebeard_tbexample_loadfloodprot_%s' % treetype
    if cache.get(key):
        return render_to_response('tbexample/loadflood.html', data)
    cache.set(key, True, 60*5)

    interval = range(1,6)
    vals = {1: 4, 2: 10, 3: 25, 4: 50, 5: 50}
    for depth in interval:
        num = vals[depth]
        if depth > 1:
            if treetype == 'al':
                nodes = [obj for obj in tbmodel.objects.all()
                         if obj.get_depth() == depth-1]
            else:
                nodes = list(tbmodel.objects.filter(depth=depth-1))
        for i in range(num):
            if depth == 1:
                meth = tbmodel.add_root
            else:
                node = nodes[random.randint(0, len(nodes)-1)]
                if treetype == 'ns':
                    node = tbmodel.objects.get(pk=node.id)
                meth = node.add_child
            obj = meth(author='author_%d' % (i,),
                 comment='lorem ipsum! %d' % (random.randint(1000000000,
                                                             9999999999), ),
                 created=datetime.datetime.now())
    return render_to_response('tbexample/loaddata.html', data)


@transaction.commit_on_success
def delete_node(request, treetype, node_id):
    """ View to remove a message and it's replies
    """
    tbmodel = treetype2model(treetype)
    data = {'sql_queries': connection.queries,
            'treetype': treetype}
    node = get_object_or_404(tbmodel, id=node_id)
    node.delete()
    return render_to_response('tbexample/delete.html', data)


@transaction.commit_on_success
def delete_all(request, treetype):
    """ Remove all messages
    """
    tbmodel = treetype2model(treetype)
    data = {'sql_queries': connection.queries,
            'treetype': treetype}
    tbmodel.objects.all().delete()
    return render_to_response('tbexample/delete.html', data)


@transaction.commit_on_success
def convo(request, treetype, node_id=None):
    """ convo view~
    """
    tbmodel = treetype2model(treetype)
    data = {'sql_queries': connection.queries,
            'treetype': treetype}
    if node_id:
        root = get_object_or_404(tbmodel, id=node_id)
        if root.get_depth() != 1:
            # meh not really a root node, redirecting...
            return HttpResponseRedirect('%s#comment_%d' %
                (root.get_root().get_absolute_url(), root.id))
    else:
        root = None

    if request.method == 'POST':
        form = CommentForm(request.POST)
        form.root = root
        form.tbmodel = tbmodel
        if form.is_valid():
            obj = form.add_method(
                author=form.cleaned_data['author'],
                comment=form.cleaned_data['comment'],
                created=datetime.datetime.now())
            if not root:
                root = obj
            data['link'] = '%s#comment_%d' % \
                           (root.get_absolute_url(), obj.id)
            return render_to_response('tbexample/posted.html', data)
    else:
        form = CommentForm()
        form.root = root
        form.tbmodel = tbmodel

    data['form'] = form
    if node_id:
        template_html = 'tbexample/convo.html'
        descendants = root.get_descendants()
        nodes = [(root, len(descendants))] + [
            (node, node.get_children_count())
            for node in descendants
        ]
        data['mainpage'] = False
        data['nodes'] = nodes
        form.fields['parent'].widget = forms.TextInput()
    else:
        template_html = 'tbexample/main.html'
        data['mainpage'] = True
        data['nodes'] = [(node, node.descendants_count)
                         for node in tbmodel.get_descendants_group_count()]
        data['total_comments'] = len(data['nodes']) + \
                                 sum([count for _, count in data['nodes']])
        data['treetype'] = treetype
    return render_to_response(template_html, data)


def choose(request):
    return render_to_response('tbexample/choose.html', {})