def init_db():
sites = Site.objects()
sites.delete()
# Should probably add something for setting up the site
title = raw_input("What are you naming this site?\n")
domain = raw_input("What is the site's domain?\n")
motto = raw_input("What is the site's motto? (optional)\n")
logo = raw_input("What is the relative path to the logo?\n \
ex: '/static/uploads/brainBadge.png' \n")
site = Site(title=title, motto=motto, domain=domain, logo=logo)
try:
site.save()
except:
print "Sorry, something went wrong..."
userreponse = raw_input("try again? [Y,n]")
if userresponse == 'Y' or userresponse == 'y' or userresponse == '\n':
init_db()
depends = [
{ 'title' : u'Flask',
'url' : u'http://flask.pocoo.org/',
'imgurl': u'http://flask.pocoo.org/static/logo.png',
'authors': ['Armin Ronacher'],
},
{ 'title' : u'jQuery',
'url' : u'http://jquery.com/',
'imgurl':
u'http://static.jquery.com/files/rocker/images/logo_jquery_215x53.gif',
'authors': ['John Resig'],
},
]
for dep in depends:
dependency =\
Dependency(title=dep['title'],authors=dep['authors'],url=dep['url'],\
imgurl=dep['imgurl'])
try:
dependency.save()
except:
flash("problem")
def reset_db():
try:
posts = Post.objects()
sites = Site.objects()
users = User.objects()
images = Image.objects()
deps = Dependency.objects()
flatpages = FlatPage.objects()
posts.delete()
sites.delete()
users.delete()
deps.delete()
flatpages.delete()
return "all images, posts, sites, and users deleted successfully!"
except:
print "There was an error..."