32 lines
No EOL
945 B
Python
32 lines
No EOL
945 B
Python
|
|
import os
|
|
|
|
from django.conf import settings
|
|
from django.contrib import admin
|
|
|
|
from intranet.models import IntranetDoc
|
|
|
|
# Register your models here.
|
|
|
|
|
|
class IntranetDocAdmin(admin.ModelAdmin):
|
|
|
|
list_display = ('__str__', 'module', 'authorization')
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
searched_file = 'intranet/{0}'.format(form.cleaned_data['doc'])
|
|
try:
|
|
doc = IntranetDoc.objects.get(doc=searched_file)
|
|
form.save()
|
|
# Override previous file
|
|
filename = os.path.join(settings.MEDIA_ROOT, searched_file)
|
|
file = request.FILES['doc']
|
|
with open(filename, 'wb+') as destination:
|
|
for chunk in file.chunks():
|
|
destination.write(chunk)
|
|
|
|
except IntranetDoc.DoesNotExist:
|
|
super(IntranetDocAdmin, self).save_model(request, obj, form, change)
|
|
|
|
|
|
admin.site.register(IntranetDoc, IntranetDocAdmin) |