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