web/ldt/ldt_utils/admin.py
author ymh <ymh.work@gmail.com>
Fri, 15 Oct 2010 11:58:18 +0200
changeset 89 30c6e597a7de
parent 67 90fd14c649bb
permissions -rw-r--r--
correct project serializer and take into account the display + new version
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     1
from django.contrib import admin
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     2
from django.conf.urls.defaults import *
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     3
from django.shortcuts import render_to_response
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     4
from django.template import RequestContext
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     5
from django.conf import settings
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     6
from models import *
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     7
from forms import *
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     8
from fileimport import *
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
     9
from ldt.ldt_utils import STORE
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    10
from ldt.ldt_utils import ANALYZER
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    11
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    12
import lucene
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    13
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    14
admin.site.register(Project)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    15
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    16
class ContentAdmin(admin.ModelAdmin):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    17
    
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    18
    def import_file(self, request):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    19
        if request.method =='POST':
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    20
            form = LdtImportForm(request.POST, request.FILES)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    21
            if form.is_valid():
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    22
                filetoprocess =form.cleaned_data['importFile']
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    23
                flatten = form.cleaned_data['flatten']
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    24
                videoPath = form.cleaned_data['videoPath']
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    25
                # fi = None
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    26
                fi = FileImport(filetoprocess, videoPath, flatten)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    27
                try:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    28
                    fi.processFile()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    29
                    args = {'message': "File imported"}
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    30
                except FileImportError:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    31
                    non_field_errors = form.non_field_errors()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    32
                    non_field_errors.append("Error when importing : unknown file type")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    33
                    form._errors["__all__"] = non_field_errors
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    34
                    args = {'message': "Can not import file, unknown file type", 'form': form}
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    35
                    
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    36
            else:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    37
                non_field_errors = form.non_field_errors()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    38
                non_field_errors.append("Error when importing : invalid form")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    39
                form._errors["__all__"] = non_field_errors
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    40
                args = {'message': "Error when importing : invalid form", 'form': form}
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    41
        else:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    42
            form = LdtImportForm()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    43
            args = {'form': form, 'current_app': self.admin_site.name, 'current_action' : 'import_file'}
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    44
        return render_to_response('admin/ldt_utils/content/upload_form.html', args, context_instance=RequestContext(request))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    45
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    46
    def reindex(self, request):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    47
        message = None
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    48
        if request.method == "POST":
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    49
            form = ReindexForm(request.POST)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    50
            if form.is_valid():    
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    51
                # try:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    52
                writer = lucene.IndexWriter(STORE, ANALYZER, True, lucene.IndexWriter.MaxFieldLength.UNLIMITED)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    53
                contentList = form.cleaned_data["contents"]
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    54
                indexer = ContentIndexer(contentList,writer)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    55
                indexer.index_all()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    56
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    57
                writer.close()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    58
                message = "Indexation ok : " + repr(form.cleaned_data["contents"]) 
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    59
                form = ReindexForm()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    60
                # except Exception, inst:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    61
                    # non_field_errors = form.non_field_errors()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    62
                    # non_field_errors.append("Error when reindexing : " + cgi.escape(repr(inst)))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    63
                    # form._errors["__all__"] = non_field_errors
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    64
                    #message = "ERROR : " + repr(non_field_errors)
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    65
        else:
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    66
            form = ReindexForm()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    67
    
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    68
        return render_to_response('admin/ldt_utils/content/reindex_form.html', {'form': form, 'message':message, 'current_app': self.admin_site.name, 'current_action' : 'reindex' }, context_instance=RequestContext(request))
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    69
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    70
    def get_urls(self):
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    71
        urls = super(ContentAdmin, self).get_urls()
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    72
        content_urls = patterns('',
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    73
            url(r'^reindex/$', self.admin_site.admin_view(self.reindex), name="ldt_content_reindex"),
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    74
    # (r'^admin/ldt/content/import/upload/$', 'ldt.ldt_utils.views.uploadFile'),                                
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    75
            url(r'^import/$', self.admin_site.admin_view(self.import_file), name="ldt_content_import_file")
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    76
        )
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    77
        return content_urls + urls
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    78
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    79
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    80
admin.site.register(Content, ContentAdmin)
67
90fd14c649bb add edition for content
ymh <ymh.work@gmail.com>
parents: 0
diff changeset
    81
admin.site.register(Media)
0
cc4a51750724 first commit
ymh <ymh.work@gmail.com>
parents:
diff changeset
    82
admin.site.register(Author)