def delete_project(id):
project = UserProjects.get_project(id)
if not project:
abort(404)
try:
UserProjects.delete_project(project)
except:
flash("Error when deleting project")
return redirect(url_for("account", user=session["user"]))
return redirect(url_for("account", username=session["user"]))
def delete_project(id):
project = UserProjects.get_project(id)
if not project:
abort(404)
try:
UserProjects.delete_project(project)
except:
flash("Error when deleting project")
return redirect(url_for("account", user=session["user"]))
return redirect(url_for("account", username=session["user"]))
def test_deleting_adn_updating_projects(self):
with test_database(db, (Users, UserProjects)):
user1 = Users.create_user("konrad", "test")
UserProjects.create_project(title = "test project", body = "test", author = user1)
project = UserProjects.get_project(1)
UserProjects.update_project(project, "changed", "changed")
project = UserProjects.get_project(1)
self.assertEquals("changed", project.title)
self.assertEquals("changed", project.body)
UserProjects.delete_project(project)
self.assertEquals(0, UserProjects.select().count())
def account(username):
""" Main account view """
if username is None:
return redirect("/admin")
user = Users.get_user_by_username(username)
if not user:
abort(404)
articles = Articles.get_user_articles(user.username)
projects = UserProjects.get_all_projects()
return render_template("dashboard.html",
user=user,
articles=articles,
projects=projects)
def account(username):
""" Main account view """
if username is None:
return redirect("/admin")
user = Users.get_user_by_username(username)
if not user:
abort(404)
articles = Articles.get_user_articles(user.username)
projects = UserProjects.get_all_projects()
return render_template("dashboard.html",
user=user,
articles=articles,
projects=projects
)
def test_creating_project(self):
with test_database(db, (Users, UserProjects)):
user1 = Users.create_user("konrad", "test")
UserProjects.create_project(title = "test project", body = "test", author = user1)
project = UserProjects.select().get()
self.assertTrue(project)
self.assertEquals("test project", project.title)
self.assertEquals("konrad", project.author.username)
UserProjects.create_project(title = "test project 2", body = "test", author = user1)
self.assertTrue(UserProjects.check_exists("test project 2"))
q = UserProjects.select()
self.assertEquals(2, q.count())
self.assertRaises(IntegrityError, lambda: UserProjects.create_project(title = "test project", body = "test", author = user1))
def projects():
projects = UserProjects.get_all_projects()
return render_template("user_projects.html", projects=projects)
def get_object(self, id):
return UserProjects.get_project(id)
def projects():
projects = UserProjects.get_all_projects()
return render_template("user_projects.html", projects=projects)
