Add out of process bulk document upload
This commit is contained in:
48
apps/sources/management/commands/bulk_upload.py
Normal file
48
apps/sources/management/commands/bulk_upload.py
Normal file
@@ -0,0 +1,48 @@
|
||||
from __future__ import absolute_import
|
||||
|
||||
import os
|
||||
from optparse import make_option
|
||||
|
||||
from django.core.management.base import BaseCommand, CommandError, LabelCommand
|
||||
|
||||
from ...models import OutOfProcess
|
||||
from ...compressed_file import CompressedFile, NotACompressedFile
|
||||
|
||||
|
||||
class Command(LabelCommand):
|
||||
args = '<filename>'
|
||||
help = 'Upload documents from a compressed file in to the database.'
|
||||
option_list = LabelCommand.option_list + (
|
||||
make_option('--noinput', action='store_false', dest='interactive',
|
||||
default=True, help='Do not ask the user for confirmation before '
|
||||
'starting.'),
|
||||
#make_option('--metadata', action='store', dest='metadata',
|
||||
# help='A metadata dictionary to apply to the documents.'),
|
||||
)
|
||||
|
||||
def handle_label(self, label, **options):
|
||||
if not os.access(label, os.R_OK):
|
||||
raise CommandError("File '%s' is not readable." % label)
|
||||
|
||||
if _confirm(options['interactive']) == 'yes':
|
||||
print 'Beginning upload...'
|
||||
fd = open(label)
|
||||
source = OutOfProcess()
|
||||
try:
|
||||
result = source.upload_file(fd, filename=None, use_file_name=False, document_type=None, expand=True, metadata_dict_list=None, user=None, document=None, new_version_data=None, verbose=True)
|
||||
except NotACompressedFile:
|
||||
print '%s is not a compressed file.'
|
||||
else:
|
||||
print 'Finished.'
|
||||
|
||||
fd.close()
|
||||
else:
|
||||
print 'Cancelled.'
|
||||
|
||||
|
||||
def _confirm(interactive):
|
||||
if not interactive:
|
||||
return 'yes'
|
||||
return raw_input('You have requested to bulk upload a number of documents from a compressed file.\n'
|
||||
'Are you sure you want to do this?\n'
|
||||
'Type \'yes\' to continue, or any other value to cancel: ')
|
||||
Reference in New Issue
Block a user