scripts/import_links.py

   1 import sys
   2 sys.path.insert(0, '../apps')
   3 sys.path.insert(0, '../lib')
   4 sys.path.insert(0, '../wolnelektury')
   5 sys.path.insert(0, '..')
   6
   7 from django.core.management import setup_environ
   8 from wolnelektury import settings
   9 import sys
  10
  11 setup_environ(settings)
  12
  13 from catalogue.models import Book, Tag
  14
  15
  16 def import_links(file_name, attribute):
  17     for line in file(file_name):
  18         slug, link = line.split()
  19         link = link.strip('\n')
  20         try:
  21             book = Book.objects.get(slug=slug)
  22             setattr(book, attribute, link)
  23             book.save()
  24             print 'Link %s for book %s added!' % (link, book)
  25         except Book.DoesNotExist:
  26             try:
  27                 tag = Tag.objects.get(slug=slug)
  28                 setattr(tag, attribute, link)
  29                 tag.save()
  30                 print 'Link %s for tag %s added!' % (link, tag)
  31             except Tag.DoesNotExist:
  32                 print 'Invalid slug %s!' % slug
  33
  34
  35 import_links('gazeta-links', 'gazeta_link')
  36 import_links('wiki-links', 'wiki_link')