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