def save_document(provider, title, href, images, embed, desc, duration, views, categories):
from documents.models import Document, Category, Image
try:
doc = Document.objects.get(href=href)
except Document.DoesNotExist:
doc = Document()
doc.provider = provider
doc.title = title
doc.href = href
doc.embed = embed
doc.desc = desc
doc.duration = duration
locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')
doc.views = locale.atoi(views)
doc.save()
if doc.images.count() == 0:
for image_url in images:
response = requests.get(image_url)
if response.status_code == 200:
image = Image()
image.document = doc
image.url = image_url
image.save()
for category in categories:
cat = Category.objects.get_or_create(title=category)
doc.categories.add(cat[0])
