def post(self):
user = users.get_current_user()
new_idea = []
idea_name = self.request.get("idea-name")
if not user:
new_idea = Idea(parent=ndb.Key("Entity", "anon"), title=idea_name)
else:
new_idea = Idea(parent=ndb.Key("Entity",
user.email().lower()),
title=idea_name)
if (idea_name != ""):
new_idea.put()
self.redirect(self.request.referer)
def post(self):
"""Posts a new project idea."""
name = self.request.get('name')
description = self.request.get('description')
Idea(name=name,
description=description,
author=auth.current_user()).put()
return self.redirect('/ideas')
def setUp(self):
"""Set up a temp database before each test"""
basedir = os.path.abspath(os.path.dirname(__file__))
app.config['TESTING'] = True
self.test_db_path = os.path.join(basedir, TEST_DB)
app.config[
'SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + self.test_db_path
self.app = app.test_client()
db.create_all()
fixtures = [Idea(text=f'idea{i}') for i in range(5)]
db.session.bulk_save_objects(fixtures)
db.session.commit()
def create_idea():
"""Receives JSON and stores new Idea to db"""
idea = request.get_json()
new_idea = Idea(
idea_session_id = idea['session'],
name = idea['name'],
creator_id = g.user.id
)
db.session.add(new_idea)
db.session.commit()
idea['id'] = new_idea.id
idea['description'] = None
idea['creator_id'] = g.user.id
return _todo_response(idea)
def ideas():
if request.method == 'POST':
title = request.form.get('title')
description = request.form.get('description')
author = request.form.get('author')
category_id = request.form.get('category_id')
idea = Idea(title, description, author, category_id).save()
db.session.add(idea)
db.session.commit()
return redirect(url_for('show_idea', id=idea.id))
context = {'title': 'Ideas', 'ideas': Idea.query.all()}
return render_template('ideas/index.html', **context)
def validate_create_idea():
if "user_id" not in session:
return redirect("/")
print(request.form["idea_content"])
is_valid = True
if not request.form["idea_content"]:
is_valid = False
flash("You cannot post an empty idea.")
if len(request.form["idea_content"]) > 255:
flash("Ideas must be less than 255 characters.")
is_valid = False
if is_valid:
new_idea = Idea(content=request.form["idea_content"],
author_id=session["user_id"]["id"])
db.session.add(new_idea)
db.session.commit()
flash("Idea saved!")
return redirect("/userpage")
def create_idea():
if "user_id" not in session:
return redirect("/")
is_valid = True
if not request.form["idea_content"]:
is_valid = False
flash("You cannot post an empty idea.")
if len(request.form["idea_content"]) > 255:
flash("Ideas must be less than 255 characters.")
is_valid = False
if is_valid:
new_idea = Idea(content=request.form["idea_content"],
author_id=session["user_id"]["id"])
db.session.add(new_idea)
db.session.commit()
flash("Idea saved!")
user_data = User.query.filter_by(
user_id=session["user_id"]["id"]).all()
idea_data = Idea.query.order_by(desc(Idea.created_at)).join(User).all()
return render_template("/partials/idea_feed.html",
user_data=user_data[0],
idea_data=idea_data)
from app import db
from models import Idea
ideas = []
with open('ideas.txt') as f:
for line in f:
ideas.append(Idea(text=line.strip()))
s = db.session
s.bulk_save_objects(ideas)
s.commit()
