def open_search_menu(window_x, window_y, db_dict, attr_dict=None, attr_list=None, settings=None):
num_results = 20
general_display = []
if attr_dict is None:
attr_dict = {}
if attr_list is None:
attr_list = []
if settings is None:
settings = {
"window": "search",
"mode": "players",
"p_db_rg": "player_db",
"f_db_rg": "formation_db",
"order_rg": True,
"messages": {"search": [], "sort": [], "results": []},
}
# ========== Window ==========
win_search = Window()
win_search.title = search_win_title
win_search.auto_position = False
win_search.position = (window_x, window_y)
win_search.size = (win_width, win_height)
win_search.resizable = 0
win_search.name = search_title + " Window"
# ========== Window Image View ==========
class FormationsWindowImageView(View):
def draw(self, c, r):
c.backcolor = view_backcolor
c.erase_rect(r)
view = FormationsWindowImageView(size=win_search.size)
# ========== Title ==========
title = Label(text=search_title)
title.font = title_font
title.width = title_width
title.height = title_height
title.x = (win_width - title_width) / 2
title.y = top_border
title.color = title_color
title.just = "center"
general_display.append(title)
# ========== Action Button Declarations ==========
start_btn = Button("Start")
back_btn = Button("Back")
# ========== Search Type Button Declarations ==========
players_btn = Button("Players")
formations_btn = Button("Formations")
teams_btn = Button("Teams")
search_type_button_list = [players_btn, formations_btn, teams_btn]
# ========== Tool Button Declarations ==========
attribute_btn = Button("Add Search Attribute")
sort_btn = Button("Add Sort Attribute")
reset_btn = Button("Reset Results")
# ========== Action Button Functions ==========
def start_btn_func():
# Start button corresponds to players
if settings["mode"] == "players":
# Search - if no attribute selected, return all
if len(attr_dict) == 0:
search_results = db_dict[p_db_radio_group.value][1]
else:
search_results = db_dict[p_db_radio_group.value][1].search(attr_dict)
search_results = PlayerDB.PlayerDB(search_results)
# Sort - if no attribute selected, use rating
if len(attr_list) == 0:
search_results.sort(["rating"], sort_order_radio_group.value)
else:
search_results.sort(attr_list, sort_order_radio_group.value)
# Get attributes list and avoid duplicates
attributes_list = []
for attr in attr_list:
if attributes_list.count(attr) == 0:
attributes_list.append(attr)
for attr_key in attr_dict.iterkeys():
if attributes_list.count(attr_key) == 0:
attributes_list.append(attr_key)
display_players(search_results, attributes_list, (0, num_results))
# Start button corresponds to formations
elif settings["mode"] == "formations":
# Search - if no attribute selected, return all
if len(attr_dict) == 0:
search_results = db_dict[f_db_radio_group.value][1]
else:
search_results = db_dict[f_db_radio_group.value][1].search(attr_dict)
search_results = FormationDB.FormationDB(search_results)
# Sort - if no attribute selected, use name
if len(attr_list) == 0:
search_results.sort(["name"], sort_order_radio_group.value)
else:
search_results.sort(attr_list, sort_order_radio_group.value)
# Get attributes list and avoid duplicates
attributes_list = []
for attr in attr_list:
if attributes_list.count(attr) == 0:
attributes_list.append(attr)
for attr_key in attr_dict.iterkeys():
if attributes_list.count(attr_key) == 0:
attributes_list.append(attr_key)
display_formations(search_results, attributes_list, (0, num_results))
# Start button corresponds to teams
elif settings["mode"] == "teams":
# Search - if no attribute selected, return all
if len(attr_dict) == 0:
search_results = db_dict["team_list"][1]
else:
search_results = db_dict["team_list"][1].search(attr_dict)
search_results = TeamDB.TeamDB(search_results)
# Sort - if no attribute selected, use rating
if len(attr_list) == 0:
search_results.sort(["rating"], sort_order_radio_group.value)
else:
search_results.sort(attr_list, sort_order_radio_group.value)
# Get attributes list and avoid duplicates
attributes_list = []
for attr in attr_list:
if attributes_list.count(attr) == 0:
attributes_list.append(attr)
for attr_key in attr_dict.iterkeys():
if attributes_list.count(attr_key) == 0:
attributes_list.append(attr_key)
display_teams(search_results, attributes_list, (0, num_results))
win_search.become_target()
def back_btn_func():
win_search.hide()
StartMenu.open_start_menu(win_search.x, win_search.y, db_dict)
# ========== Search Type Button Functions ==========
def players_btn_func():
for btn in search_type_button_list:
btn.enabled = 1
players_btn.enabled = 0
settings["mode"] = "players"
# Reset screen
reset_btn_func()
view.remove(formation_db_radio_btn)
view.remove(f_db_file_btn)
view.remove(formation_list_radio_btn)
view.remove(f_list_file_btn)
view.remove(team_list_button)
view.add(player_db_radio_btn)
view.add(p_db_file_btn)
view.add(player_list_radio_btn)
view.add(p_list_file_btn)
win_search.become_target()
def formations_btn_func():
for btn in search_type_button_list:
btn.enabled = 1
formations_btn.enabled = 0
settings["mode"] = "formations"
# Reset screen
reset_btn_func()
view.remove(player_db_radio_btn)
view.remove(p_db_file_btn)
view.remove(player_list_radio_btn)
view.remove(p_list_file_btn)
view.remove(team_list_button)
view.add(formation_db_radio_btn)
view.add(f_db_file_btn)
view.add(formation_list_radio_btn)
view.add(f_list_file_btn)
win_search.become_target()
def teams_btn_func():
for btn in search_type_button_list:
btn.enabled = 1
teams_btn.enabled = 0
settings["mode"] = "teams"
# Reset screen
reset_btn_func()
view.remove(player_db_radio_btn)
view.remove(p_db_file_btn)
view.remove(player_list_radio_btn)
view.remove(p_list_file_btn)
view.remove(formation_db_radio_btn)
view.remove(f_db_file_btn)
view.remove(formation_list_radio_btn)
view.remove(f_list_file_btn)
view.add(team_list_button)
win_search.become_target()
# ========== Tool Button Functions ==========
def attribute_btn_func():
# Delete results
del settings["messages"]["results"][:]
# Set type argument
attr_type = ""
if settings["mode"] == "players":
attr_type = "player_search"
elif settings["mode"] == "formations":
attr_type = "formation_search"
elif settings["mode"] == "teams":
attr_type = "team_search"
else:
print "Invalid mode."
# Open new window and close current window
win_search.hide()
AddAttribute.open_attribute_window(
win_search.x, win_search.y, db_dict, attr_dict, attr_list, attr_type, settings
)
def sort_btn_func():
# Delete results
del settings["messages"]["results"][:]
# Set type argument
attr_type = ""
if settings["mode"] == "players":
attr_type = "player_sort"
elif settings["mode"] == "formations":
attr_type = "formation_sort"
elif settings["mode"] == "teams":
attr_type = "team_sort"
else:
print "Invalid mode."
# Open new window and close current window
win_search.hide()
AddAttribute.open_attribute_window(
win_search.x, win_search.y, db_dict, attr_dict, attr_list, attr_type, settings
)
def reset_btn_func():
# Remove messages off page
for message in settings["messages"]["search"]:
view.remove(message)
for message in settings["messages"]["sort"]:
view.remove(message)
for message in settings["messages"]["results"]:
view.remove(message)
# Delete the attribute parameters for search and sort
attr_dict.clear()
del attr_list[:]
del settings["messages"]["results"][:]
win_search.become_target()
def player_bio_btn_func(player):
win_search.hide()
PlayerBio.open_player_bio_window(
win_search.x,
win_search.y,
player,
win_search,
db_dict,
db_dict["player_list"][0],
db_dict["player_list"][1],
)
win_search.become_target()
def formation_bio_btn_func(formation):
win_search.hide()
FormationBio.open_formation_bio_window(
win_search.x,
win_search.y,
formation,
win_search,
db_dict["formation_list"][0],
db_dict["formation_list"][1],
)
win_search.become_target()
def team_bio_btn_func(team):
win_search.hide()
TeamBio.open_team_bio_window(
win_search.x, win_search.y, team, win_search, db_dict["team_list"][0], db_dict["team_list"][1]
)
win_search.become_target()
# ========== Pick DB/List Button Functions ==========
def pick_file():
# Remove old messages off page
for message in settings["messages"]["results"]:
view.remove(message)
del settings["messages"]["results"][:]
settings["file_changes"] = False
settings["prev_window"] = "search"
settings["attr_dict"] = attr_dict
settings["attr_list"] = attr_list
PickFile.open_pick_file_window(win_search.x, win_search.y, db_dict, settings)
win_search.hide()
def pick_p_db_func():
settings["file_type"] = "current_player_db"
pick_file()
def pick_p_list_func():
settings["file_type"] = "current_player_list"
pick_file()
def pick_f_db_func():
settings["file_type"] = "current_formation_db"
pick_file()
def pick_f_list_func():
settings["file_type"] = "current_formation_list"
pick_file()
def pick_t_list_func():
settings["file_type"] = "current_team_list"
pick_file()
# ========== Action Buttons ==========
start_btn.x = (win_width - 2 * small_button_width - small_button_spacing) / 2
start_btn.y = title.bottom + small_button_top_spacing
start_btn.height = small_button_height
start_btn.width = small_button_width
start_btn.font = small_button_font
start_btn.action = start_btn_func
start_btn.style = "default"
start_btn.color = small_button_color
start_btn.just = "right"
general_display.append(start_btn)
back_btn.x = start_btn.right + small_button_spacing
back_btn.y = start_btn.top
back_btn.height = small_button_height
back_btn.width = small_button_width
back_btn.font = small_button_font
back_btn.action = back_btn_func
back_btn.style = "default"
back_btn.color = small_button_color
back_btn.just = "right"
general_display.append(back_btn)
# ========== Search Type Buttons ==========
players_btn.x = (win_width - 3 * small_button_width - 2 * small_button_spacing) / 2
players_btn.y = start_btn.bottom + small_button_top_spacing
players_btn.height = small_button_height
players_btn.width = small_button_width
players_btn.font = small_button_font
players_btn.action = players_btn_func
players_btn.style = "default"
players_btn.color = small_button_color
players_btn.just = "right"
if settings["mode"] == "players":
players_btn.enabled = 0
general_display.append(players_btn)
formations_btn.x = players_btn.right + small_button_spacing
formations_btn.y = players_btn.top
formations_btn.height = small_button_height
formations_btn.width = small_button_width
formations_btn.font = small_button_font
formations_btn.action = formations_btn_func
formations_btn.style = "default"
formations_btn.color = small_button_color
formations_btn.just = "right"
if settings["mode"] == "formations":
formations_btn.enabled = 0
general_display.append(formations_btn)
teams_btn.x = formations_btn.right + small_button_spacing
teams_btn.y = players_btn.top
teams_btn.height = small_button_height
teams_btn.width = small_button_width
teams_btn.font = small_button_font
teams_btn.action = teams_btn_func
teams_btn.style = "default"
teams_btn.color = small_button_color
teams_btn.just = "right"
if settings["mode"] == "teams":
teams_btn.enabled = 0
general_display.append(teams_btn)
# ========== Tool Buttons ==========
attribute_btn.x = (win_width - 3 * small_button_width - 2 * small_button_spacing) / 2
attribute_btn.y = players_btn.bottom + 5
attribute_btn.height = small_button_height
attribute_btn.width = small_button_width
attribute_btn.font = small_button_font
attribute_btn.action = attribute_btn_func
attribute_btn.style = "default"
attribute_btn.color = small_button_color
attribute_btn.just = "right"
general_display.append(attribute_btn)
sort_btn.x = attribute_btn.right + small_button_spacing
sort_btn.y = attribute_btn.top
sort_btn.height = small_button_height
sort_btn.width = small_button_width
sort_btn.font = small_button_font
sort_btn.action = sort_btn_func
sort_btn.style = "default"
sort_btn.color = small_button_color
sort_btn.just = "right"
general_display.append(sort_btn)
reset_btn.x = sort_btn.right + small_button_spacing
reset_btn.y = attribute_btn.top
reset_btn.height = small_button_height
reset_btn.width = small_button_width
reset_btn.font = small_button_font
reset_btn.action = reset_btn_func
reset_btn.style = "default"
reset_btn.color = small_button_color
reset_btn.just = "right"
general_display.append(reset_btn)
# ========== DB Radio Buttons ==========
def get_attribute_p_db_rg():
settings["p_db_rg"] = p_db_radio_group.value
win_search.become_target()
def get_attribute_f_db_rg():
settings["f_db_rg"] = f_db_radio_group.value
win_search.become_target()
p_db_radio_group = RadioGroup(action=get_attribute_p_db_rg)
f_db_radio_group = RadioGroup(action=get_attribute_f_db_rg)
db_msg_width = 35
db_radio_btn_width = 150
db_file_btn_width = 40
db_radio_btn_space = 5
db_file_btn_space = 2
file_label = Label(text="File:", font=std_tf_font, width=db_msg_width, height=std_tf_height, color=title_color)
file_label.x = (
win_search.width
- 2 * db_radio_btn_width
- 2 * db_file_btn_width
- db_radio_btn_space
- 2 * db_file_btn_space
- db_msg_width
) / 2
file_label.y = reset_btn.bottom + db_radio_btn_space
view.add(file_label)
# Players DB RG
player_db_radio_btn = RadioButton(db_dict["player_db"][0])
player_db_radio_btn.width = db_radio_btn_width
player_db_radio_btn.x = file_label.right
player_db_radio_btn.y = file_label.top
player_db_radio_btn.group = p_db_radio_group
player_db_radio_btn.value = "player_db"
p_db_file_btn = Button("Pick")
p_db_file_btn.x = player_db_radio_btn.right + db_file_btn_space
p_db_file_btn.y = file_label.top
p_db_file_btn.height = std_tf_height
p_db_file_btn.width = db_file_btn_width
p_db_file_btn.font = small_button_font
p_db_file_btn.action = pick_p_db_func
p_db_file_btn.style = "default"
p_db_file_btn.color = small_button_color
p_db_file_btn.just = "right"
player_list_radio_btn = RadioButton(db_dict["player_list"][0])
player_list_radio_btn.width = db_radio_btn_width
player_list_radio_btn.x = p_db_file_btn.right + db_radio_btn_space
player_list_radio_btn.y = file_label.top
player_list_radio_btn.group = p_db_radio_group
player_list_radio_btn.value = "player_list"
p_list_file_btn = Button("Pick")
p_list_file_btn.x = player_list_radio_btn.right + db_file_btn_space
p_list_file_btn.y = file_label.top
p_list_file_btn.height = std_tf_height
p_list_file_btn.width = db_file_btn_width
p_list_file_btn.font = small_button_font
p_list_file_btn.action = pick_p_list_func
p_list_file_btn.style = "default"
p_list_file_btn.color = small_button_color
p_list_file_btn.just = "right"
p_db_radio_group.value = settings["p_db_rg"]
# Formations DB RG
formation_db_radio_btn = RadioButton(db_dict["formation_db"][0])
formation_db_radio_btn.width = db_radio_btn_width
formation_db_radio_btn.x = file_label.right
formation_db_radio_btn.y = file_label.top
formation_db_radio_btn.group = f_db_radio_group
formation_db_radio_btn.value = "formation_db"
f_db_file_btn = Button("Pick")
f_db_file_btn.x = formation_db_radio_btn.right + db_file_btn_space
f_db_file_btn.y = file_label.top
f_db_file_btn.height = std_tf_height
f_db_file_btn.width = db_file_btn_width
f_db_file_btn.font = small_button_font
f_db_file_btn.action = pick_f_db_func
f_db_file_btn.style = "default"
f_db_file_btn.color = small_button_color
f_db_file_btn.just = "right"
formation_list_radio_btn = RadioButton(db_dict["formation_list"][0])
formation_list_radio_btn.width = db_radio_btn_width
formation_list_radio_btn.x = f_db_file_btn.right + db_radio_btn_space
formation_list_radio_btn.y = file_label.top
formation_list_radio_btn.group = f_db_radio_group
formation_list_radio_btn.value = "formation_list"
f_list_file_btn = Button("Pick")
f_list_file_btn.x = formation_list_radio_btn.right + db_file_btn_space
f_list_file_btn.y = file_label.top
f_list_file_btn.height = std_tf_height
f_list_file_btn.width = db_file_btn_width
f_list_file_btn.font = small_button_font
f_list_file_btn.action = pick_f_list_func
f_list_file_btn.style = "default"
f_list_file_btn.color = small_button_color
f_list_file_btn.just = "right"
f_db_radio_group.value = settings["f_db_rg"]
# Teams DB Button
teams_db_msg_width = 2 * db_radio_btn_width + 2 * db_file_btn_width + db_radio_btn_space + 2 * db_file_btn_space
team_list_button = Button(
title=db_dict["team_list"][0],
font=std_tf_font,
just="left",
width=teams_db_msg_width,
height=std_tf_height,
color=title_color,
)
team_list_button.x = file_label.right
team_list_button.y = reset_btn.bottom + db_radio_btn_space
team_list_button.action = pick_t_list_func
if settings["mode"] == "players":
view.add(player_db_radio_btn)
view.add(p_db_file_btn)
view.add(player_list_radio_btn)
view.add(p_list_file_btn)
elif settings["mode"] == "formations":
view.add(formation_db_radio_btn)
view.add(f_db_file_btn)
view.add(formation_list_radio_btn)
view.add(f_list_file_btn)
elif settings["mode"] == "teams":
view.add(team_list_button)
else:
print "Error: Invalid mode."
# ========== Sort Order Radio Buttons ==========
def get_attribute_sort_order_rg():
settings["order_rg"] = sort_order_radio_group.value
win_search.become_target()
sort_order_radio_group = RadioGroup(action=get_attribute_sort_order_rg)
asc_desc_radio_btn_width = 75
asc_msg_width = 80
radio_btn_space = 5
asc_desc_rg_msg = Label(
text="Sort Order:", font=std_tf_font, width=asc_msg_width, height=std_tf_height, color=title_color
)
asc_desc_rg_msg.x = (win_search.width - 2 * asc_desc_radio_btn_width - radio_btn_space - asc_msg_width) / 2
asc_desc_rg_msg.y = formation_db_radio_btn.bottom + radio_btn_space
general_display.append(asc_desc_rg_msg)
descend_radio_btn = RadioButton("Descending")
descend_radio_btn.width = asc_desc_radio_btn_width
descend_radio_btn.x = asc_desc_rg_msg.right
descend_radio_btn.y = asc_desc_rg_msg.top
descend_radio_btn.group = sort_order_radio_group
descend_radio_btn.value = True
general_display.append(descend_radio_btn)
ascend_radio_btn = RadioButton("Ascending")
ascend_radio_btn.width = asc_desc_radio_btn_width
ascend_radio_btn.x = descend_radio_btn.right + radio_btn_space
ascend_radio_btn.y = descend_radio_btn.top
ascend_radio_btn.group = sort_order_radio_group
ascend_radio_btn.value = False
general_display.append(ascend_radio_btn)
sort_order_radio_group.value = settings["order_rg"]
# ========== Messages ==========
lowest_msg_l = start_btn.top
lowest_msg_r = start_btn.top
attr_msg_offset = 25
# Attribute Messages
del settings["messages"]["search"][:]
del settings["messages"]["sort"][:]
if len(attr_dict) > 0:
settings["messages"]["search"].append(
Label(
text="Search Attributes:",
font=title_tf_font,
width=std_tf_width,
height=std_tf_height,
x=attr_msg_offset,
y=lowest_msg_l,
color=title_color,
)
)
lowest_msg_l += std_tf_height
for key, value in attr_dict.iteritems():
msg_text = format_attr_name(key) + ": "
if key in ["id", "baseId", "nationId", "leagueId", "clubId"]:
msg_text += str(value[0])
elif type(value[0]) is int:
msg_text += value[1].capitalize() + " " + str(value[0])
elif value[1] == "not":
msg_text += value[1].capitalize() + ' "' + str(value[0]) + '"'
else:
msg_text += '"' + str(value[0]) + '"'
attr_label = Label(
text=msg_text,
font=std_tf_font,
width=std_tf_width,
height=std_tf_height,
x=attr_msg_offset,
y=lowest_msg_l,
color=title_color,
)
lowest_msg_l += std_tf_height
settings["messages"]["search"].append(attr_label)
if len(attr_list) > 0:
settings["messages"]["sort"].append(
Label(
text="Sort Attributes:",
font=title_tf_font,
width=std_tf_width,
height=std_tf_height,
x=teams_btn.right + 3 * attr_msg_offset,
y=lowest_msg_r,
color=title_color,
)
)
lowest_msg_r += std_tf_height
for value in attr_list:
attr_label = Label(
text=(format_attr_name(value)),
font=std_tf_font,
width=std_tf_width,
height=std_tf_height,
x=teams_btn.right + 3 * attr_msg_offset,
y=lowest_msg_r,
color=title_color,
)
lowest_msg_r += std_tf_height
settings["messages"]["sort"].append(attr_label)
# ========== Previous, Add to List, Next Buttons ==========
previous_btn = Button("<<< Previous %d" % num_results)
add_to_list_btn = Button()
next_btn = Button("Next %d >>>" % num_results)
total_num_results_label = Label()
pages_label = Label()
def add_to_list_btn_func(input_list, func_type):
results_list = copy.deepcopy(input_list)
if func_type == "add all":
if settings["mode"] == "players":
added_players = []
# Add current results to player list
for player in results_list:
if db_dict["player_list"][1].db.count(player) == 0:
db_dict["player_list"][1].db.append(player)
added_players.append(player)
# Sort
db_dict["player_list"][1].sort(["rating"])
# Save
db_dict["player_list"][1].save(db_dict["player_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Remove Added Players"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "remove select")
# Keep track of just added players
settings["messages"]["players_changed"] = added_players
elif settings["mode"] == "formations":
added_formations = []
# Add current results to formation list
for formation in results_list:
if db_dict["formation_list"][1].db.count(formation) == 0:
db_dict["formation_list"][1].db.append(formation)
added_formations.append(formation)
# Sort
db_dict["formation_list"][1].sort(["name"])
# Save
db_dict["formation_list"][1].save(db_dict["formation_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Remove Added Forms"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "remove select")
# Keep track of just added formations
settings["messages"]["formations_changed"] = added_formations
elif func_type == "remove all":
if settings["mode"] == "players":
removed_players = []
# Remove current results from player list
for player in results_list:
if db_dict["player_list"][1].db.count(player) > 0:
db_dict["player_list"][1].db.remove(player)
removed_players.append(player)
# Sort
db_dict["player_list"][1].sort(["rating"])
# Save
db_dict["player_list"][1].save(db_dict["player_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Add Removed Players"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "add select")
# Keep track of just removed players
settings["messages"]["players_changed"] = removed_players
elif settings["mode"] == "formations":
removed_formations = []
# Remove current results from formation list
for formation in results_list:
if db_dict["formation_list"][1].db.count(formation) > 0:
db_dict["formation_list"][1].db.remove(formation)
removed_formations.append(formation)
# Sort
db_dict["formation_list"][1].sort(["name"])
# Save
db_dict["formation_list"][1].save(db_dict["formation_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Add Removed Forms"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "add select")
# Keep track of just removed players
settings["messages"]["formations_changed"] = removed_formations
elif func_type == "add select":
if settings["mode"] == "players":
# Add select players back to player list
for player in settings["messages"]["players_changed"]:
if db_dict["player_list"][1].db.count(player) == 0:
db_dict["player_list"][1].db.append(player)
# Sort
db_dict["player_list"][1].sort(["rating"])
# Save
db_dict["player_list"][1].save(db_dict["player_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Remove Added Players"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "remove select")
elif settings["mode"] == "formations":
# Add select formations back to formation list
for formation in settings["messages"]["formations_changed"]:
if db_dict["formation_list"][1].db.count(formation) == 0:
db_dict["formation_list"][1].db.append(formation)
# Sort
db_dict["formation_list"][1].sort(["name"])
# Save
db_dict["formation_list"][1].save(db_dict["formation_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Remove Added Forms"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "remove select")
elif func_type == "remove select":
if settings["mode"] == "players":
# Remove select players from player list
for player in settings["messages"]["players_changed"]:
if db_dict["player_list"][1].db.count(player) > 0:
db_dict["player_list"][1].db.remove(player)
# Sort
db_dict["player_list"][1].sort(["rating"])
# Save
db_dict["player_list"][1].save(db_dict["player_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Add Removed Players"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "add select")
elif settings["mode"] == "formations":
# Remove select formations from formation list
for formation in settings["messages"]["formations_changed"]:
if db_dict["formation_list"][1].db.count(formation) > 0:
db_dict["formation_list"][1].db.remove(formation)
# Sort
db_dict["formation_list"][1].sort(["name"])
# Save
db_dict["formation_list"][1].save(db_dict["formation_list"][0], "list", True)
# Change button title and action
add_to_list_btn.title = "Add Removed Forms"
add_to_list_btn.action = (add_to_list_btn_func, results_list, "add select")
win_search.become_target()
def previous_btn_func(results_list, attributes, index_range):
# display previous results
if settings["mode"] == "players":
display_players(results_list, attributes, index_range)
elif settings["mode"] == "formations":
display_formations(results_list, attributes, index_range)
elif settings["mode"] == "teams":
display_teams(results_list, attributes, index_range)
win_search.become_target()
def next_btn_func(results_list, attributes, index_range):
# display next results
if settings["mode"] == "players":
display_players(results_list, attributes, index_range)
elif settings["mode"] == "formations":
display_formations(results_list, attributes, index_range)
elif settings["mode"] == "teams":
display_teams(results_list, attributes, index_range)
win_search.become_target()
add_to_list_btn.x = attribute_btn.right + small_button_spacing
add_to_list_btn.y = descend_radio_btn.bottom + 5
add_to_list_btn.height = tiny_button_height
add_to_list_btn.width = small_button_width
add_to_list_btn.font = small_button_font
add_to_list_btn.style = "default"
add_to_list_btn.color = small_button_color
add_to_list_btn.just = "right"
previous_btn.height = tiny_button_height
previous_btn.width = small_button_width
previous_btn.x = add_to_list_btn.left - previous_btn.width - small_button_spacing
previous_btn.y = add_to_list_btn.top
previous_btn.font = small_button_font
previous_btn.style = "default"
previous_btn.color = small_button_color
previous_btn.just = "right"
next_btn.height = tiny_button_height
next_btn.width = small_button_width
next_btn.x = add_to_list_btn.right + small_button_spacing
next_btn.y = add_to_list_btn.top
next_btn.font = small_button_font
next_btn.style = "default"
next_btn.color = small_button_color
next_btn.just = "right"
total_num_results_label.font = std_tf_font
total_num_results_label.width = 100
total_num_results_label.height = std_tf_height
total_num_results_label.x = previous_btn.left + -100 - 10
total_num_results_label.y = add_to_list_btn.top
total_num_results_label.color = title_color
total_num_results_label.just = "right"
pages_label.font = std_tf_font
pages_label.width = 125
pages_label.height = std_tf_height
pages_label.x = next_btn.right + 10
pages_label.y = add_to_list_btn.top
pages_label.color = title_color
pages_label.just = "left"
# ========== Display players from search ==========
def display_players(results_list, attributes, index_range):
# Remove old messages off page
for message in settings["messages"]["results"]:
view.remove(message)
del settings["messages"]["results"][:]
# Add navigation buttons to page
if settings["p_db_rg"] == "player_db":
add_to_list_btn.title = "Add All Players"
add_to_list_btn.action = (add_to_list_btn_func, results_list.db, "add all")
elif settings["p_db_rg"] == "player_list":
add_to_list_btn.title = "Remove All Players"
add_to_list_btn.action = (add_to_list_btn_func, results_list.db, "remove all")
else:
print "Invalid edit type."
previous_range = (index_range[0] - num_results, index_range[0])
previous_btn.action = (previous_btn_func, results_list, attributes, previous_range)
next_range = (index_range[1], index_range[1] + num_results)
next_btn.action = (next_btn_func, results_list, attributes, next_range)
total_num_results_label.text = str(len(results_list.db)) + " Players"
pages_label.text = "Page %d of %d" % (
int(index_range[1] / num_results),
math.ceil(len(results_list.db) / float(num_results)),
)
if index_range[0] > 0:
previous_btn.enabled = 1
else:
previous_btn.enabled = 0
if index_range[1] <= len(results_list.db) - 1:
next_btn.enabled = 1
else:
next_btn.enabled = 0
settings["messages"]["results"].append(add_to_list_btn)
settings["messages"]["results"].append(previous_btn)
settings["messages"]["results"].append(next_btn)
settings["messages"]["results"].append(total_num_results_label)
settings["messages"]["results"].append(pages_label)
# Print out labels
labels = player_info_labels(attributes)
stat_index = 0
# Spacing values for each of the stats
spacing_list = [125, 40, 40, 65, 115, 115, 115, 40]
# Calculate the maximum number of stat fields that will fit on screen
max_player_fields = len(spacing_list[:-1]) + (win_search.width - sum(spacing_list[:-1])) / spacing_list[-1]
# Calculate the left border of the stats based on the number and width of the stats
left_border = (
win_width
- sum(spacing_list[:-1])
- (len(labels[:max_player_fields]) - len(spacing_list) + 1) * spacing_list[-1]
) / 2
msg_x = left_border
msg_y = add_to_list_btn.bottom + 5
for info_label in labels[:max_player_fields]:
player_label = Label(
text=info_label,
font=std_tf_font_bold,
width=spacing_list[stat_index] - 5,
height=std_tf_height,
x=msg_x,
y=msg_y,
color=title_color,
)
settings["messages"]["results"].append(player_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height + 5
# Print out players
for idx, player in enumerate(results_list.db[index_range[0] : index_range[1]]):
msg_x = left_border
player_stats = player_info(player, attributes)
stat_index = 0
# Check for names that are too long
name = player_stats[0]
if len(name) > 20:
name = player["lastName"]
bio_btn = Button(
title=name,
width=spacing_list[stat_index] - 5,
height=15,
x=msg_x,
y=msg_y,
action=(player_bio_btn_func, player),
)
settings["messages"]["results"].append(bio_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
for player_stat in player_stats[1:max_player_fields]:
player_label = Label(
text=player_stat,
font=small_button_font,
width=spacing_list[stat_index] - 5,
height=std_tf_height,
x=msg_x,
y=msg_y,
color=title_color,
)
settings["messages"]["results"].append(player_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height
for results_msg in settings["messages"]["results"]:
view.add(results_msg)
# ========== Display formations from search ==========
def display_formations(results_list, attributes, index_range):
# Remove old messages off page
for message in settings["messages"]["results"]:
view.remove(message)
del settings["messages"]["results"][:]
# Add navigation buttons to page
if settings["f_db_rg"] == "formation_db":
add_to_list_btn.title = "Add All Formations"
add_to_list_btn.action = (add_to_list_btn_func, results_list.db, "add all")
elif settings["f_db_rg"] == "formation_list":
add_to_list_btn.title = "Remove All Formations"
add_to_list_btn.action = (add_to_list_btn_func, results_list.db, "remove all")
else:
print "Invalid edit type."
previous_range = (index_range[0] - num_results, index_range[0])
previous_btn.action = (previous_btn_func, results_list, attributes, previous_range)
next_range = (index_range[1], index_range[1] + num_results)
next_btn.action = (next_btn_func, results_list, attributes, next_range)
total_num_results_label.text = str(len(results_list.db)) + " Formations"
pages_label.text = "Page %d of %d" % (
int(index_range[1] / num_results),
math.ceil(len(results_list.db) / float(num_results)),
)
if index_range[0] > 0:
previous_btn.enabled = 1
else:
previous_btn.enabled = 0
if index_range[1] <= len(results_list.db) - 1:
next_btn.enabled = 1
else:
next_btn.enabled = 0
settings["messages"]["results"].append(add_to_list_btn)
settings["messages"]["results"].append(previous_btn)
settings["messages"]["results"].append(next_btn)
settings["messages"]["results"].append(total_num_results_label)
settings["messages"]["results"].append(pages_label)
# Print out labels
labels = formation_info_labels()
stat_index = 0
# Spacing values for each of the stats
spacing_list = [100, 100, 55, 55, 55, 55, 140, 160]
# Calculate the left border of the stats based on the number and width of the stats
left_border = (win_search.width - sum(spacing_list)) / 2
msg_x = left_border
msg_y = add_to_list_btn.bottom + 5
for info_label in labels:
formation_label = Label(
text=info_label,
font=std_tf_font_bold,
width=spacing_list[stat_index] - 5,
height=std_tf_height,
x=msg_x,
y=msg_y,
color=title_color,
)
settings["messages"]["results"].append(formation_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height + 5
# Print out formations
for formation in results_list.db[index_range[0] : index_range[1]]:
msg_x = left_border
formation_stats = formation_info(formation)
stat_index = 0
bio_btn = Button(
title=formation_stats[0],
width=spacing_list[stat_index] - 5,
height=15,
x=msg_x,
y=msg_y,
action=(formation_bio_btn_func, formation),
)
settings["messages"]["results"].append(bio_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
for formation_stat in formation_stats[1:]:
formation_label = Label(
text=formation_stat,
font=small_button_font,
width=spacing_list[stat_index] - 5,
height=std_tf_height,
x=msg_x,
y=msg_y,
color=title_color,
)
settings["messages"]["results"].append(formation_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height
for results_msg in settings["messages"]["results"]:
view.add(results_msg)
# ========== Display teams from search ==========
def display_teams(results_list, attributes, index_range):
# Remove old messages off page
for message in settings["messages"]["results"]:
view.remove(message)
del settings["messages"]["results"][:]
# Add navigation buttons to page
'''if settings['f_db_rg'] == 'formation_db':
add_to_list_btn.title = 'Add All Formations'
add_to_list_btn.action = (add_to_list_btn_func, results_list.db, 'add all')
elif settings['f_db_rg'] == 'formation_list':
add_to_list_btn.title = 'Remove All Formations'
add_to_list_btn.action = (add_to_list_btn_func, results_list.db, 'remove all')
else:
print "Invalid edit type."'''
add_to_list_btn.title = ""
add_to_list_btn.action = ()
previous_range = (index_range[0] - num_results, index_range[0])
previous_btn.action = (previous_btn_func, results_list, attributes, previous_range)
next_range = (index_range[1], index_range[1] + num_results)
next_btn.action = (next_btn_func, results_list, attributes, next_range)
total_num_results_label.text = str(len(results_list.db)) + " Teams"
pages_label.text = "Page %d of %d" % (
int(index_range[1] / num_results),
math.ceil(len(results_list.db) / float(num_results)),
)
if index_range[0] > 0:
previous_btn.enabled = 1
else:
previous_btn.enabled = 0
if index_range[1] <= len(results_list.db) - 1:
next_btn.enabled = 1
else:
next_btn.enabled = 0
settings["messages"]["results"].append(add_to_list_btn)
settings["messages"]["results"].append(previous_btn)
settings["messages"]["results"].append(next_btn)
settings["messages"]["results"].append(total_num_results_label)
settings["messages"]["results"].append(pages_label)
# Print out labels
labels = team_info_labels(attributes)
stat_index = 0
# Spacing values for each of the stats
spacing_list = [60, 40, 55, 90, 100, 115, 115, 40]
# Calculate the maximum number of stat fields that will fit on screen
max_team_fields = len(spacing_list[:-1]) + (win_search.width - sum(spacing_list[:-1])) / spacing_list[-1]
# Calculate the left border of the stats based on the number and width of the stats
left_border = (
win_width
- sum(spacing_list[:-1])
- (len(labels[:max_team_fields]) - len(spacing_list) + 1) * spacing_list[-1]
) / 2
msg_x = left_border
msg_y = add_to_list_btn.bottom + 5
for info_label in labels[:max_team_fields]:
team_label = Label(
text=info_label,
font=std_tf_font_bold,
width=spacing_list[stat_index] - 5,
height=std_tf_height,
x=msg_x,
y=msg_y,
color=title_color,
)
settings["messages"]["results"].append(team_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height + 5
# Print out teams
for team in results_list.db[index_range[0] : index_range[1]]:
msg_x = left_border
team_stats = team_info(team, attributes)
stat_index = 0
bio_btn = Button(
title=team_stats[0],
width=spacing_list[stat_index] - 5,
height=15,
x=msg_x,
y=msg_y,
action=(team_bio_btn_func, team),
)
settings["messages"]["results"].append(bio_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
for team_stat in team_stats[1:max_team_fields]:
team_label = Label(
text=team_stat,
font=small_button_font,
width=spacing_list[stat_index] - 5,
height=std_tf_height,
x=msg_x,
y=msg_y,
color=title_color,
)
settings["messages"]["results"].append(team_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height
for results_msg in settings["messages"]["results"]:
view.add(results_msg)
# ========== Add components to view and add view to window ==========
for msg in general_display:
view.add(msg)
for msg in settings["messages"]["search"]:
view.add(msg)
for msg in settings["messages"]["sort"]:
view.add(msg)
for msg in settings["messages"]["results"]:
view.add(msg)
win_search.add(view)
view.become_target()
win_search.show()
def open_edit_menu(window_x, window_y, db_dict, attr_dict=None, attr_list=None, settings=None):
list_players = PlayerDB.PlayerDB()
list_formations = FormationDB.FormationDB()
list_teams = TeamDB.TeamDB()
if settings['edit_subject'] == 'players':
list_players.load(settings['file_name'], 'list')
elif settings['edit_subject'] == 'formations':
list_formations.load(settings['file_name'], 'list')
elif settings['edit_subject'] == 'teams':
list_teams.load(settings['file_name'])
if attr_dict is None:
attr_dict = {}
if attr_list is None:
attr_list = []
if settings is None:
settings = {
'window': 'edit',
'edit_type': 'add',
'edit_mode': 'simple',
'edit_subject': 'players',
'sort_order': True,
'messages': {
'search': [],
'sort': [],
'results': []
}
}
else:
if 'window' not in settings:
settings['window'] = 'edit'
if 'edit_type' not in settings:
settings['edit_type'] = 'add'
if 'edit_mode' not in settings:
settings['edit_mode'] = 'simple'
if 'edit_subject' not in settings:
settings['edit_subject'] = 'players'
if 'sort_order' not in settings:
settings['sort_order'] = True
if 'messages' not in settings:
settings['messages'] = {'search': [], 'sort': [], 'results': []}
num_results = 20
general_display = []
simple_display = []
advanced_display = []
# ========== Window ==========
win_edit = Window()
win_edit.title = edit_win_title
win_edit.auto_position = False
win_edit.position = (window_x, window_y)
win_edit.size = (win_width, win_height)
win_edit.resizable = 0
win_edit.name = edit_title + " Window"
# ========== Window Image View ==========
class FormationsWindowImageView(View):
def draw(self, c, r):
c.backcolor = view_backcolor
c.erase_rect(r)
view = FormationsWindowImageView(size=win_edit.size)
# ========== Title ==========
title = Label(text=edit_title + ' File: ' + settings['file_name'])
title.font = title_font
title.width = title_width
title.height = title_height
title.x = (win_width - title_width) / 2
title.y = top_border
title.color = title_color
title.just = 'center'
general_display.append(title)
# ========== Action Button Declarations ==========
start_btn = Button("Start")
back_btn = Button("Back")
# ========== Mode Button Declarations ==========
simple_btn = Button("Simple")
advanced_btn = Button("Advanced")
# ========== Tool Button Declarations ==========
attribute_btn = Button("Add Search Attribute")
sort_btn = Button("Add Sort Attribute")
reset_btn = Button("Reset Results")
# ========== Simple Field Declarations ==========
rating_label = Label("Rating: ")
rating_tf = TextField()
name_label = Label("Name: ")
name_tf = TextField()
# ========== Action Button Functions ==========
def start_btn_func():
# Search for players based on attributes
if settings['edit_subject'] == 'players':
# Get the attributes to search with based on what mode is in use
if simple_btn.enabled == 0:
if len(name_tf.value) > 0:
search_dict = {'name_custom': (name_tf.value, 'exact')}
else:
search_dict = {}
if rating_tf.value.isdigit():
search_dict['rating'] = (int(rating_tf.value), 'exact')
else:
search_dict = attr_dict
# Get players from database to add and search
if settings['edit_type'] == 'add':
db_players = db_dict['player_db'][1]
if len(search_dict) == 0:
search_results = db_players
else:
search_results = db_players.search(search_dict)
search_results = PlayerDB.PlayerDB(search_results)
# Get players from list to edit or delete and search
else:
if len(search_dict) == 0:
search_results = list_players
else:
search_results = list_players.search(search_dict)
search_results = PlayerDB.PlayerDB(search_results)
# Sort players - if no attribute selected, use rating
if len(attr_list) == 0:
search_results.sort(['rating'], sort_order_radio_group.value)
else:
search_results.sort(attr_list, sort_order_radio_group.value)
# Get attributes list and avoid duplicates
attributes_list = []
for attr in attr_list:
if attributes_list.count(attr) == 0:
attributes_list.append(attr)
for attr_key in attr_dict.iterkeys():
if attributes_list.count(attr_key) == 0:
attributes_list.append(attr_key)
display_players(search_results, attributes_list, (0, num_results))
# Start button corresponds to formations
elif settings['edit_subject'] == 'formations':
# Get the attributes to search with based on what mode is in use
if simple_btn.enabled == 0:
if len(name_tf.value) > 0:
search_dict = {'name': (name_tf.value, 'exact')}
else:
search_dict = {}
else:
search_dict = attr_dict
# Get formations from database to add and search
if settings['edit_type'] == 'add':
db_formations = db_dict['formation_db'][1]
if len(search_dict) == 0:
search_results = db_formations
else:
search_results = db_formations.search(search_dict)
search_results = FormationDB.FormationDB(search_results)
# Get formations from list to edit or delete and search
else:
if len(search_dict) == 0:
search_results = list_formations
else:
search_results = list_formations.search(search_dict)
search_results = FormationDB.FormationDB(search_results)
# Sort formations - if no attribute selected, use name
if len(attr_list) == 0:
search_results.sort(['name'], sort_order_radio_group.value)
else:
search_results.sort(attr_list, sort_order_radio_group.value)
# Get attributes list and avoid duplicates
attributes_list = []
for attr in attr_list:
if attributes_list.count(attr) == 0:
attributes_list.append(attr)
for attr_key in attr_dict.iterkeys():
if attributes_list.count(attr_key) == 0:
attributes_list.append(attr_key)
display_formations(search_results, attributes_list, (0, num_results))
# Start button corresponds to teams
elif settings['edit_subject'] == 'teams':
stuff = 0
win_edit.become_target()
def back_btn_func():
# Clean up
reset_btn_func()
win_edit.hide()
PickFile.open_pick_file_window(win_edit.x, win_edit.y, db_dict, settings)
# ========== Search Type Button Functions ==========
def simple_btn_func():
simple_btn.enabled = 0
advanced_btn.enabled = 1
settings['edit_mode'] = 'simple'
for display_item in advanced_display:
view.remove(display_item)
clean_results()
for display_item in simple_display:
view.add(display_item)
name_tf.become_target()
def advanced_btn_func():
advanced_btn.enabled = 0
simple_btn.enabled = 1
settings['edit_mode'] = 'advanced'
for display_item in simple_display:
view.remove(display_item)
clean_results()
display_attributes()
for display_item in advanced_display:
view.add(display_item)
win_edit.become_target()
# ========== Tool Button Functions ==========
def attribute_btn_func():
# Delete results
del settings['messages']['results'][:]
attr_type = ''
if settings['edit_subject'] == 'players':
attr_type = 'player_search'
elif settings['edit_subject'] == 'formations':
attr_type = 'formation_search'
elif settings['team_subject'] == 'teams':
attr_type = 'player_search'
else:
print "Invalid edit_subject settings."
# Open new window and close current window
win_edit.hide()
AddAttribute.open_attribute_window(win_edit.x, win_edit.y,
db_dict, attr_dict, attr_list, attr_type, settings)
def sort_btn_func():
# Delete results
del settings['messages']['results'][:]
attr_type = ''
if settings['edit_subject'] == 'players':
attr_type = 'player_sort'
elif settings['edit_subject'] == 'formations':
attr_type = 'formation_sort'
elif settings['team_subject'] == 'teams':
attr_type = 'player_sort'
else:
print "Invalid edit_subject settings."
# Open new window and close current window
win_edit.hide()
AddAttribute.open_attribute_window(win_edit.x, win_edit.y, db_dict,
attr_dict, attr_list, attr_type, settings)
def display_attributes():
for search_item in settings['messages']['search']:
view.add(search_item)
for sort_item in settings['messages']['sort']:
view.add(sort_item)
def clean_results():
# Remove messages off page
for message in settings['messages']['search']:
view.remove(message)
for message in settings['messages']['sort']:
view.remove(message)
for message in settings['messages']['results']:
view.remove(message)
del settings['messages']['results'][:]
win_edit.become_target()
def reset_btn_func():
# Remove messages off page
for message in settings['messages']['search']:
view.remove(message)
for message in settings['messages']['sort']:
view.remove(message)
for message in settings['messages']['results']:
view.remove(message)
# Delete the attribute parameters for search and sort
attr_dict.clear()
del attr_list[:]
del settings['messages']['results'][:]
win_edit.become_target()
def player_bio_btn_func(player):
win_edit.become_target()
PlayerBio.open_player_bio_window(win_edit.x, win_edit.y, player, win_edit,
file_name=settings['file_name'], current_list=list_players)
win_edit.hide()
def formation_bio_btn_func(formation):
win_edit.become_target()
FormationBio.open_formation_bio_window(
win_edit.x, win_edit.y, formation, win_edit, settings['file_name'], list_formations)
win_edit.hide()
def add_btn_func(list_item, btn):
if settings['edit_subject'] == 'players':
# Check if player is already on selected players list
# Remove player from list
player_data = list_players.search({'id': (list_item['id'], 'exact')})
if len(player_data) > 0:
# Remove
list_players.db.remove(player_data[0])
# Save
list_players.sort(['rating'])
list_players.save(settings['file_name'], 'list', True)
# Switch button title
btn.title = "+"
# Add player to the list
else:
# Add
list_players.db.append(list_item)
# Save
list_players.sort(['rating'])
list_players.save(settings['file_name'], 'list', True)
# Switch button title
btn.title = "-"
elif settings['edit_subject'] == 'formations':
# Check if formation is already on selected formations list
# Remove formation from list
if list_item in list_formations.db:
# Remove
list_formations.db.remove(list_item)
# Save
list_formations.sort(['name'])
list_formations.save(settings['file_name'], 'list', True)
# Switch button title
btn.title = "+"
# Add formation to the list
else:
# Add
list_formations.db.append(list_item)
# Save
list_formations.sort(['name'])
list_formations.save(settings['file_name'], 'list', True)
# Switch button title
btn.title = "-"
win_edit.become_target()
def pos_btn_func(player, btn):
# Open window to get position
win_edit.hide()
PickPosition.open_pick_position_window(win_edit.x, win_edit.y, player, list_players, settings, btn, win_edit)
win_edit.become_target()
# ========== Action Buttons ==========
start_btn.x = (win_width - 2*small_button_width - small_button_spacing) / 2
start_btn.y = title.bottom + small_button_top_spacing
start_btn.height = small_button_height
start_btn.width = small_button_width
start_btn.font = small_button_font
start_btn.action = start_btn_func
start_btn.style = 'default'
start_btn.color = small_button_color
start_btn.just = 'right'
general_display.append(start_btn)
back_btn.x = start_btn.right + small_button_spacing
back_btn.y = start_btn.top
back_btn.height = small_button_height
back_btn.width = small_button_width
back_btn.font = small_button_font
back_btn.action = back_btn_func
back_btn.style = 'default'
back_btn.color = small_button_color
back_btn.just = 'right'
general_display.append(back_btn)
# ========== Edit Type Radio Buttons ==========
def get_edit_type_rg():
settings['edit_type'] = edit_type_radio_group.value
win_edit.become_target()
edit_type_radio_group = RadioGroup(action=get_edit_type_rg)
radio_btn_width = 125
radio_btn_space = 5
# Edit Type
type_add_radio_btn = RadioButton()
if settings['edit_subject'] == 'players':
type_add_radio_btn.title = 'Add Players'
elif settings['edit_subject'] == 'formations':
type_add_radio_btn.title = 'Add Formations'
elif settings['edit_subject'] == 'teams':
type_add_radio_btn.title = 'Add Teams'
type_add_radio_btn.width = radio_btn_width
type_add_radio_btn.x = (win_edit.width - 2*radio_btn_width - radio_btn_space) / 2
type_add_radio_btn.y = start_btn.bottom + small_button_top_spacing
type_add_radio_btn.group = edit_type_radio_group
type_add_radio_btn.value = 'add'
general_display.append(type_add_radio_btn)
type_edit_radio_btn = RadioButton()
if settings['edit_subject'] == 'players':
type_edit_radio_btn.title = 'Edit/Delete Players'
elif settings['edit_subject'] == 'formations':
type_edit_radio_btn.title = 'Edit/Delete Forms'
elif settings['edit_subject'] == 'teams':
type_edit_radio_btn.title = 'Edit/Delete Teams'
type_edit_radio_btn.width = radio_btn_width
type_edit_radio_btn.x = type_add_radio_btn.right + radio_btn_space
type_edit_radio_btn.y = type_add_radio_btn.top
type_edit_radio_btn.group = edit_type_radio_group
type_edit_radio_btn.value = 'edit'
general_display.append(type_edit_radio_btn)
edit_type_radio_group.value = settings['edit_type']
# ========== Mode Buttons ==========
simple_btn.x = (win_width - 2*small_button_width - small_button_spacing) / 2
simple_btn.y = type_add_radio_btn.bottom + small_button_top_spacing
simple_btn.height = small_button_height
simple_btn.width = small_button_width
simple_btn.font = small_button_font
simple_btn.action = simple_btn_func
simple_btn.style = 'default'
simple_btn.color = small_button_color
simple_btn.just = 'right'
if settings['edit_mode'] == 'simple':
simple_btn.enabled = 0
general_display.append(simple_btn)
advanced_btn.x = simple_btn.right + small_button_spacing
advanced_btn.y = simple_btn.top
advanced_btn.height = small_button_height
advanced_btn.width = small_button_width
advanced_btn.font = small_button_font
advanced_btn.action = advanced_btn_func
advanced_btn.style = 'default'
advanced_btn.color = small_button_color
advanced_btn.just = 'right'
if settings['edit_mode'] == 'advanced':
advanced_btn.enabled = 0
general_display.append(advanced_btn)
# ========== Sort Order Radio Buttons ==========
def get_attribute_sort_order_rg():
settings['sort_order'] = sort_order_radio_group.value
win_edit.become_target()
sort_order_radio_group = RadioGroup(action=get_attribute_sort_order_rg)
asc_desc_radio_btn_width = 75
asc_msg_width = 80
radio_btn_space = 5
asc_desc_rg_msg = Label(text="Sort Order:", font=std_tf_font, width=asc_msg_width, height=std_tf_height,
color=title_color)
asc_desc_rg_msg.x = (win_edit.width - 2*asc_desc_radio_btn_width - radio_btn_space - asc_msg_width) / 2
asc_desc_rg_msg.y = simple_btn.bottom + radio_btn_space
general_display.append(asc_desc_rg_msg)
descend_radio_btn = RadioButton("Descending")
descend_radio_btn.width = asc_desc_radio_btn_width
descend_radio_btn.x = asc_desc_rg_msg.right
descend_radio_btn.y = asc_desc_rg_msg.top
descend_radio_btn.group = sort_order_radio_group
descend_radio_btn.value = True
general_display.append(descend_radio_btn)
ascend_radio_btn = RadioButton("Ascending")
ascend_radio_btn.width = asc_desc_radio_btn_width
ascend_radio_btn.x = descend_radio_btn.right + radio_btn_space
ascend_radio_btn.y = descend_radio_btn.top
ascend_radio_btn.group = sort_order_radio_group
ascend_radio_btn.value = False
general_display.append(ascend_radio_btn)
sort_order_radio_group.value = settings['sort_order']
# ========== Tool Buttons ==========
attribute_btn.x = (win_width - 3*small_button_width - 2*small_button_spacing) / 2
attribute_btn.y = asc_desc_rg_msg.bottom + 5
attribute_btn.height = small_button_height
attribute_btn.width = small_button_width
attribute_btn.font = small_button_font
attribute_btn.action = attribute_btn_func
attribute_btn.style = 'default'
attribute_btn.color = small_button_color
attribute_btn.just = 'right'
advanced_display.append(attribute_btn)
sort_btn.x = attribute_btn.right + small_button_spacing
sort_btn.y = attribute_btn.top
sort_btn.height = small_button_height
sort_btn.width = small_button_width
sort_btn.font = small_button_font
sort_btn.action = sort_btn_func
sort_btn.style = 'default'
sort_btn.color = small_button_color
sort_btn.just = 'right'
advanced_display.append(sort_btn)
reset_btn.x = sort_btn.right + small_button_spacing
reset_btn.y = attribute_btn.top
reset_btn.height = small_button_height
reset_btn.width = small_button_width
reset_btn.font = small_button_font
reset_btn.action = reset_btn_func
reset_btn.style = 'default'
reset_btn.color = small_button_color
reset_btn.just = 'right'
advanced_display.append(reset_btn)
# ========== Simple Fields ==========
label_width = 50
rating_tf_width = 30
field_spacing = 20
rating_label.font = std_tf_font
rating_label.width = label_width
rating_label.height = std_tf_height
rating_label.x = (win_edit.width - 2*label_width - std_tf_width - rating_tf_width - field_spacing) / 2
rating_label.y = asc_desc_rg_msg.bottom + 5
rating_label.color = title_color
if settings['edit_subject'] != 'formations':
simple_display.append(rating_label)
rating_tf.font = std_tf_font
rating_tf.width = rating_tf_width
rating_tf.height = 25
rating_tf.x = rating_label.right
rating_tf.y = rating_label.top
if settings['edit_subject'] != 'formations':
simple_display.append(rating_tf)
name_label.font = std_tf_font
name_label.width = label_width
name_label.height = std_tf_height
if settings['edit_subject'] != 'formations':
name_label.x = rating_tf.right + field_spacing
else:
name_label.x = (win_edit.width - label_width - std_tf_width) / 2
name_label.y = rating_label.top
name_label.color = title_color
simple_display.append(name_label)
name_tf.font = std_tf_font
name_tf.width = std_tf_width
name_tf.height = 25
name_tf.x = name_label.right
name_tf.y = rating_label.top
simple_display.append(name_tf)
# ========== Messages ==========
lowest_msg_l = start_btn.top
lowest_msg_r = start_btn.top
attr_msg_offset = 25
# Attribute Messages
del settings['messages']['search'][:]
del settings['messages']['sort'][:]
if len(attr_dict) > 0:
settings['messages']['search'].append(Label(text="Search Attributes:", font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=attr_msg_offset, y=lowest_msg_l,
color=title_color))
lowest_msg_l += std_tf_height
for key, value in attr_dict.iteritems():
msg_text = format_attr_name(key) + ": "
if key in ['id', 'baseId', 'nationId', 'leagueId', 'clubId']:
msg_text += str(value[0])
elif type(value[0]) is int:
msg_text += value[1].capitalize() + ' ' + str(value[0])
elif value[1] == 'not':
msg_text += value[1].capitalize() + ' "' + str(value[0]) + '"'
else:
msg_text += '"' + str(value[0]) + '"'
attr_label = Label(text=msg_text, font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=attr_msg_offset, y=lowest_msg_l, color=title_color)
lowest_msg_l += std_tf_height
settings['messages']['search'].append(attr_label)
if len(attr_list) > 0:
settings['messages']['sort'].append(Label(text="Sort Attributes:", font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=advanced_btn.right + 3*attr_msg_offset,
y=lowest_msg_r, color=title_color))
lowest_msg_r += std_tf_height
for value in attr_list:
attr_label = Label(text=(format_attr_name(value)), font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=advanced_btn.right + 3*attr_msg_offset, y=lowest_msg_r,
color=title_color)
lowest_msg_r += std_tf_height
settings['messages']['sort'].append(attr_label)
# ========== Previous, Add to List, Next Buttons ==========
previous_btn = Button("<<< Previous %d" % num_results)
add_to_list_btn = Button()
next_btn = Button("Next %d >>>" % num_results)
total_num_results_label = Label()
pages_label = Label()
def add_to_list_btn_func(input_list, func_type):
item_list = copy.deepcopy(input_list)
if settings['edit_subject'] == 'players':
if func_type == 'add all':
added_players = []
# Add current results to player list
for player in item_list:
if list_players.db.count(player) == 0:
list_players.db.append(player)
added_players.append(player)
# Sort
list_players.sort(['rating'])
# Save
list_players.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Remove Added Players"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'remove select')
# Keep track of just added players
settings['messages']['players_changed'] = added_players
elif func_type == 'remove all':
removed_players = []
# Remove current results from player list
for player in item_list:
if list_players.db.count(player) > 0:
list_players.db.remove(player)
removed_players.append(player)
# Sort
list_players.sort(['rating'])
# Save
list_players.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Add Removed Players"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'add select')
# Keep track of just removed players
settings['messages']['players_changed'] = removed_players
elif func_type == 'add select':
# Add select players back to player list
for player in settings['messages']['players_changed']:
if list_players.db.count(player) == 0:
list_players.db.append(player)
# Sort
list_players.sort(['rating'])
# Save
list_players.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Remove Added Players"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'remove select')
elif func_type == 'remove select':
# Remove select players from player list
for player in settings['messages']['players_changed']:
if list_players.db.count(player) > 0:
list_players.db.remove(player)
# Sort
list_players.sort(['rating'])
# Save
list_players.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Add Removed Players"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'add select')
elif settings['edit_subject'] == 'formations':
if func_type == 'add all':
added_formations = []
# Add current results to formation list
for formation in item_list:
if list_formations.db.count(formation) == 0:
list_formations.db.append(formation)
added_formations.append(formation)
# Sort
list_formations.sort(['name'])
# Save
list_formations.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Remove Added Forms"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'remove select')
# Keep track of just added formations
settings['messages']['formations_changed'] = added_formations
elif func_type == 'remove all':
removed_formations = []
# Remove current results from formation list
for formation in item_list:
if formation in list_formations.db:
list_formations.db.remove(formation)
removed_formations.append(formation)
# Sort
list_formations.sort(['name'])
# Save
list_formations.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Add Removed Forms"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'add select')
# Keep track of just removed formations
settings['messages']['formations_changed'] = removed_formations
elif func_type == 'add select':
# Add select formations back to formation list
for formation in settings['messages']['formations_changed']:
if list_formations.db.count(formation) == 0:
list_formations.db.append(formation)
# Sort
list_formations.sort(['name'])
# Save
list_formations.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Remove Added Forms"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'remove select')
elif func_type == 'remove select':
# Remove select formations from formation list
for formation in settings['messages']['formations_changed']:
if list_formations.db.count(formation) > 0:
list_formations.db.remove(formation)
# Sort
list_formations.sort(['name'])
# Save
list_formations.save(settings['file_name'], 'list', True)
# Change button title and action
add_to_list_btn.title = "Add Removed Forms"
add_to_list_btn.action = (add_to_list_btn_func, item_list, 'add select')
del item_list
win_edit.become_target()
def previous_btn_func(display_db=None, attributes=None, index_range=None):
if display_db is not None:
# display previous results
if settings['edit_subject'] == 'players':
display_players(display_db, attributes, index_range)
elif settings['edit_subject'] == 'formations':
display_formations(display_db, attributes, index_range)
#elif settings['edit_subject'] == 'teams':
#display_teams(display_db, attributes, index_range)
win_edit.become_target()
def next_btn_func(display_db=None, attributes=None, index_range=None):
if display_db is not None:
# display next results
if settings['edit_subject'] == 'players':
display_players(display_db, attributes, index_range)
elif settings['edit_subject'] == 'formations':
display_formations(display_db, attributes, index_range)
# elif settings['edit_subject'] == 'teams':
# display_teams(display_db, attributes, index_range)
win_edit.become_target()
add_to_list_btn.x = attribute_btn.right + small_button_spacing
add_to_list_btn.y = attribute_btn.bottom + 5
add_to_list_btn.height = tiny_button_height
add_to_list_btn.width = small_button_width
add_to_list_btn.font = small_button_font
add_to_list_btn.style = 'default'
add_to_list_btn.color = small_button_color
add_to_list_btn.just = 'right'
previous_btn.height = tiny_button_height
previous_btn.width = small_button_width
previous_btn.x = add_to_list_btn.left - previous_btn.width - small_button_spacing
previous_btn.y = add_to_list_btn.top
previous_btn.font = small_button_font
previous_btn.action = previous_btn_func
previous_btn.style = 'default'
previous_btn.color = small_button_color
previous_btn.just = 'right'
next_btn.height = tiny_button_height
next_btn.width = small_button_width
next_btn.x = add_to_list_btn.right + small_button_spacing
next_btn.y = add_to_list_btn.top
next_btn.font = small_button_font
next_btn.action = next_btn_func
next_btn.style = 'default'
next_btn.color = small_button_color
next_btn.just = 'right'
total_num_results_label.font = std_tf_font
total_num_results_label.width = 100
total_num_results_label.height = std_tf_height
total_num_results_label.x = previous_btn.left + - 100 - 10
total_num_results_label.y = add_to_list_btn.top
total_num_results_label.color = title_color
total_num_results_label.just = 'right'
pages_label.font = std_tf_font
pages_label.width = 125
pages_label.height = std_tf_height
pages_label.x = next_btn.right + 10
pages_label.y = add_to_list_btn.top
pages_label.color = title_color
pages_label.just = 'left'
# ========== Display players from search ==========
def display_players(display_db, attributes, index_range):
# Remove old messages off page
for message in settings['messages']['results']:
view.remove(message)
del settings['messages']['results'][:]
# Add navigation buttons to page
if settings['edit_type'] == 'add':
add_to_list_btn.title = 'Add All Players'
add_to_list_btn.action = (add_to_list_btn_func, display_db.db, 'add all')
elif settings['edit_type'] == 'edit':
add_to_list_btn.title = 'Remove All Players'
add_to_list_btn.action = (add_to_list_btn_func, display_db.db, 'remove all')
else:
print "Invalid edit type."
previous_range = (index_range[0]-num_results, index_range[0])
previous_btn.action = (previous_btn_func, display_db, attributes, previous_range)
next_range = (index_range[1], index_range[1]+num_results)
next_btn.action = (next_btn_func, display_db, attributes, next_range)
total_num_results_label.text = str(len(display_db.db)) + " Players"
pages_label.text = "Page %d of %d" % (int(index_range[1]/num_results),
math.ceil(len(display_db.db) / float(num_results)))
if index_range[0] > 0:
previous_btn.enabled = 1
else:
previous_btn.enabled = 0
if index_range[1] <= len(display_db.db) - 1:
next_btn.enabled = 1
else:
next_btn.enabled = 0
settings['messages']['results'].append(add_to_list_btn)
settings['messages']['results'].append(previous_btn)
settings['messages']['results'].append(next_btn)
settings['messages']['results'].append(total_num_results_label)
settings['messages']['results'].append(pages_label)
# Print out labels
labels = player_info_labels(attributes)
stat_index = 1
spacing_list = [25, 125, 40, 40, 65, 115, 115, 115, 40]
left_border = (win_width - sum(spacing_list[:-1]) - (len(labels) - len(spacing_list) + 2) * spacing_list[-1])/2
msg_x = left_border + spacing_list[0]
msg_y = add_to_list_btn.bottom + 5
for info_label in labels:
player_label = Label(text=info_label, font=std_tf_font_bold, width=spacing_list[stat_index]-5,
height=std_tf_height, x=msg_x, y=msg_y, color=title_color)
settings['messages']['results'].append(player_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list)-1:
stat_index += 1
msg_y += std_tf_height + 5
# Print out players
for idx, player in enumerate(display_db.db[index_range[0]:index_range[1]]):
msg_x = left_border
player_stats = player_info(player, attributes)
stat_index = 0
player_data = list_players.search({'id': (player['id'], 'exact')})
if len(player_data) > 0:
add_btn_text = '-'
temp_player = player_data[0]
else:
add_btn_text = '+'
temp_player = player
add_btn = Button(title=add_btn_text, width=spacing_list[stat_index]-5, height=15, x=msg_x, y=msg_y)
add_btn.action = (add_btn_func, temp_player, add_btn)
settings['messages']['results'].append(add_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
# Check for names that are too long
name = player_stats[0]
if len(name) > 20:
name = player['lastName']
bio_btn = Button(title=name, width=spacing_list[stat_index]-5, height=15, x=msg_x, y=msg_y,
action=(player_bio_btn_func, temp_player))
settings['messages']['results'].append(bio_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
for player_stat in player_stats[1:]:
player_label = Label(text=player_stat, font=small_button_font, width=spacing_list[stat_index]-5,
height=std_tf_height, x=msg_x, y=msg_y, color=title_color)
settings['messages']['results'].append(player_label)
# Save values for position edit button
if stat_index == 3:
edit_x = msg_x
edit_y = msg_y
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
if settings['edit_type'] == 'edit':
pos_btn = Button(title=player_stats[2], width=28, height=15, x=edit_x, y=edit_y)
pos_btn.action = (pos_btn_func, player, pos_btn)
settings['messages']['results'].append(pos_btn)
msg_y += std_tf_height
for results_msg in settings['messages']['results']:
view.add(results_msg)
# ========== Display formations from search ==========
def display_formations(display_db, attributes, index_range):
# Remove old messages off page
for message in settings['messages']['results']:
view.remove(message)
del settings['messages']['results'][:]
# Add navigation buttons to page
if settings['edit_type'] == 'add':
add_to_list_btn.title = 'Add All Formations'
add_to_list_btn.action = (add_to_list_btn_func, display_db.db, 'add all')
elif settings['edit_type'] == 'edit':
add_to_list_btn.title = 'Remove All Formations'
add_to_list_btn.action = (add_to_list_btn_func, display_db.db, 'remove all')
else:
print "Invalid edit type."
previous_range = (index_range[0]-num_results, index_range[0])
previous_btn.action = (previous_btn_func, display_db, attributes, previous_range)
next_range = (index_range[1], index_range[1]+num_results)
next_btn.action = (next_btn_func, display_db, attributes, next_range)
total_num_results_label.text = str(len(display_db.db)) + " Formations"
pages_label.text = "Page %d of %d" % (int(index_range[1]/num_results),
math.ceil(len(display_db.db)/float(num_results)))
if index_range[0] > 0:
previous_btn.enabled = 1
else:
previous_btn.enabled = 0
if index_range[1] <= len(display_db.db) - 1:
next_btn.enabled = 1
else:
next_btn.enabled = 0
settings['messages']['results'].append(add_to_list_btn)
settings['messages']['results'].append(previous_btn)
settings['messages']['results'].append(next_btn)
settings['messages']['results'].append(total_num_results_label)
settings['messages']['results'].append(pages_label)
# Print out labels
labels = formation_info_labels()
stat_index = 1
spacing_list = [25, 100, 100, 55, 55, 55, 55, 140, 160]
left_border = (win_edit.width - sum(spacing_list))/2
msg_x = left_border + spacing_list[0]
msg_y = add_to_list_btn.bottom + 5
for info_label in labels:
formation_label = Label(text=info_label, font=std_tf_font_bold, width=spacing_list[stat_index]-5,
height=std_tf_height, x=msg_x, y=msg_y, color=title_color)
settings['messages']['results'].append(formation_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list)-1:
stat_index += 1
msg_y += std_tf_height + 5
# Print out formations
for idx, formation in enumerate(display_db.db[index_range[0]:index_range[1]]):
msg_x = left_border
formation_stats = formation_info(formation)
stat_index = 0
if formation in list_formations.db:
add_btn_text = '-'
else:
add_btn_text = '+'
add_btn = Button(title=add_btn_text, width=spacing_list[stat_index]-5, height=15, x=msg_x, y=msg_y)
add_btn.action = (add_btn_func, formation, add_btn)
settings['messages']['results'].append(add_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
bio_btn = Button(title=formation['name'], width=spacing_list[stat_index]-5, height=15, x=msg_x, y=msg_y,
action=(formation_bio_btn_func, formation))
settings['messages']['results'].append(bio_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
for formation_stat in formation_stats[1:]:
formation_label = Label(text=formation_stat, font=small_button_font, width=spacing_list[stat_index]-5,
height=std_tf_height, x=msg_x, y=msg_y, color=title_color)
settings['messages']['results'].append(formation_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height
for results_msg in settings['messages']['results']:
view.add(results_msg)
# ========== Add components to view and add view to window ==========
for item in general_display:
view.add(item)
if settings['edit_mode'] == 'simple':
for item in simple_display:
view.add(item)
elif settings['edit_mode'] == 'advanced':
for item in advanced_display:
view.add(item)
display_attributes()
for msg in settings['messages']['results']:
view.add(msg)
win_edit.add(view)
view.become_target()
win_edit.show()
def open_enter_text_window(window_x, window_y, db_dict, settings, box_type, fill_text='', file_prefix=''):
general_display = []
if box_type == 'rename':
title = 'Rename File'
old_file_name = fill_text
message_text = 'Enter new file name.'
elif box_type == 'duplicate':
title = 'Duplicate File'
old_file_name = fill_text
message_text = 'Enter new file name.'
elif box_type == 'download':
title = 'Download Player Database'
message_text = 'Enter player database name.'
download_settings = {'overwrite_counter': 0, 'last_entered_text': ''}
else:
title = 'Enter Text'
message_text = 'Enter text.'
# ========== Window ==========
win_enter_text = Window()
win_enter_text.title = title
win_enter_text.auto_position = False
win_enter_text.position = (window_x+100, window_y+100)
win_enter_text.size = (win_width-200, win_height-400)
win_enter_text.resizable = 0
win_enter_text.name = enter_text_title + " Window"
win_enter_text.show()
# ========== Window Image View ==========
class StartWindowImageView(View):
def draw(self, c, r):
c.backcolor = view_backcolor
c.erase_rect(r)
view = StartWindowImageView(size=win_enter_text.size)
# ========== Title ==========
title = Label(text=title)
title.font = title_font
title.width = title_width
title.height = title_height
title.x = (win_enter_text.width - title_width) / 2
title.y = top_border
title.color = title_color
title.just = 'center'
general_display.append(title)
# ========== Message Label ==========
message = Label(font=title_font_2, width=win_enter_text.width - 50, height=title_height,
x=25, y=title.bottom + top_border, color=title_color, just='center')
message.text = message_text
general_display.append(message)
# ========== Button Declarations ==========
enter_btn = Button("Enter")
back_btn = Button("Back")
# ========== Button Functions ==========
def enter_btn_func():
# Rename file
if box_type == 'rename':
# Get new name
new_file_name = value_tf.value
if len(new_file_name) > 0:
if not isfile('JSONs/' + file_prefix + new_file_name + '.json'):
# Rename file
rename('JSONs/' + file_prefix + old_file_name + '.json',
'JSONs/' + file_prefix + new_file_name + '.json')
# Disable pick file back button in case the selected file has changed
settings['file_changes'] = True
PickFile.open_pick_file_window(window_x, window_y, db_dict, settings)
win_enter_text.hide()
else:
message.text = "A file with that name already exists."
else:
message.text = "File name must be at least 1 character."
# Duplicate file
elif box_type == 'duplicate':
# Get new name
duplicate_file_name = value_tf.value
if len(duplicate_file_name) > 0:
if not isfile('JSONs/' + file_prefix + duplicate_file_name + '.json'):
# Create duplicate file
copyfile('JSONs/' + file_prefix + old_file_name + '.json',
'JSONs/' + file_prefix + duplicate_file_name + '.json')
PickFile.open_pick_file_window(window_x, window_y, db_dict, settings)
win_enter_text.hide()
else:
message.text = "A file with that name already exists."
else:
message.text = "File name must be at least 1 character."
elif box_type == 'download':
valid_name = False
# Get new player database name
new_player_db_name = value_tf.value
if new_player_db_name != download_settings['last_entered_text']:
download_settings['overwrite_counter'] = 0
message.text = message_text
if len(new_player_db_name) > 0:
if not isfile('JSONs/play_db_' + new_player_db_name + '.json'):
valid_name = True
else:
download_settings['last_entered_text'] = new_player_db_name
if download_settings['overwrite_counter'] == 0:
download_settings['overwrite_counter'] += 1
message.text = "File already exists. Overwrite?"
elif download_settings['overwrite_counter'] == 1:
download_settings['overwrite_counter'] += 1
message.text = "Are you sure you want to overwrite?"
elif download_settings['overwrite_counter'] == 2:
download_settings['overwrite_counter'] += 1
message.text = "Really, really, really sure?"
elif download_settings['overwrite_counter'] >= 3:
valid_name = True
if valid_name:
# Open status window to start download of players
StatusWindow.open_status_window(win_enter_text.x, win_enter_text.y, db_dict,
get_prices=sort_order_radio_group.value,
file_name=new_player_db_name, settings=settings,
win_previous=win_enter_text, win_next='FilesMenu')
win_enter_text.hide()
else:
message.text = "File name must be at least 1 character."
def back_btn_func():
if box_type == 'download':
FilesMenu.open_files_menu(window_x, window_y, db_dict, settings)
else:
PickFile.open_pick_file_window(window_x, window_y, db_dict, settings)
win_enter_text.hide()
# ========== Buttons ==========
enter_btn.x = (win_enter_text.width - 2*button_width - button_spacing) / 2
enter_btn.y = win_enter_text.height - 70
enter_btn.height = button_height
enter_btn.width = button_width
enter_btn.font = button_font
enter_btn.action = enter_btn_func
enter_btn.style = 'default'
enter_btn.color = button_color
enter_btn.just = 'right'
general_display.append(enter_btn)
back_btn.x = enter_btn.right + button_spacing
back_btn.y = enter_btn.top
back_btn.height = button_height
back_btn.width = button_width
back_btn.font = button_font
back_btn.action = back_btn_func
back_btn.style = 'default'
back_btn.color = button_color
back_btn.just = 'right'
general_display.append(back_btn)
# ========== Value Textfield ==========
value_tf = TextField()
value_tf.width = 200
value_tf.x = (win_enter_text.width - value_tf.width) / 2
value_tf.y = enter_btn.top - 65
value_tf.height = 25
value_tf.font = std_tf_font
value_tf.value = fill_text
general_display.append(value_tf)
def get_attribute_sort_order_rg():
settings['order_rg'] = sort_order_radio_group.value
win_enter_text.become_target()
sort_order_radio_group = RadioGroup(action=get_attribute_sort_order_rg)
prices_label_width = 85
prices_button_width = 50
prices_label = Label(text="Get Prices?:", font=std_tf_font, width=prices_label_width, height=std_tf_height,
color=title_color)
prices_label.x = (win_enter_text.width - 2*prices_button_width - 10 - prices_label_width) / 2
prices_label.y = enter_btn.top - 38
prices_label.just = 'center'
prices_yes = RadioButton("Yes")
prices_yes.width = prices_button_width
prices_yes.x = prices_label.right + 5
prices_yes.y = prices_label.top
prices_yes.group = sort_order_radio_group
prices_yes.value = True
prices_no = RadioButton("No")
prices_no.width = prices_button_width
prices_no.x = prices_yes.right + 5
prices_no.y = prices_label.top
prices_no.group = sort_order_radio_group
prices_no.value = False
sort_order_radio_group.value = True
if box_type == 'download':
value_tf.y = enter_btn.top - 75
general_display.append(prices_label)
general_display.append(prices_yes)
general_display.append(prices_no)
# ========== Add buttons to window ==========
for item in general_display:
view.add(item)
win_enter_text.add(view)
view.become_target()
win_enter_text.show()
def open_start_menu(window_x, window_y, db_dict):
view_item_list = []
# Load console type
settings = {'console_type': ''}
with open(config_filename, 'r') as f:
settings["console_type"] = json.load(f)['console_type']
f.close()
# ========== Window ==========
win_start = Window()
win_start.title = start_win_title
win_start.auto_position = False
win_start.position = (window_x, window_y)
win_start.size = (win_width, win_height)
win_start.resizable = 0
win_start.name = start_title + " Window"
# ========== Window Image View ==========
class StartWindowImageView(View):
def draw(self, c, r):
c.backcolor = view_backcolor
c.erase_rect(r)
image_pos = ((win_start.width - start_window_image.width)/2, search_btn.bottom + title_border)
src_rect = start_window_image.bounds
dst_rect = Geometry.offset_rect(src_rect, image_pos)
start_window_image.draw(c, src_rect, dst_rect)
view = StartWindowImageView(size=win_start.size)
# ========== Start Window Image ==========
start_window_image = Image(file='Images/start_menu_image.jpg')
# ========== Title ==========
title = Label(text=start_title)
title.font = title_font
title.width = title_width
title.height = title_height
title.x = (win_width - title_width) / 2
title.y = top_border
title.color = title_color
title.just = 'center'
view_item_list.append(title)
# ========== Sort Order Radio Buttons ==========
def save_console_type():
"""
Save the console type
"""
# Load configurations
from AppConfig import config_filename
with open(config_filename, 'r') as config_file:
configurations = json.load(config_file)
config_file.close()
# Edit configurations
configurations['console_type'] = settings["console_type"]
# Save the settings
with open(config_filename, 'w') as config_file:
json.dump(configurations, config_file)
config_file.close()
def get_attribute_console_rg():
settings["console_type"] = console_radio_group.value
win_start.become_target()
save_console_type()
console_radio_group = RadioGroup(action=get_attribute_console_rg)
console_radio_btn_width = 75
console_label_width = 80
radio_btn_space = 5
console_rg_label = Label(text="Sort Order:", font=std_tf_font, width=console_label_width, height=std_tf_height,
color=title_color)
console_rg_label.x = (win_start.width - 2 * console_radio_btn_width - radio_btn_space - console_label_width) / 2
console_rg_label.y = title.bottom + radio_btn_space
view_item_list.append(console_rg_label)
playstation_radio_btn = RadioButton("PlayStation")
playstation_radio_btn.width = console_radio_btn_width
playstation_radio_btn.x = console_rg_label.right
playstation_radio_btn.y = console_rg_label.top
playstation_radio_btn.group = console_radio_group
playstation_radio_btn.value = 'PLAYSTATION'
view_item_list.append(playstation_radio_btn)
xbox_radio_btn = RadioButton("Xbox")
xbox_radio_btn.width = console_radio_btn_width
xbox_radio_btn.x = playstation_radio_btn.right + radio_btn_space
xbox_radio_btn.y = playstation_radio_btn.top
xbox_radio_btn.group = console_radio_group
xbox_radio_btn.value = 'XBOX'
view_item_list.append(xbox_radio_btn)
if settings["console_type"] in ['PLAYSTATION', 'XBOX']:
console_radio_group.value = settings["console_type"]
else:
print "Invalid console type."
# ========== Button Declarations ==========
search_btn = Button("Search")
teams_btn = Button("Teams")
files_btn = Button("Files")
button_list = [search_btn,
teams_btn,
files_btn]
# ========== Button Functions ==========
def search_btn_func():
win_start.hide()
SearchMenu.open_search_menu(win_start.x, win_start.y, db_dict)
def teams_btn_func():
win_start.hide()
TeamsMenu.open_teams_menu(win_start.x, win_start.y, db_dict)
def files_btn_func():
win_start.hide()
FilesMenu.open_files_menu(win_start.x, win_start.y, db_dict)
# ========== Buttons ==========
search_btn.x = (win_width - len(button_list)*button_width - (len(button_list)-1)*button_spacing) / 2
search_btn.y = console_rg_label.bottom + title_border
search_btn.height = button_height
search_btn.width = button_width
search_btn.font = button_font
search_btn.action = search_btn_func
search_btn.style = 'default'
search_btn.color = button_color
search_btn.just = 'right'
view_item_list.append(search_btn)
teams_btn.x = button_spacing + search_btn.right
teams_btn.y = search_btn.top
teams_btn.height = button_height
teams_btn.width = button_width
teams_btn.font = button_font
teams_btn.action = teams_btn_func
teams_btn.style = 'default'
teams_btn.color = button_color
teams_btn.just = 'right'
view_item_list.append(teams_btn)
files_btn.x = button_spacing + teams_btn.right
files_btn.y = search_btn.top
files_btn.height = button_height
files_btn.width = button_width
files_btn.font = button_font
files_btn.action = files_btn_func
files_btn.style = 'default'
files_btn.color = button_color
files_btn.just = 'right'
view_item_list.append(files_btn)
# ========== Add components to view and add view to window ==========
for item in view_item_list:
view.add(item)
win_start.add(view)
view.become_target()
win_start.show()
def open_attribute_window(window_x, window_y, db_dict, attr_dict, attr_list, attr_type, settings):
# ========== Window ==========
win_attribute = Window()
win_attribute.title = attribute_win_title
win_attribute.auto_position = False
win_attribute.position = (window_x, window_y)
win_attribute.size = (win_width, 500)
win_attribute.resizable = 0
win_attribute.name = attribute_title + " Window"
win_attribute.show()
# ========== Window Image View ==========
class AddAttributeWindowImageView(View):
def draw(self, c, r):
c.backcolor = view_backcolor
c.erase_rect(r)
view = AddAttributeWindowImageView(size=win_attribute.size)
attribute_display_list = []
# ========== Title ==========
title = Label(text=attribute_title)
title.font = title_font
title.width = title_width
title.height = title_height
title.x = (win_width - title_width) / 2
title.y = top_border
title.color = title_color
title.just = 'center'
view.add(title)
# ========== Button Declarations ==========
enter_btn = Button("Enter")
erase_btn = Button("Erase List")
back_btn = Button("Back")
# ========== Radio Button Action ==========
def selection_made():
enter_btn.enabled = 1
win_attribute.become_target()
def comp_selection_made():
win_attribute.become_target()
def attribute_button_func(attr_value):
# Check for doubles
if attr_list.count(attr_value) == 0:
attr_list.append(attr_value)
show_attributes(attribute_display_list)
erase_btn.enabled = 1
win_attribute.become_target()
# ========== Attribute Radio Buttons ==========
radio_group = RadioGroup(action=selection_made)
radio_button_list = []
buttons_per_column = 11
config_file = '_attributes'
if attr_type[:4] == 'play':
config_file = 'player' + config_file
elif attr_type[:4] == 'form':
config_file = 'formation' + config_file
elif attr_type[:4] == 'team':
config_file = 'team' + config_file
else:
print "Invalid attribute type."
with open(config_filename, 'r') as f:
attributes_list = json.load(f)[config_file]['all']
f.close()
for idx, attribute in enumerate(attributes_list):
if attr_type[-6:] == 'search':
button = RadioButton(attribute)
button.group = radio_group
button.value = attribute
button.title = format_attr_name(attribute)
elif attr_type[-4:] == 'sort':
button = Button()
button.action = (attribute_button_func, attribute)
button.title = format_attr_name(attribute)
else:
print "Invalid attribute type."
if attr_type[:4] == 'play':
if idx < buttons_per_column:
button.width = 45
button.x = (idx / buttons_per_column) * (button.width + 5) + 5
elif idx < 3*buttons_per_column:
button.width = 100
button.x = (idx / buttons_per_column) * (button.width + 5) - 50
elif idx < 4*buttons_per_column:
button.width = 110
button.x = (idx / buttons_per_column) * (button.width + 5) - 80
else:
button.width = 100
button.x = (idx / buttons_per_column) * (button.width + 5) - 40
button.y = (idx % buttons_per_column) * 25 + title.bottom - 5
# Added for 81st item
"""if idx == 80:
button.x = 7 * (button.width + 5) - 40
button.y = 10 * 25 + title.bottom + 5"""
elif attr_type[:4] == 'form':
button.width = 150
button.x = (win_attribute.width - button.width) / 2
button.y = idx * 25 + title.bottom + 5
elif attr_type[:4] == 'team':
if idx < buttons_per_column:
button.width = 150
button.x = win_attribute.width / 2 - (button.width + 5)
else:
button.width = 150
button.x = win_attribute.width / 2 + 5
button.y = (idx % buttons_per_column) * 25 + title.bottom - 5
else:
print "Invalid attribute type."
radio_button_list.append(button)
# ========== Button Functions ==========
def enter_btn_func():
valid = False
return_value = None
# Player
if attr_type == 'player_search':
# Value checks
if radio_group.value in ["PAC", "SHO", "PAS", "DRI", "DEF", "PHY", "DIV", "HAN", "KIC", "REF", "SPD", "POS",
"acceleration", "aggression", "agility", "balance", "ballcontrol", "crossing",
"curve", "dribbling", "finishing", "freekickaccuracy", "gkdiving", "gkhandling",
"gkkicking", "gkpositioning", "gkreflexes", "headingaccuracy", "interceptions",
"jumping", "longpassing", "longshots", "marking", "penalties", "positioning",
"potential", "rating", "reactions", "shortpassing", "shotpower", "skillMoves",
"slidingtackle", "sprintspeed", "stamina", "standingtackle", "strength", "vision",
"volleys", "weakFoot"]:
# Value should be an integer between 0 and 100
if value_tf.value.isdigit():
return_value = int(value_tf.value)
if 0 < return_value < 100:
valid = True
elif radio_group.value in ["age", "baseId", "clubId", "id", "leagueId", "nationId", "price"]:
# Value should be an integer
if value_tf.value.isdigit():
return_value = int(value_tf.value)
valid = True
elif radio_group.value in ["height"]:
# Check if value is a decimal
decimal_value = 0.0
if '.' in value_tf.value:
try:
decimal_value = float(value_tf.value)
# If value is less than 10, assume unit is feet. Else, assume unit is centimeters.
if decimal_value < 10:
return_value = int(decimal_value * 30.48)
else:
return_value = decimal_value
valid = True
except Exception:
print "Invalid attribute value."
value_tf.value = "Invalid attribute value."
elif value_tf.value.isdigit():
# If value is less than 10, assume unit is feet. Else, assume unit is centimeters.
if int(value_tf.value) < 10:
return_value = int(int(value_tf.value) * 30.48)
else:
return_value = int(value_tf.value)
valid = True
elif radio_group.value in ["weight"]:
# Value should be an integer
if value_tf.value.isdigit():
# If value is greater than 110, assume unit is pounds. Else, assume unit is kilograms.
if int(value_tf.value) > 110:
return_value = int(int(value_tf.value) * 0.453592)
else:
return_value = int(value_tf.value)
valid = True
elif radio_group.value in ["atkWorkRate", "birthdate", "club", "color", "commonName", "defWorkRate",
"firstName", "foot", "itemType", "lastName", "league", "modelName", "name",
"name_custom", "nation", "playStyle", "playerType", "position", "positionFull",
"quality", "specialities", "traits"]:
# Value should be a string
if type(value_tf.value) is str:
return_value = value_tf.value
valid = True
elif radio_group.value in ["isGK", "isSpecialType"]:
# Value should be a string
if value_tf.value.upper() in ['T', 'F', 'TRUE', 'FALSE', 'Y', 'N', 'YES', 'NO']:
if value_tf.value.upper()[0] in ['T', 'Y']:
return_value = True
else:
return_value = False
valid = True
if valid:
attr_dict[radio_group.value] = (return_value, compare_group.value)
value_tf.value = ''
show_attributes(attribute_display_list)
erase_btn.enabled = 1
else:
print "Invalid attribute value."
value_tf.value = "Invalid attribute value."
# Formation
elif attr_type == 'formation_search':
# Value checks
if radio_group.value in ["num_attackers", "num_midfielders", "num_defenders"]:
# Value should be an integer between 0 and 10
if value_tf.value.isdigit():
return_value = int(value_tf.value)
if 0 < return_value < 10:
valid = True
elif radio_group.value in ["num_links"]:
# Value should be an integer
if value_tf.value.isdigit():
return_value = int(value_tf.value)
valid = True
elif radio_group.value in ["name", "style", "description", "position_all"]:
# Value should be a string
if type(value_tf.value) is str:
return_value = value_tf.value
valid = True
if valid:
attr_dict[radio_group.value] = (return_value, compare_group.value)
value_tf.value = ''
show_attributes(attribute_display_list)
erase_btn.enabled = 1
else:
print "Invalid attribute value."
value_tf.value = "Invalid attribute value."
# Team
elif attr_type == 'team_search':
# Value checks
if radio_group.value in ["rating", "chemistry", "total_ic"]:
# Value should be an integer between -1 and 111
if value_tf.value.isdigit():
return_value = int(value_tf.value)
if -1 < return_value < 111:
valid = True
elif radio_group.value in ["total_skillMoves"]:
# Value should be an integer between 10 and 56
if value_tf.value.isdigit():
return_value = int(value_tf.value)
if 10 < return_value < 56:
valid = True
elif radio_group.value in ["strength", "total_price", "total_PAC", "total_SHO", "total_PAS", "total_DRI",
"total_DEF", "total_PHY"]:
# Value should be an integer
if value_tf.value.isdigit():
return_value = int(value_tf.value)
valid = True
elif radio_group.value in ["player", "formation", "manager_league", "manager_nation", "style"]:
# Value should be a string
if type(value_tf.value) is str:
return_value = value_tf.value
valid = True
if valid:
attr_dict[radio_group.value] = (return_value, compare_group.value)
value_tf.value = ''
show_attributes(attribute_display_list)
erase_btn.enabled = 1
else:
print "Invalid attribute value."
value_tf.value = "Invalid attribute value."
else:
print 'Invalid attr_type for AddAttribute.'
win_attribute.become_target()
def erase_btn_func():
if attr_type[-4:] == 'sort':
del attr_list[:]
elif attr_type[-6:] == 'search':
attr_dict.clear()
show_attributes(attribute_display_list)
win_attribute.become_target()
erase_btn.enabled = 0
def back_btn_func():
if settings['window'] == 'search':
SearchMenu.open_search_menu(win_attribute.x, win_attribute.y, db_dict, attr_dict, attr_list, settings)
elif settings['window'] == 'edit':
EditMenu.open_edit_menu(win_attribute.x, win_attribute.y, db_dict, attr_dict, attr_list, settings)
elif settings['window'] == 'ultimate_player_judging':
CreateUltimateTeams.open_create_ultimate_teams_window(
win_attribute.x, win_attribute.y, db_dict, settings['prev_window_value'], player_judge_list=attr_list,
file_name=settings['file_name'], roster=settings['roster'], input_formation=settings['input_formation'])
elif settings['window'] == 'ultimate_team_judging':
CreateUltimateTeams.open_create_ultimate_teams_window(
win_attribute.x, win_attribute.y, db_dict, settings['prev_window_value'], team_judge_list=attr_list,
file_name=settings['file_name'], roster=settings['roster'], input_formation=settings['input_formation'])
elif settings['window'] == 'pick_player':
PickPlayer.open_pick_player_window(win_attribute.x, win_attribute.y, db_dict,
settings['input_formation'], settings['win_previous'],
settings['roster'], settings['pos_symbol'],
settings['pick_formations_page'], attr_dict, attr_list, settings)
else:
print "Invalid window setting."
win_attribute.become_target()
win_attribute.hide()
def attr_btn_func(attr_to_remove):
# Remove from attribute dict
if attr_type[-6:] == 'search':
attr_dict.pop(attr_to_remove, None)
show_attributes(attribute_display_list)
if len(attr_dict) < 1:
erase_btn.enabled = 0
# Remove from attribute list
elif attr_type[-4:] == 'sort':
if attr_to_remove in attr_list:
attr_list.remove(attr_to_remove)
show_attributes(attribute_display_list)
if len(attr_list) < 1:
erase_btn.enabled = 0
win_attribute.become_target()
# ========== Buttons ==========
custom_button_width = 100
erase_btn.x = (win_width - custom_button_width) / 2
erase_btn.y = win_attribute.height - 70
erase_btn.height = button_height
erase_btn.width = custom_button_width
erase_btn.font = button_font
erase_btn.action = erase_btn_func
erase_btn.style = 'default'
erase_btn.color = button_color
erase_btn.just = 'right'
if (attr_type[-4:] == 'sort' and len(attr_list) == 0) or \
(attr_type[-6:] == 'search' and len(attr_dict) == 0):
erase_btn.enabled = 0
if attr_type[-4:] == 'sort':
erase_btn.x = (win_width - 2*custom_button_width - button_spacing) / 2
view.add(erase_btn)
enter_btn.x = erase_btn.left - button_spacing - custom_button_width
enter_btn.y = erase_btn.top
enter_btn.height = button_height
enter_btn.width = custom_button_width
enter_btn.font = button_font
enter_btn.action = enter_btn_func
enter_btn.style = 'default'
enter_btn.color = button_color
enter_btn.just = 'right'
enter_btn.enabled = 0
if attr_type[-6:] == 'search':
view.add(enter_btn)
back_btn.x = erase_btn.right + button_spacing
back_btn.y = erase_btn.top
back_btn.height = button_height
back_btn.width = custom_button_width
back_btn.font = button_font
back_btn.action = back_btn_func
back_btn.style = 'default'
back_btn.color = button_color
back_btn.just = 'right'
view.add(back_btn)
# ========== Value Textfield ==========
value_tf = TextField()
value_tf.width = 200
value_tf.x = (win_width - value_tf.width) / 2
value_tf.y = erase_btn.top - 35
value_tf.height = 25
value_tf.font = std_tf_font
# ========== Compare Radio Buttons ==========
compare_group = RadioGroup(action=comp_selection_made)
comp_button_list = []
comp_button_width = 55
comp_button_x = (win_attribute.width - 4*comp_button_width)/2
for value in ['higher', 'exact', 'lower', 'not']:
button = RadioButton()
button.width = comp_button_width
button.x = comp_button_x
button.y = value_tf.top - value_tf.height - title_border
button.group = compare_group
button.value = value
button.title = value.capitalize()
comp_button_x += comp_button_width + 5
comp_button_list.append(button)
compare_group.value = 'higher'
def show_attributes(display_list):
# Remove old attribute buttons from screen
for message in display_list:
view.remove(message)
del display_list[:]
# Display new search attributes on screen
message_x = 10
message_y = win_attribute.height - 150
if attr_type[-6:] == 'search':
display_list.append(Label(text="Search Attributes:", font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color))
message_y += std_tf_height
index = 0
for key, search_value in attr_dict.iteritems():
if index == 6:
message_x = win_attribute.width - std_tf_width - 10
message_y = win_attribute.height - 150 + std_tf_height
msg_text = format_attr_name(key) + ": "
if key in ['id', 'baseId', 'nationId', 'leagueId', 'clubId']:
msg_text += str(search_value[0])
elif type(search_value[0]) is int:
msg_text += search_value[1].capitalize() + ' ' + str(search_value[0])
elif search_value[1] == 'not':
msg_text += search_value[1].capitalize() + ' "' + str(search_value[0]) + '"'
else:
msg_text += '"' + str(search_value[0]) + '"'
attr_btn = Button(title=msg_text, font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color,
action=(attr_btn_func, key))
message_y += std_tf_height + 1
display_list.append(attr_btn)
index += 1
# Display new sort attributes on screen
elif attr_type[-4:] == 'sort':
display_list.append(Label(text="Sort Attributes:", font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color))
message_y += std_tf_height
for index, sort_value in enumerate(attr_list):
if index == 6:
message_x = win_attribute.width - std_tf_width - 10
message_y = win_attribute.height - 150 + std_tf_height
attr_btn = Button(title=(format_attr_name(sort_value)), font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color,
action=(attr_btn_func, sort_value))
message_y += std_tf_height + 1
display_list.append(attr_btn)
for attribute_label in display_list:
view.add(attribute_label)
show_attributes(attribute_display_list)
# ========== Add buttons to window ==========
# Shows only for getting attribute for search, not sort
if attr_type[-6:] == 'search':
view.add(value_tf)
for button in comp_button_list:
view.add(button)
for button in radio_button_list:
view.add(button)
win_attribute.add(view)
view.become_target()
win_attribute.show()
def open_pick_player_window(window_x, window_y, db_dict, input_formation, win_previous, roster, pos_symbol,
pick_formations_page, attr_dict=None, attr_list=None, settings=None):
num_results = 20
general_display = []
if attr_dict is None:
attr_dict = {}
if attr_list is None:
attr_list = []
if settings is None:
settings = {
'window': 'pick_player',
'p_db_rg': 'player_list',
'pos_search_rg': 'green',
'order_rg': True,
'messages': {
'search': [],
'sort': [],
'results': []
},
'input_formation': input_formation,
'roster': roster,
'pos_symbol': pos_symbol,
'win_previous': win_previous,
'pick_formations_page': pick_formations_page}
# Assign exact position as default position to search for.
attr_dict['position'] = (input_formation['positions'][pos_symbol]['symbol'], 'exact')
# Not position dependent - search all.
if input_formation['name'] == 'Generic':
settings['pos_search_rg'] = 'red'
attr_dict.pop('position', None)
# ========== Window ==========
win_pick_player = Window()
win_pick_player.title = pick_player_win_title + ' - ' + pos_symbol
win_pick_player.auto_position = False
win_pick_player.position = (window_x, window_y)
win_pick_player.size = (win_width, win_height)
win_pick_player.resizable = 0
win_pick_player.name = pick_player_title + " Window"
# ========== Window Image View ==========
class PickPlayerWindowImageView(View):
def draw(self, c, r):
c.backcolor = view_backcolor
c.erase_rect(r)
view = PickPlayerWindowImageView(size=win_pick_player.size)
# ========== Title ==========
title = Label(text=pick_player_title + ' - ' + pos_symbol)
title.font = title_font
title.width = title_width
title.height = title_height
title.x = (win_width - title_width) / 2
title.y = top_border
title.color = title_color
title.just = 'center'
general_display.append(title)
# ========== Action Button Declarations ==========
start_btn = Button("Start")
back_btn = Button("Back")
# ========== Tool Button Declarations ==========
attribute_btn = Button("Add Search Attribute")
sort_btn = Button("Add Sort Attribute")
reset_btn = Button("Reset Results")
# ========== Action Button Functions ==========
def start_btn_func():
# Search - if no attribute selected, return all
if len(attr_dict) == 0:
search_results = db_dict[p_db_radio_group.value][1]
else:
search_results = db_dict[p_db_radio_group.value][1].search(attr_dict)
search_results = PlayerDB.PlayerDB(search_results)
# Sort - if no attribute selected, use rating
if len(attr_list) == 0:
search_results.sort(['rating'], sort_order_radio_group.value)
else:
search_results.sort(attr_list, sort_order_radio_group.value)
# Get attributes list and avoid duplicates
attributes_list = []
for attr in attr_list:
if attributes_list.count(attr) == 0:
attributes_list.append(attr)
for attr_key in attr_dict.iterkeys():
if attributes_list.count(attr_key) == 0:
attributes_list.append(attr_key)
display_players(search_results, attributes_list, (0, num_results))
win_pick_player.become_target()
def back_btn_func():
win_pick_player.hide()
win_previous.show()
# ========== Tool Button Functions ==========
def attribute_btn_func():
# Delete results
del settings['messages']['results'][:]
# Set type argument
attr_type = 'player_search'
# Open new window and close current window
win_pick_player.hide()
AddAttribute.open_attribute_window(win_pick_player.x, win_pick_player.y, db_dict,
attr_dict, attr_list, attr_type, settings)
def sort_btn_func():
# Delete results
del settings['messages']['results'][:]
# Set type argument
attr_type = 'player_sort'
# Open new window and close current window
win_pick_player.hide()
AddAttribute.open_attribute_window(win_pick_player.x, win_pick_player.y, db_dict,
attr_dict, attr_list, attr_type, settings)
def reset_btn_func():
# Remove messages off page
for message in settings['messages']['search']:
view.remove(message)
for message in settings['messages']['sort']:
view.remove(message)
for message in settings['messages']['results']:
view.remove(message)
# Delete the attribute parameters for search and sort
attr_dict.clear()
del attr_list[:]
del settings['messages']['results'][:]
win_pick_player.become_target()
def create_attribute_messages():
# ========== Messages ==========
lowest_msg_l = start_btn.top
lowest_msg_r = start_btn.top
attr_msg_offset = 25
# Attribute Messages
del settings['messages']['search'][:]
del settings['messages']['sort'][:]
if len(attr_dict) > 0:
settings['messages']['search'].append(
Label(text="Search Attributes:", font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=attr_msg_offset, y=lowest_msg_l,
color=title_color))
lowest_msg_l += std_tf_height
for key, value in attr_dict.iteritems():
msg_text = format_attr_name(key) + ": "
if key in ['id', 'baseId', 'nationId', 'leagueId', 'clubId']:
msg_text += str(value[0])
elif type(value[0]) is int:
msg_text += value[1].capitalize() + ' ' + str(value[0])
elif value[1] == 'not':
msg_text += value[1].capitalize() + ' "' + str(value[0]) + '"'
else:
msg_text += '"' + str(value[0]) + '"'
attr_label = Label(text=msg_text, font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=attr_msg_offset, y=lowest_msg_l, color=title_color)
lowest_msg_l += std_tf_height
settings['messages']['search'].append(attr_label)
if len(attr_list) > 0:
settings['messages']['sort'].append(Label(text="Sort Attributes:", font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=reset_btn.right + 3 * attr_msg_offset,
y=lowest_msg_r, color=title_color))
lowest_msg_r += std_tf_height
for value in attr_list:
attr_label = Label(text=(format_attr_name(value)), font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=reset_btn.right + 3 * attr_msg_offset, y=lowest_msg_r,
color=title_color)
lowest_msg_r += std_tf_height
settings['messages']['sort'].append(attr_label)
create_attribute_messages()
def player_bio_btn_func(player):
win_pick_player.hide()
PlayerBio.open_player_bio_window(win_pick_player.x, win_pick_player.y, player, win_pick_player, db_dict,
db_dict['player_list'][0], db_dict['player_list'][1],
roster, pos_symbol, input_formation, pick_formations_page)
win_pick_player.become_target()
# ========== Action Buttons ==========
start_btn.x = (win_width - 2*small_button_width - small_button_spacing) / 2
start_btn.y = title.bottom + small_button_top_spacing
start_btn.height = small_button_height
start_btn.width = small_button_width
start_btn.font = small_button_font
start_btn.action = start_btn_func
start_btn.style = 'default'
start_btn.color = small_button_color
start_btn.just = 'right'
general_display.append(start_btn)
back_btn.x = start_btn.right + small_button_spacing
back_btn.y = start_btn.top
back_btn.height = small_button_height
back_btn.width = small_button_width
back_btn.font = small_button_font
back_btn.action = back_btn_func
back_btn.style = 'default'
back_btn.color = small_button_color
back_btn.just = 'right'
general_display.append(back_btn)
# ========== Tool Buttons ==========
attribute_btn.x = (win_width - 3*small_button_width - 2*small_button_spacing) / 2
attribute_btn.y = start_btn.bottom + 10
attribute_btn.height = small_button_height
attribute_btn.width = small_button_width
attribute_btn.font = small_button_font
attribute_btn.action = attribute_btn_func
attribute_btn.style = 'default'
attribute_btn.color = small_button_color
attribute_btn.just = 'right'
general_display.append(attribute_btn)
sort_btn.x = attribute_btn.right + small_button_spacing
sort_btn.y = attribute_btn.top
sort_btn.height = small_button_height
sort_btn.width = small_button_width
sort_btn.font = small_button_font
sort_btn.action = sort_btn_func
sort_btn.style = 'default'
sort_btn.color = small_button_color
sort_btn.just = 'right'
general_display.append(sort_btn)
reset_btn.x = sort_btn.right + small_button_spacing
reset_btn.y = attribute_btn.top
reset_btn.height = small_button_height
reset_btn.width = small_button_width
reset_btn.font = small_button_font
reset_btn.action = reset_btn_func
reset_btn.style = 'default'
reset_btn.color = small_button_color
reset_btn.just = 'right'
general_display.append(reset_btn)
# ========== Pick DB/List Button Functions ==========
def pick_file():
# Remove old messages off page
for message in settings['messages']['results']:
view.remove(message)
del settings['messages']['results'][:]
settings['file_changes'] = False
settings['prev_window'] = 'pick_player'
settings['input_formation'] = input_formation
settings['win_previous'] = win_previous
settings['roster'] = roster
settings['pos_symbol'] = pos_symbol
settings['pick_formations_page'] = pick_formations_page
settings['attr_dict'] = attr_dict
settings['attr_list'] = attr_list
PickFile.open_pick_file_window(win_pick_player.x, win_pick_player.y, db_dict, settings)
win_pick_player.hide()
def pick_p_db_func():
settings['file_type'] = 'current_player_db'
pick_file()
def pick_p_list_func():
settings['file_type'] = 'current_player_list'
pick_file()
# ========== DB Radio Buttons ==========
def get_attribute_p_db_rg():
settings['p_db_rg'] = p_db_radio_group.value
win_pick_player.become_target()
def get_attribute_pos_search_rg():
if settings['pos_search_rg'] != pos_search_radio_group.value:
button_click = True
else:
button_click = False
settings['pos_search_rg'] = pos_search_radio_group.value
# If new radio button was clicked, clear position in attribute dictionary.
if button_click:
attr_dict.pop('position', None)
# Add position search terms if position and positionFull and not specified and not searching all players.
if pos_search_radio_group.value != 'red' and button_click:
temp_team = Team.Team()
# Get positions of corresponding relevance
search_positions_list = temp_team.related_positions(
input_formation['positions'][pos_symbol]['symbol'], pos_search_radio_group.value)
# Concatenate strings
search_positions = ""
for position in search_positions_list:
search_positions += position + ', '
search_positions = search_positions[:-2]
if len(search_positions) < 1:
search_positions = 'none'
attr_dict['position'] = (search_positions, 'exact')
# Erase current search attributes
for message in settings['messages']['search']:
view.remove(message)
# Create new search attributes
create_attribute_messages()
# Display search attributes
for message in settings['messages']['search']:
view.add(message)
win_pick_player.become_target()
p_db_radio_group = RadioGroup(action=get_attribute_p_db_rg)
pos_search_radio_group = RadioGroup(action=get_attribute_pos_search_rg)
# File Label
db_msg_width = 35
db_radio_btn_width = 150
db_file_btn_width = 40
db_radio_btn_space = 5
db_file_btn_space = 2
file_label = Label(text="File", font=std_tf_font, width=db_msg_width,
height=std_tf_height, color=title_color)
file_label.x = (win_pick_player.width - 2*db_radio_btn_width - 2*db_file_btn_width - db_radio_btn_space
- 2*db_file_btn_space - db_msg_width) / 2
file_label.y = reset_btn.bottom + db_radio_btn_space
# Players DB RG
player_db_radio_btn = RadioButton(db_dict['player_db'][0])
player_db_radio_btn.width = db_radio_btn_width
player_db_radio_btn.x = file_label.right
player_db_radio_btn.y = file_label.top
player_db_radio_btn.group = p_db_radio_group
player_db_radio_btn.value = 'player_db'
player_db_file_btn = Button("Pick")
player_db_file_btn.x = player_db_radio_btn.right + db_file_btn_space
player_db_file_btn.y = file_label.top
player_db_file_btn.height = std_tf_height
player_db_file_btn.width = db_file_btn_width
player_db_file_btn.font = small_button_font
player_db_file_btn.action = pick_p_db_func
player_db_file_btn.style = 'default'
player_db_file_btn.color = small_button_color
player_db_file_btn.just = 'right'
player_list_radio_btn = RadioButton(db_dict['player_list'][0])
player_list_radio_btn.width = db_radio_btn_width
player_list_radio_btn.x = player_db_file_btn.right + db_radio_btn_space
player_list_radio_btn.y = file_label.top
player_list_radio_btn.group = p_db_radio_group
player_list_radio_btn.value = 'player_list'
player_list_file_btn = Button("Pick")
player_list_file_btn.x = player_list_radio_btn.right + db_file_btn_space
player_list_file_btn.y = file_label.top
player_list_file_btn.height = std_tf_height
player_list_file_btn.width = db_file_btn_width
player_list_file_btn.font = small_button_font
player_list_file_btn.action = pick_p_list_func
player_list_file_btn.style = 'default'
player_list_file_btn.color = small_button_color
player_list_file_btn.just = 'right'
p_db_radio_group.value = settings['p_db_rg']
view.add(file_label)
view.add(player_db_radio_btn)
view.add(player_db_file_btn)
view.add(player_list_radio_btn)
view.add(player_list_file_btn)
# Position Search RG
pos_search_radio_btn_width = 60
pos_search_msg_width = 140
pos_search_rg_msg = Label(text="Positions to Search:", font=std_tf_font, width=pos_search_msg_width,
height=std_tf_height, color=title_color)
pos_search_rg_msg.x = (win_pick_player.width-4*pos_search_radio_btn_width-db_radio_btn_space-pos_search_msg_width)/2
pos_search_rg_msg.y = file_label.bottom + db_radio_btn_space
pos_search_green_radio_btn = RadioButton("Exact")
pos_search_green_radio_btn.width = pos_search_radio_btn_width
pos_search_green_radio_btn.x = pos_search_rg_msg.right
pos_search_green_radio_btn.y = pos_search_rg_msg.top
pos_search_green_radio_btn.group = pos_search_radio_group
pos_search_green_radio_btn.value = 'green'
pos_search_yellow_radio_btn = RadioButton("Similar")
pos_search_yellow_radio_btn.width = pos_search_radio_btn_width
pos_search_yellow_radio_btn.x = pos_search_green_radio_btn.right + db_radio_btn_space
pos_search_yellow_radio_btn.y = pos_search_green_radio_btn.top
pos_search_yellow_radio_btn.group = pos_search_radio_group
pos_search_yellow_radio_btn.value = 'yellow'
pos_search_orange_radio_btn = RadioButton("Related")
pos_search_orange_radio_btn.width = pos_search_radio_btn_width
pos_search_orange_radio_btn.x = pos_search_yellow_radio_btn.right + db_radio_btn_space
pos_search_orange_radio_btn.y = pos_search_yellow_radio_btn.top
pos_search_orange_radio_btn.group = pos_search_radio_group
pos_search_orange_radio_btn.value = 'orange'
pos_search_red_radio_btn = RadioButton("All")
pos_search_red_radio_btn.width = pos_search_radio_btn_width
pos_search_red_radio_btn.x = pos_search_orange_radio_btn.right + db_radio_btn_space
pos_search_red_radio_btn.y = pos_search_orange_radio_btn.top
pos_search_red_radio_btn.group = pos_search_radio_group
pos_search_red_radio_btn.value = 'red'
pos_search_radio_group.value = settings['pos_search_rg']
view.add(pos_search_rg_msg)
view.add(pos_search_green_radio_btn)
view.add(pos_search_yellow_radio_btn)
view.add(pos_search_orange_radio_btn)
view.add(pos_search_red_radio_btn)
if input_formation['name'] == 'Generic':
pos_search_green_radio_btn.enabled = 0
pos_search_yellow_radio_btn.enabled = 0
pos_search_orange_radio_btn.enabled = 0
# ========== Sort Order Radio Buttons ==========
def get_attribute_sort_order_rg():
settings['order_rg'] = sort_order_radio_group.value
win_pick_player.become_target()
sort_order_radio_group = RadioGroup(action=get_attribute_sort_order_rg)
asc_desc_radio_btn_width = 75
asc_msg_width = 80
radio_btn_space = 5
asc_desc_rg_msg = Label(text="Sort Order:", font=std_tf_font, width=asc_msg_width, height=std_tf_height,
color=title_color)
asc_desc_rg_msg.x = (win_pick_player.width - 2*asc_desc_radio_btn_width - radio_btn_space - asc_msg_width) / 2
asc_desc_rg_msg.y = pos_search_green_radio_btn.bottom + radio_btn_space
general_display.append(asc_desc_rg_msg)
descend_radio_btn = RadioButton("Descending")
descend_radio_btn.width = asc_desc_radio_btn_width
descend_radio_btn.x = asc_desc_rg_msg.right
descend_radio_btn.y = asc_desc_rg_msg.top
descend_radio_btn.group = sort_order_radio_group
descend_radio_btn.value = True
general_display.append(descend_radio_btn)
ascend_radio_btn = RadioButton("Ascending")
ascend_radio_btn.width = asc_desc_radio_btn_width
ascend_radio_btn.x = descend_radio_btn.right + radio_btn_space
ascend_radio_btn.y = descend_radio_btn.top
ascend_radio_btn.group = sort_order_radio_group
ascend_radio_btn.value = False
general_display.append(ascend_radio_btn)
sort_order_radio_group.value = settings['order_rg']
# ========== Previous, Add to List, Next Buttons ==========
previous_btn = Button("<<< Previous %d" % num_results)
add_to_list_btn = Button("")
next_btn = Button("Next %d >>>" % num_results)
total_num_results_label = Label()
pages_label = Label()
def previous_btn_func(results_list, attributes, index_range):
# display previous results
display_players(results_list, attributes, index_range)
win_pick_player.become_target()
def next_btn_func(results_list, attributes, index_range):
# display next results
display_players(results_list, attributes, index_range)
win_pick_player.become_target()
add_to_list_btn.x = attribute_btn.right + small_button_spacing
add_to_list_btn.y = descend_radio_btn.bottom + 5
add_to_list_btn.height = tiny_button_height
add_to_list_btn.width = small_button_width
add_to_list_btn.font = small_button_font
add_to_list_btn.style = 'default'
add_to_list_btn.color = small_button_color
add_to_list_btn.just = 'right'
add_to_list_btn.action = win_pick_player.become_target()
previous_btn.height = tiny_button_height
previous_btn.width = small_button_width
previous_btn.x = add_to_list_btn.left - previous_btn.width - small_button_spacing
previous_btn.y = add_to_list_btn.top
previous_btn.font = small_button_font
previous_btn.style = 'default'
previous_btn.color = small_button_color
previous_btn.just = 'right'
next_btn.height = tiny_button_height
next_btn.width = small_button_width
next_btn.x = add_to_list_btn.right + small_button_spacing
next_btn.y = add_to_list_btn.top
next_btn.font = small_button_font
next_btn.style = 'default'
next_btn.color = small_button_color
next_btn.just = 'right'
total_num_results_label.font = std_tf_font
total_num_results_label.width = 100
total_num_results_label.height = std_tf_height
total_num_results_label.x = previous_btn.left + - 100 - 10
total_num_results_label.y = add_to_list_btn.top
total_num_results_label.color = title_color
total_num_results_label.just = 'right'
pages_label.font = std_tf_font
pages_label.width = 125
pages_label.height = std_tf_height
pages_label.x = next_btn.right + 10
pages_label.y = add_to_list_btn.top
pages_label.color = title_color
pages_label.just = 'left'
# ========== Display players from search ==========
def display_players(results_list, attributes, index_range):
# Remove old messages off page
for message in settings['messages']['results']:
view.remove(message)
del settings['messages']['results'][:]
previous_range = (index_range[0]-num_results, index_range[0])
previous_btn.action = (previous_btn_func, results_list, attributes, previous_range)
next_range = (index_range[1], index_range[1]+num_results)
next_btn.action = (next_btn_func, results_list, attributes, next_range)
total_num_results_label.text = str(len(results_list.db)) + " Players"
pages_label.text = "Page %d of %d" % (int(index_range[1]/num_results),
math.ceil(len(results_list.db)/float(num_results)))
if index_range[0] > 0:
previous_btn.enabled = 1
else:
previous_btn.enabled = 0
if index_range[1] <= len(results_list.db) - 1:
next_btn.enabled = 1
else:
next_btn.enabled = 0
settings['messages']['results'].append(add_to_list_btn)
settings['messages']['results'].append(previous_btn)
settings['messages']['results'].append(next_btn)
settings['messages']['results'].append(total_num_results_label)
settings['messages']['results'].append(pages_label)
# Print out labels
labels = player_info_labels(attributes)
stat_index = 0
# Spacing values for each of the stats
spacing_list = [125, 40, 40, 65, 115, 115, 115, 40]
# Calculate the maximum number of stat fields that will fit on screen
max_player_fields = len(spacing_list[:-1]) + (win_pick_player.width - sum(spacing_list[:-1]))/spacing_list[-1]
# Calculate the left border of the stats based on the number and width of the stats
left_border = (win_width - sum(spacing_list[:-1]) -
(len(labels[:max_player_fields]) - len(spacing_list) + 1) * spacing_list[-1])/2
msg_x = left_border
msg_y = add_to_list_btn.bottom + 5
for info_label in labels[:max_player_fields]:
player_label = Label(text=info_label, font=std_tf_font_bold, width=spacing_list[stat_index]-5,
height=std_tf_height, x=msg_x, y=msg_y, color=title_color)
settings['messages']['results'].append(player_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list)-1:
stat_index += 1
msg_y += std_tf_height + 5
# Print out players
for idx, player in enumerate(results_list.db[index_range[0]:index_range[1]]):
msg_x = left_border
player_stats = player_info(player, attributes)
stat_index = 0
# Check for names that are too long
name = player_stats[0]
if len(name) > 20:
name = player['lastName']
bio_btn = Button(title=name, width=spacing_list[stat_index]-5, height=15, x=msg_x, y=msg_y,
action=(player_bio_btn_func, player))
settings['messages']['results'].append(bio_btn)
msg_x += spacing_list[stat_index]
stat_index += 1
for player_stat in player_stats[1:max_player_fields]:
player_label = Label(text=player_stat, font=small_button_font, width=spacing_list[stat_index]-5,
height=std_tf_height, x=msg_x, y=msg_y, color=title_color)
settings['messages']['results'].append(player_label)
msg_x += spacing_list[stat_index]
if stat_index < len(spacing_list) - 1:
stat_index += 1
msg_y += std_tf_height
for results_msg in settings['messages']['results']:
view.add(results_msg)
# ========== Add components to view and add view to window ==========
for msg in general_display:
view.add(msg)
for msg in settings['messages']['search']:
view.add(msg)
for msg in settings['messages']['sort']:
view.add(msg)
for msg in settings['messages']['results']:
view.add(msg)
win_pick_player.add(view)
view.become_target()
win_pick_player.show()
def open_attribute_window(window_x, window_y, db_dict, attr_dict, attr_list, attr_type, settings):
# ========== Window ==========
win_attribute = Window()
win_attribute.title = attribute_win_title
win_attribute.auto_position = False
win_attribute.position = (window_x, window_y)
win_attribute.size = (win_width, 500)
win_attribute.resizable = 0
win_attribute.name = attribute_title + " Window"
win_attribute.show()
# ========== Window Image View ==========
class StartWindowImageView(View):
def draw(self, c, r):
c.backcolor = view_backcolor
c.erase_rect(r)
view = StartWindowImageView(size=win_attribute.size)
attribute_display_list = []
# ========== Title ==========
title = Label(text=attribute_title)
title.font = title_font
title.width = title_width
title.height = title_height
title.x = (win_width - title_width) / 2
title.y = top_border
title.color = title_color
title.just = 'center'
# ========== Button Declarations ==========
enter_btn = Button("Enter")
erase_btn = Button("Erase List")
back_btn = Button("Back")
# ========== Radio Button Action ==========
def selection_made():
enter_btn.enabled = 1
win_attribute.become_target()
def comp_selection_made():
win_attribute.become_target()
# ========== Attribute Radio Buttons ==========
radio_group = RadioGroup(action=selection_made)
radio_button_list = []
with open(config_filename, 'r') as f:
attributes_list = json.load(f)['player_attributes']['all']
f.close()
for idx, attribute in enumerate(attributes_list):
button = RadioButton(attribute)
if idx < 10:
button.width = 45
button.x = (idx / 10) * (button.width + 5) + 5
elif idx < 30:
button.width = 100
button.x = (idx / 10) * (button.width + 5) - 50
elif idx < 40:
button.width = 110
button.x = (idx / 10) * (button.width + 5) - 80
else:
button.width = 100
button.x = (idx / 10) * (button.width + 5) - 40
button.y = (idx % 10) * 25 + title.bottom + 5
button.group = radio_group
button.value = attribute
button.title = format_attr_name(attribute)
radio_button_list.append(button)
# ========== Button Functions ==========
def enter_btn_func():
valid = False
return_value = None
if attr_type == 'sort':
# Check for doubles
if attr_list.count(radio_group.value) == 0:
attr_list.append(radio_group.value)
show_attributes(attribute_display_list)
erase_btn.enabled = 1
elif attr_type == 'search':
# Value checks
if radio_group.value in ["PAC", "SHO", "PAS", "DRI", "DEF", "PHY", "DIV", "HAN", "KIC", "REF", "SPD", "POS",
"acceleration", "aggression", "agility", "balance", "ballcontrol", "crossing",
"curve", "dribbling", "finishing", "freekickaccuracy", "gkdiving", "gkhandling",
"gkkicking", "gkpositioning", "gkreflexes", "headingaccuracy", "interceptions",
"jumping", "longpassing", "longshots", "marking", "penalties", "positioning",
"potential", "rating", "reactions", "shortpassing", "shotpower", "skillMoves",
"slidingtackle", "sprintspeed", "stamina", "standingtackle", "strength", "vision",
"volleys", "weakFoot"]:
# Value should be an integer between 0 and 100
if value_tf.value.isdigit():
return_value = int(value_tf.value)
if 0 < return_value < 100:
valid = True
elif radio_group.value in ["age", "baseId", "clubId", "height", "id", "leagueId", "nationId", "weight"]:
# Value should be an integer
if value_tf.value.isdigit():
return_value = int(value_tf.value)
valid = True
elif radio_group.value in ["atkWorkRate", "birthdate", "club", "color", "commonName", "defWorkRate",
"firstName", "foot", "itemType", "lastName", "league", "modelName", "name",
"name_custom", "nation", "playStyle", "playerType", "position", "positionFull",
"quality"]:
# Value should be a string
if type(value_tf.value) is str:
return_value = value_tf.value
valid = True
elif radio_group.value in ["isGK", "isSpecialType"]:
# Value should be a string
if value_tf.value.upper() in ['T', 'F', 'TRUE', 'FALSE', 'Y', 'N', 'YES', 'NO']:
if value_tf.value.upper()[0] in ['T', 'Y']:
return_value = True
else:
return_value = False
valid = True
if valid:
attr_dict[radio_group.value] = (return_value, compare_group.value)
value_tf.value = ''
show_attributes(attribute_display_list)
erase_btn.enabled = 1
else:
print "Invalid attribute value."
else:
print 'Invalid attr_type for AddAttribute.'
def erase_btn_func():
if attr_type == 'sort':
del attr_list[:]
elif attr_type == 'search':
attr_dict.clear()
show_attributes(attribute_display_list)
erase_btn.enabled = 0
def back_btn_func():
SearchMenu.open_search_menu(win_attribute.x, win_attribute.y,
db_dict, attr_dict, attr_list, settings)
win_attribute.hide()
# ========== Buttons ==========
erase_btn.x = (win_width - button_width) / 2
erase_btn.y = win_attribute.height - 70
erase_btn.height = button_height
erase_btn.width = button_width
erase_btn.font = button_font
erase_btn.action = erase_btn_func
erase_btn.style = 'default'
erase_btn.color = button_color
erase_btn.just = 'right'
if (attr_type == 'sort' and len(attr_list) == 0) or \
(attr_type == 'search' and len(attr_dict) == 0):
erase_btn.enabled = 0
enter_btn.x = erase_btn.left - button_spacing - button_width
enter_btn.y = erase_btn.top
enter_btn.height = button_height
enter_btn.width = button_width
enter_btn.font = button_font
enter_btn.action = enter_btn_func
enter_btn.style = 'default'
enter_btn.color = button_color
enter_btn.just = 'right'
enter_btn.enabled = 0
back_btn.x = erase_btn.right + button_spacing
back_btn.y = erase_btn.top
back_btn.height = button_height
back_btn.width = button_width
back_btn.font = button_font
back_btn.action = back_btn_func
back_btn.style = 'default'
back_btn.color = button_color
back_btn.just = 'right'
# ========== Value Textfield ==========
value_tf = TextField()
value_tf.width = 200
value_tf.x = (win_width - value_tf.width) / 2
value_tf.y = enter_btn.top - 35
value_tf.height = 25
value_tf.font = std_tf_font
# ========== Compare Radio Buttons ==========
compare_group = RadioGroup(action=comp_selection_made)
comp_button_list = []
comp_button_width = 55
comp_button_x = (win_attribute.width - 4*comp_button_width)/2
for value in ['higher', 'exact', 'lower', 'not']:
button = RadioButton(attribute)
button.width = comp_button_width
button.x = comp_button_x
button.y = value_tf.top - value_tf.height - title_border
button.group = compare_group
button.value = value
button.title = value.capitalize()
comp_button_x += comp_button_width + 5
comp_button_list.append(button)
compare_group.value = 'higher'
def show_attributes(display_list):
# Remove old attribute labels from screen
for message in display_list:
view.remove(message)
del display_list[:]
# Display new search attributes on screen
message_x = 10
message_y = win_attribute.height - 150
if attr_type == 'search':
display_list.append(Label(text=("Search Attributes:"), font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color))
message_y += std_tf_height
index = 0
for key, search_value in attr_dict.iteritems():
if index == 6:
message_x = back_btn.right + 10
message_y = win_attribute.height - 150
msg_text = format_attr_name(key) + ": "
if key in ['id', 'baseId', 'nationId', 'leagueId', 'clubId']:
msg_text += str(search_value[0])
elif type(search_value[0]) is int:
msg_text += search_value[1].capitalize() + ' ' + str(search_value[0])
elif search_value[1] == 'not':
msg_text += search_value[1].capitalize() + ' "' + str(search_value[0]) + '"'
else:
msg_text += '"' + str(search_value[0]) + '"'
attr_label = Label(text=msg_text, font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color)
message_y += std_tf_height
display_list.append(attr_label)
index += 1
# Display new sort attributes on screen
elif attr_type == 'sort':
display_list.append(Label(text=("Sort Attributes:"), font=title_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color))
message_y += std_tf_height
for index, sort_value in enumerate(attr_list):
if index == 6:
message_x = back_btn.right + 10
message_y = win_attribute.height - 150
attr_label = Label(text=(format_attr_name(sort_value)), font=std_tf_font, width=std_tf_width,
height=std_tf_height, x=message_x, y=message_y, color=title_color)
message_y += std_tf_height
display_list.append(attr_label)
for attribute_label in display_list:
view.add(attribute_label)
show_attributes(attribute_display_list)
# ========== Add buttons to window ==========
view.add(title)
view.add(enter_btn)
view.add(erase_btn)
view.add(back_btn)
# Shows only for getting attribute for search, not sort
if attr_type == 'search':
view.add(value_tf)
for button in comp_button_list:
view.add(button)
for button in radio_button_list:
view.add(button)
win_attribute.add(view)
view.become_target()
win_attribute.show()