def root():
top_ten = database.get_ten()
top_recent = database.get_recent()
# trying to separate into lists for columns
top_five = list()
bottom_five = list()
recent_thalf = list()
recent_bhalf = list()
recent_another = list()
for team in range(0, len(top_ten)):
if team < len(top_ten)/2:
top_five.append(top_ten[team])
else:
bottom_five.append(top_ten[team])
for team in range(0, len(top_recent)):
if team < (len(top_recent)+1)/3:
recent_thalf.append(top_recent[team])
elif team < 2 * ((len(top_recent)+1)/3):
recent_bhalf.append(top_recent[team])
else:
recent_another.append(top_recent[team])
return render_template("index.html",
loggedin = auth.is_logged_in(),
top_five = top_five,
bottom_five = bottom_five,
recent_thalf = recent_thalf,
recent_bhalf = recent_bhalf,
recent_another = recent_another)
def search():
search_term = request.args['search'].replace("\'", "\'\'")
print search_term
results = database.search_name(search_term)
return render_template("search.html",
results = results,
loggedin = auth.is_logged_in())
def view_team():
if request.method == 'POST':
# edit team; go to edit_team
if 'edit' in request.form:
return redirect(url_for("edit_team", id = request.args['id']))
# delete team; database call; return to profile
elif 'delete' in request.form:
database.delete_team(int(request.args['id']))
return redirect(url_for("profile"))
# favorite team; dataase call; stay on team page
elif 'favorite' in request.form:
if 'user' in session:
id = request.args["id"]
database.add_favorite(session["user"], int(id))
return redirect(url_for("view_team", id = id))
else:
return redirect(url_for('login'))
# favorited; unfavorite button; database call; stay on page
else:
id = request.args["id"]
database.remove_favorite(session["user"], int(id))
return redirect(url_for("view_team", id = id))
else:
id = int(request.args["id"])
team = database.find_team(id)
mine = 'user' in session and session["user"] == team[1]
# get poke list from team; find pokemon info
pokedict2 = {}
alltypes = list()
for poke in team[8].split(","):
print poke
if poke != '':
pokeinfo = database.return_pkmn(int(poke))[0]
pokedict2[str(poke)] = pokeinfo
#create comprehensive list of types
for ptype in pokeinfo[7].split(","):
if ptype not in alltypes:
alltypes.append(ptype)
sandw = ["", ""]
if len(alltypes) != 0:
sandw = api.get_sandw(alltypes)
#check if it's favorited already
faves = list()
if 'user' in session:
faves = database.return_favorites(session["user"])[0][0].split(",")
return render_template("view_team.html",
loggedin = auth.is_logged_in(),
team = team,
favorited = str(id) in faves,
strengths = sandw[0],
weaknesses = sandw[1],
mine = mine,
poke_teams = pokedict2)
def edit_pokemon():
if request.method == 'POST':
#team id and pokemon id separated by comma
print request.args['id']
teampkmn = request.args['id'].split(",")
#list of types
typelist = request.form['type'][:-1]
#get selected moves and put in list form
moves = ""
movelist = []
for x in range(0, 3):
moves += (request.form['move' + str(x)]) + ", "
movelist.append(request.form['move' + str(x)])
moves += request.form['move3']
movelist.append(request.form['move3'])
# if one move was selected multiple times
print "MOVELIST"
print movelist
try:
if movelist.index("") != -1:
flash("Please select all 4 moves")
return render_template("edit_pokemon.html", loggedin = auth.is_logged_in(), action = "editpokemon?id=" + request.args['id'], which = "edit")
except ValueError:
print
if len(movelist) == 4 and len(movelist) != len(set(movelist)):
flash("Do not select a move more than once")
return render_template("edit_pokemon.html", loggedin = auth.is_logged_in(), action = "editpokemon?id=" + request.args['id'], which = "edit")
else:
#update database with new pokemon traits
database.update_poke(int(teampkmn[0]), request.form['pokemon'], "N/A", 0, request.form['ability'], moves, "N/A", typelist, request.form['img'])
#return to edit_team
return redirect(url_for("edit_team", id=int(teampkmn[1])))
else:
#you're going to need the id of the pokemon and the team
return render_template("edit_pokemon.html",
loggedin = auth.is_logged_in(),
which = "edit",
action = "editpokemon?id=" + request.args['pkmnid'])
def profile():
fav_list = database.return_favorites(session["user"])[0][0].split(",")
fav_teams = list()
for team in fav_list:
if team != '':
fav_teams.append(database.find_team(int(team)))
return render_template("profile.html",
user = session['user'],
loggedin = auth.is_logged_in(),
fav_teams = fav_teams,
my_teams = database.get_teams(session['user']))
def signup():
# CREATE ACCOUNT
if request.method == 'POST':
cr_acc_res = auth.sign_up( request.form['usr'], request.form['pwd'] )
# if successful
if cr_acc_res == 0:
flash("Account created!")
return redirect( url_for('login') )
# if username already exists
if cr_acc_res == 1:
return redirect( url_for('signup') )
return render_template("signup.html", loggedin=auth.is_logged_in())
def create():
if request.method == 'POST':
database.update_team(int(request.args['id']), request.form['teamname'], request.form['teamdesc'], request.form['teamvers'], "NONE", "NONE")
return redirect("/editteam?id=" + request.args['id'])
else:
new_teamid = database.next_teamid(session['user'])
return render_template("edit_team.html",
showPokemon = False,
action = "createteam?id=" + str(new_teamid),
created = "False",
new_teamid = new_teamid,
loggedin = auth.is_logged_in())
def create_pokemon():
if request.method == 'POST':
print "IMAGE: " + request.form['img']
teamid = request.args['id']
#list of types
typelist = request.form['type'][:-1]
#get selected moves and put in list form
moves = ""
movelist = []
for x in range(0, 3):
moves += (request.form['move' + str(x)]) + ", "
movelist.append(request.form['move' + str(x)])
moves += request.form['move3']
movelist.append(request.form['move3'])
# if one move was selected multiple times
print "MOVELIST"
print movelist
try:
if movelist.index("") != -1:
flash("Please select all 4 moves")
return render_template("edit_pokemon.html", loggedin = auth.is_logged_in(), action = "editpokemon?id=" + request.args['id'], which = "create")
except ValueError:
print
if len(movelist) == 4 and len(movelist) != len(set(movelist)):
flash("Do not select a move more than once")
return render_template("edit_pokemon.html", loggedin = auth.is_logged_in(), action = "createpokemon?id=" + request.args['id'], which = "create")
else:
#add to database
database.create_poke(request.form['pokemon'], "N/A", 0, request.form['ability'], moves, "N/A", typelist, int(teamid), request.form['img'])
return redirect(url_for("edit_team", id = request.args['id']))
else:
teamid = request.args['teamid']
return render_template("edit_pokemon.html",
loggedin = auth.is_logged_in(),
which = "create",
action = "createpokemon?id=" + str(teamid))
def login():
# checks for post method to respond to submit button
if request.method == 'POST':
# uses the database method to check the login
# print "username: "******"\npassword: "******"login.html", loggedin=auth.is_logged_in())
def edit_team():
if request.method == 'POST':
print request.args
if 'deletepkmn' in request.args:
teamid = request.args['deletepkmn'].split(",")[1]
pkmnid = request.args['deletepkmn'].split(",")[0]
database.delete_poke(int(teamid), int(pkmnid))
#go back to edit page
return redirect(url_for('edit_team', id=teamid))
else :
#save the team again
database.update_team(request.args['id'], request.form['teamname'], request.form['teamdesc'], request.form['teamvers'], "NONE", "NONE")
return redirect(url_for("view_team", id = request.args['id']))
else:
id = int(request.args["id"])
team = database.find_team(id)
pokemon = team[8].split(",")
# list of tuples- id, name&moves & list of types
pokedict2 = {}
alltypes = list()
for poke in pokemon:
print poke
pokelist = []
if poke != '':
# append name and moves to tuple
pokeinfo = database.return_pkmn(int(poke))[0]
pokelist.append(pokeinfo[1])
pokelist.append(pokeinfo[8])
pokedict2[str(poke)] = pokelist
#create comprehensive list of types
for ptype in pokeinfo[7].split(","):
if ptype not in alltypes:
alltypes.append(ptype)
sandw = ["", ""]
if len(alltypes) != 0:
sandw = api.get_sandw(alltypes)
print pokedict2
return render_template("edit_team.html",
showPokemon = True,
loggedin = auth.is_logged_in(),
action = "editteam?id=" + str(team[0]),
created = True,
pokemon = pokedict2,
strengths = sandw[0],
weaknesses = sandw[1],
more = len(pokedict2) < 6,
team = team)