This repository has been archived by the owner on Apr 13, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
populate_rango.py
65 lines (51 loc) · 2.07 KB
/
populate_rango.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
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():
python_pages = [
{'title': 'Official Python Tutorial',
'url': 'http://docs.python.org/3/tutorial/', 'views' : 42},
{'title': 'How to Think like a Computer Scientist',
'url': 'http://www.greenteapress.com/thinkpython/', 'views' : 46},
{'title': 'Learn Python in 10 Minutes',
'url': 'http://www.korokithakis.net/tutorials/python/', 'views' : 24}]
django_pages = [
{'title':'Official Django Tutorial',
'url':'https://docs.djangoproject.com/en/2.1/intro/tutorial01/','views':9},
{'title':'Django Rocks',
'url':'http://www.djangorocks.com/', 'views': 10},
{'title':'How to Tango with Django',
'url':'http://www.tangowithdjango.com/','views': 25}]
other_pages = [
{'title':'Bottle',
'url':'http://bottlepy.org/docs/dev/', 'views': 105},
{'title':'Flask',
'url':'http://flask.pocoo.org', 'views': 23}]
cats = {'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 cats.items():
c = add_cat(cat, cat_data['views'], 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(f'- {c}: {p}')
def add_page(cat, title, url, views):
p = Page.objects.get_or_create(category=cat, title=title)[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()