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