From 964ba554d569b120be2203e37f9199aa7b893480 Mon Sep 17 00:00:00 2001 From: yomguy Date: Thu, 19 Jul 2012 13:50:05 +0200 Subject: [PATCH] add cleanup --- .../commands/teleforma-import-conferences.py | 21 +++++++++++-------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/teleforma/management/commands/teleforma-import-conferences.py b/teleforma/management/commands/teleforma-import-conferences.py index 6ff5e549..1025092c 100644 --- a/teleforma/management/commands/teleforma-import-conferences.py +++ b/teleforma/management/commands/teleforma-import-conferences.py @@ -35,6 +35,15 @@ class Command(BaseCommand): hdlr.setFormatter(formatter) logger.addHandler(hdlr) + + def cleanup(self): + medias = Media.objects.all() + for media in medias: + media.delete() + items = MediaItem.objects.all() + for item in items: + item.delete() + def handle(self, *args, **options): organization_name = args[0] organization = Organization.objects.get(name=organization_name) @@ -43,14 +52,8 @@ class Command(BaseCommand): all_conferences = Conference.objects.all() i = 1 - #FIXME: -# medias = Media.objects.all() -# for media in medias: -# media.delete() -# items = MediaItem.objects.all() -# for item in items: -# item.delete() - + self.cleanup() + for root, dirs, files in os.walk(self.media_dir): for filename in files: name = os.path.splitext(filename)[0] @@ -75,8 +78,8 @@ class Command(BaseCommand): exist = False medias = conference.media.all() for media in medias: - print media.item.file if media.item.file == path: + print path exist = True break -- 2.39.5