# HG changeset patch # User bellierp # Date 1493203436 -7200 # Node ID 63af78d78387f4ca2b418c51d9235db88e98636b # Parent 34fec425afc79b42341b749278b174c8b009bcae corrections when multiple projects returned diff -r 34fec425afc7 -r 63af78d78387 src/ldtplatform/management/commands/importweb.py --- a/src/ldtplatform/management/commands/importweb.py Fri Apr 21 18:27:32 2017 +0200 +++ b/src/ldtplatform/management/commands/importweb.py Wed Apr 26 12:43:56 2017 +0200 @@ -146,17 +146,15 @@ videopath = '' tag = create_tag(data[3]) try: - mymedia = models.Media.objects.get(src=mysource) - mycontent = models.Content.objects.get(media_obj_id=mymedia.id) - myproject = models.Project.objects.get(title='front project : %s' % title) + media = models.Media.objects.get(src=mysource) + content = models.Content.objects.get(media_obj_id=media.id) + continue except (models.Media.MultipleObjectsReturned, - models.Content.MultipleObjectsReturned, - models.Project.MultipleObjectsReturned + models.Content.MultipleObjectsReturned ): continue except (models.Media.DoesNotExist, - models.Content.DoesNotExist, - models.Project.DoesNotExist + models.Content.DoesNotExist ): if requests.head(mysource).status_code == 200: management.call_command( @@ -171,14 +169,18 @@ update=True, tags=tag.encode('utf-8') if tag else '' ) - myfrontproj = models.Project.objects.get(title='front project : %s' % title) - myfrontproj.ldt = self.change_annotations(ldtproject, myfrontproj.ldt) - myfrontproj.save() - self.stdout.write("%s done"%title) - self.writefile.writerow([mysource, - models.Content.objects.get(title=title).iri_id, - title.encode('utf-8') if title else '', - myfrontproj.ldt_id, - ]) + media = models.Media.objects.get(src=mysource) + content = models.Content.objects.get(media_obj_id=media.id) + myfrontproj = models.Project.objects.get(id=content.front_project.id) + myfrontproj.ldt = self.change_annotations(ldtproject, myfrontproj.ldt) + myfrontproj.save() + self.stdout.write("%s done"%title) + self.writefile.writerow([mysource, + content.iri_id, + title.encode('utf-8') if title else '', + myfrontproj.ldt_id, + ]) + else: + self.stdout.write("Source %s doesn't return 200"%mysource) csvfile.close() csvfile2.close()