132 lines
4.8 KiB
Python
132 lines
4.8 KiB
Python
from datetime import datetime
|
|
from xml.sax import SAXParseException
|
|
|
|
from django.contrib import messages
|
|
from django.core.urlresolvers import reverse, reverse_lazy
|
|
from django.db.models import get_model
|
|
from django.http import HttpResponseRedirect, HttpResponse, HttpResponseNotFound
|
|
from django.views.generic.edit import FormView
|
|
|
|
from . import models
|
|
from . import forms
|
|
|
|
PREAMBLE = ('<?xml version="1.0" encoding="UTF-8" standalone="yes"?>'
|
|
'<collection '
|
|
'xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" '
|
|
'xmlns="http://www.loc.gov/MARC21/slim" '
|
|
'xsi:schemaLocation="http://www.loc.gov/MARC21/slim '
|
|
'http://www.loc.gov/standards/marcxml/schema/'
|
|
'MARC21slim.xsd">')
|
|
|
|
Edition = get_model(*models.EDITION_MODEL.split('.'))
|
|
|
|
|
|
def marc_records(request, selected_records=None):
|
|
# TODO convert to streaming when we move to Django 1.5+
|
|
if request.method == 'POST':
|
|
params=request.POST
|
|
elif request.method == 'GET':
|
|
params=request.GET
|
|
else:
|
|
return HttpResponseNotFound
|
|
format = params.get('format', 'xml')
|
|
link_target = params.get('link_target', 'via')
|
|
|
|
if selected_records==None:
|
|
# extract the user-selected records from the params QueryDict
|
|
selected_records = list(
|
|
k for k in params if params[k] == u'on'
|
|
)
|
|
if not selected_records:
|
|
messages.error(
|
|
request,
|
|
"There were no records to download."
|
|
)
|
|
return HttpResponseRedirect(reverse('marc', args=[]))
|
|
|
|
outfile = HttpResponse('', content_type='application/marc')
|
|
outfile['Content-Disposition'] = 'attachment; filename='+ datetime.now().strftime('%Y%m%d%H%M%S') + '.' + format
|
|
|
|
if format == 'xml':
|
|
outfile.write(PREAMBLE)
|
|
|
|
for record_name in selected_records:
|
|
if isinstance(record_name, models.MARCRecord):
|
|
record = record_name
|
|
elif isinstance(record_name, Edition):
|
|
record = models.MARCRecord(edition=record_name)
|
|
elif hasattr(record_name, 'edition'):
|
|
record = models.MARCRecord(edition=record_name.edition)
|
|
elif record_name.startswith('edition_'):
|
|
record_id = long(record_name[8:])
|
|
try:
|
|
edition = Edition.objects.get(pk=record_id)
|
|
except Edition.DoesNotExist:
|
|
continue
|
|
record = models.MARCRecord(edition=edition)
|
|
elif record_name.startswith('record_'):
|
|
record_id = long(record_name[7:])
|
|
try:
|
|
record = models.MARCRecord.objects.get(id=record_id )
|
|
except models.MARCRecord.DoesNotExist:
|
|
continue
|
|
|
|
outfile.write(record.record(format=format, link_target=link_target))
|
|
|
|
if format == 'xml':
|
|
outfile.write('</collection>')
|
|
|
|
return outfile
|
|
|
|
def all_marc_records(request):
|
|
selected_records = list(
|
|
'record_'+ str(record.pk) for record in models.MARCRecord.objects.all()
|
|
)
|
|
return marc_records(request, selected_records=selected_records)
|
|
|
|
def qs_marc_records(request, qs):
|
|
# translate a queryset into a record list.
|
|
selected_records = []
|
|
for obj in qs:
|
|
# each object must have a marc_records() attribute which returns a record list.
|
|
# records can be strings: 'record_xxx', 'edition_xxx', MARCRecord objs or objs with edition attributes
|
|
selected_records.extend(obj.marc_records())
|
|
return marc_records(request, selected_records=selected_records)
|
|
|
|
class MARCUpload(FormView):
|
|
template_name = 'marc/upload.html'
|
|
form_class = forms.MARCUploadForm
|
|
success_url = reverse_lazy('upload_marc')
|
|
|
|
# allow a get param to specify the edition
|
|
def get_initial(self):
|
|
if self.request.method == 'GET':
|
|
edition = self.request.GET.get('edition',None)
|
|
if Edition.objects.filter(id=edition).count():
|
|
edition = Edition.objects.filter(id=edition)[0]
|
|
if edition.ebooks.count() or edition.ebook_files.count():
|
|
return {'edition':edition.id}
|
|
return {}
|
|
|
|
def form_valid(self, form):
|
|
edition = form.cleaned_data['edition']
|
|
source = form.cleaned_data['source']
|
|
try:
|
|
marcfile=self.request.FILES['file']
|
|
new_record = models.MARCRecord(
|
|
guts=marcfile,
|
|
edition=edition,
|
|
user= self.request.user
|
|
)
|
|
new_record.load_from_file(source)
|
|
messages.success(
|
|
self.request,
|
|
"You have successfully added a MARC record. Hooray! Add another?"
|
|
)
|
|
except SAXParseException:
|
|
messages.error(
|
|
self.request,
|
|
"Sorry, couldn't parse that file."
|
|
)
|
|
return super(MARCUpload,self).form_valid(form)
|
|
|