X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/153bc4288d5941dafcaca24fe1f4859419d744e0..2ae6c6bc078cdc7ec6e3131bbbb430714bf36230:/apps/catalogue/management/commands/importbooks.py diff --git a/apps/catalogue/management/commands/importbooks.py b/apps/catalogue/management/commands/importbooks.py index c1a743077..323755e58 100644 --- a/apps/catalogue/management/commands/importbooks.py +++ b/apps/catalogue/management/commands/importbooks.py @@ -10,16 +10,16 @@ from django.conf import settings from django.core.management.base import BaseCommand from django.core.management.color import color_style from django.core.files import File +from catalogue.utils import trim_query_log from wolnelektury_core.management.profile import profile -import objgraph -import gc from catalogue.models import Book from picture.models import Picture from search import Index + class Command(BaseCommand): option_list = BaseCommand.option_list + ( make_option('-q', '--quiet', action='store_false', dest='verbose', default=True, @@ -109,6 +109,7 @@ class Command(BaseCommand): files = sorted(os.listdir(dir_name)) postponed = {} while files: + trim_query_log(0) file_name = files.pop(0) file_path = os.path.join(dir_name, file_name) file_base, ext = os.path.splitext(file_path) @@ -128,14 +129,10 @@ class Command(BaseCommand): if import_picture: self.import_picture(file_path, options) else: - objgraph.show_growth() self.import_book(file_path, options) - objgraph.show_growth() - print "--------------------" - + files_imported += 1 transaction.commit() - ## track. except (Book.AlreadyExists, Picture.AlreadyExists): print self.style.ERROR('%s: Book or Picture already imported. Skipping. To overwrite use --force.' %