def add_to_db(self, entry, media_folder):
new_story = Story()
if (entry[6].lower) in ['audio', 'video', 'image', 'pdf']:
try:
new_story.media_file = File(open(media_folder + "/" + entry[0]))
except IOError:
self.stderr.write(RED_C + "Media %s not found for story %s." % (entry[0], entry[1]) + \
DEFAULT_C + "\n" )
return
else:
if entry[6].lower() == "link":
new_story.link_url = entry[0]
new_story.title = entry[1]
new_story.year = entry[2]
try:
new_story.branch = Branch.objects.get(name=entry[3])
except ObjectDoesNotExist:
self.stderr.write(RED_C + "Branch %s not found for story %s." %
(entry[3], new_story.title) + DEFAULT_C + "\n")
return
new_story.user = User.objects.get(pk=1)
new_story.content_type = CONTENT_TYPE[entry[6].lower()]
new_story.description = entry[7]
new_story.public_approved = True
print new_story
new_story.save()
for keyword in entry[5].split(','):
new_story.keywords.add(keyword)
