populate_rango.py

import os
os.environ.setdefault('DJANGO_SETTINGS_MODULE','tangowithdjango.settings')
import django
django.setup()
from rango.models import Category, Page


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

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


def populate():
    python_pages = [{"title": "Official Python Tutorial",
                    "url":"http://docs.python.org/2/tutorial/"},

                    {"title": "How to Think like a Computer Scientist",
                    "url": "http://www.greenteapress.com/thinkpython/"},

                    {"title": "Learn Python in 10 Minutes",
                    "url": "http://www.korokithakis.net/tutorials/python/"}]

    django_pages = [{"title": "Official Django Tutorial",
                    "url":"https://docs.djangoproject.com/en/1.9/intro/tutorial01/"},

                    {"title": "Django Rocks",
                    "url": "http://www.djangorocks.com/"},

                    {"title": "How to Tango with Django",
                    "url": "http://www.tangowithdjango.com/"}]

    other_pages = [{"title": "Bottle",
                   "url":"http://bottlepy.org/docs/dev/"},

                   {"title": "Flask",
                    "url": "http://flask.pocoo.org"}]

    categories = {"Python": {"pages": python_pages, "views":128, "likes":64},
                  "Django": {"pages": django_pages, "views":64, "likes":32},
                  "Other Frameworks": {"pages": other_pages,"views":32,"likes":16}}

    for cat, cat_data in categories.items():

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

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




if __name__ == '__main__':
    print("starting to populate")
    populate()