populate_rango.py

import os
os.environ.setdefault('DJANGO_SETTINGS_MODULE',
                      'tango_with_django_project.settings')

import django
django.setup()
from rango.models import Category, Page

def populate():
    #Create lists of dictionaries containing the pages we want to add into each
    #category. Then create a dictionary opf dictionaries for our categories.

    python_pages= [
        {"title": "Official Python Tutorial",
         "url":"http://docs.python.org/2/tutorial/",
         "views" : 44},
        {"title":"How to Think like a Computer Scientist",
         "url":"http://www.greenteapress.com/thinkpython/",
         "views" : 65},
        {"title":"Learn Python in 10 Minutes",
         "url":"http://www.korokithakis.net/tutorials/python/",
         "views" : 26}]

    django_pages = [
        {"title":"Official Django Tutorial",
         "url":"https://docs.djangoproject.com/en/1.9/intro/tutorial01/",
         "views" : 23},
        {"title":"Django Rocks",
         "url":"http://www.djangorocks.com/",
         "views" : 24},
        {"title":"How to Tango with Django",
         "url":"http://www.tangowithdjango.com/",
         "views" : 11113}]

    other_pages = [
        {"title":"Bottle",
         "url":"http://bottlepy.org/docs/dev/",
         "views" : 4},
        {"title":"Flask",
         "url":"http://flask.pocoo.org",
         "views" : 223}]

    cats = {"Python": {"pages": python_pages, "views": 123, "likes": 65},
            "Django": {"pages": django_pages, "views": 65, "likes": 35},
            "Other Frameworks": {"pages": other_pages, "views": 36, "likes": 18} }


    for cat, cat_data in cats.items():
        c = add_cat(cat, views = cat_data["views"], likes = cat_data["likes"])
        for p in cat_data["pages"]:
            add_page(c, p["title"], p["url"], p["views"])

    for c in Category.objects.all():
        for p in Page.objects.filter(category=c):
            print("- {0} - {1}".format(str(c), str(p)))

def add_page(cat, title, url, views=0):
    p = Page.objects.get_or_create(category=cat, title=title, views= views)[0]
    p.url = url
    p.views = views
    p.save()
    return p

def add_cat(name, views=0, likes=0):
    c = Category.objects.get_or_create(name=name)[0]
    c.views = views
    c.likes = likes
    c.save()
    return c

if __name__ == '__main__':
    print("Starting Rango population script...")
    populate()