add a script to generate app categories for Marketplace

This commit is contained in:
Chris Van 2013-02-15 17:58:43 -08:00
Родитель f5508e2ca3
Коммит 099e9bc8c6
2 изменённых файлов: 39 добавлений и 0 удалений

0
scripts/__init__.py Normal file
Просмотреть файл

Просмотреть файл

@ -0,0 +1,39 @@
import amo
from addons.models import AddonCategory, Category, Webapp
def run():
"""
Usage::
python -B manage.py runscript scripts.generate-mkt-cats
"""
cats = {
'books-reference': 'Books & Reference',
'education': 'Education',
'entertainment-sports': 'Entertainment & Sports',
'games': 'Games',
'health-fitness': 'Health & Fitness',
'lifestyle': 'Lifestyle',
'music': 'Music',
'news-weather': 'News & Weather',
'photos-media': 'Photos & Media',
'productivity': 'Productivity',
'shopping': 'Shopping'
}
for slug, name in cats.iteritems():
cat, created = Category.objects.get_or_create(type=amo.ADDON_WEBAPP,
slug=slug)
if created:
cat.name = name
cat.save()
print 'Created "%s" category' % name
try:
w = Webapp.objects.visible()[0]
except IndexError:
pass
else:
AddonCategory.objects.get_or_create(category=cat, addon=w)
print 'Added "%s" to "%s" category' % (w.name, name)