From a0704a73e68c71973b07406da56707231c58846c Mon Sep 17 00:00:00 2001 From: Guillaume Pellerin Date: Fri, 22 May 2015 19:17:05 +0200 Subject: [PATCH] fix paths --- .../telemeta-import-corpus-from-dir.py | 32 ++++++++++--------- 1 file changed, 17 insertions(+), 15 deletions(-) diff --git a/telemeta/management/commands/telemeta-import-corpus-from-dir.py b/telemeta/management/commands/telemeta-import-corpus-from-dir.py index 30a84bf3..23f8336b 100644 --- a/telemeta/management/commands/telemeta-import-corpus-from-dir.py +++ b/telemeta/management/commands/telemeta-import-corpus-from-dir.py @@ -5,7 +5,7 @@ from django.core.files.base import ContentFile from django.contrib.auth.models import User from telemeta.models import * from telemeta.util.unaccent import unaccent -import os, re +import os, re, glob try: from django.utils.text import slugify @@ -113,26 +113,28 @@ class Command(BaseCommand): metadata[data[0]] = data[1:] i += 1 print metadata - collection_name = chapter - collection_id = corpus_id + '_' + slugify(unicode(collection_name)) - collection_title = collection_name.replace('_', ' ') + ' - ' + chapter_title - print collection_title - cc = MediaCollection.objects.filter(code=collection_id, title=collection_title) - if cc: - collection = cc[0] - else: - collection = MediaCollection(code=collection_id) - collection.title = collection_title - collection.save() - if not collection in corpus.children.all(): - corpus.children.add(collection) + collection_name = chapter + collection_id = corpus_id + '_' + slugify(unicode(collection_name)) + collection_title = collection_name.replace('_', ' ') + ' - ' + chapter_title + print collection_title + cc = MediaCollection.objects.filter(code=collection_id, title=collection_title) + if cc: + collection = cc[0] + else: + collection = MediaCollection(code=collection_id) + collection.title = collection_title + collection.save() + if not collection in corpus.children.all(): + corpus.children.add(collection) + + for filename in os.listdir(chapter_dir): + path = os.path.join(chapter_dir, filename) if os.path.isfile(path) and '.jpg' == os.path.splitext(filename)[1]: related_path = path.replace(self.media_root, '') related, c = MediaCollectionRelated.objects.get_or_create(collection=collection, file=related_path) - for root, dirs, files in os.walk(chapter_dir): for media_file in files: path = os.path.join(root, media_file) -- 2.39.5