def __init__(self, parent):
# Main window initialization.
Frame.__init__(self, parent)
self.parent = parent
self.character_library = CharacterLibrary()
self.fitting_library = FittingLibrary()
self.weapon_library = WeaponLibrary()
self.module_library = ModuleLibrary()
self.current_char = Character('No Skills')
self.current_fit = self.fitting_library.get_fitting(self.fitting_library.get_fitting_list()[0])
# Call pertinent methods to display main window.
self.menubar_main()
self.combobox_character()
self.combobox_fitting()
self.tree_modules()
self.fitting_display()
self.stats_display()
class DftUi(Frame):
""" This is the Main Window for the Dust Fitting Tool. """
def __init__(self, parent):
# Main window initialization.
Frame.__init__(self, parent)
self.parent = parent
self.character_library = CharacterLibrary()
self.fitting_library = FittingLibrary()
self.weapon_library = WeaponLibrary()
self.module_library = ModuleLibrary()
self.current_char = Character('No Skills')
self.current_fit = self.fitting_library.get_fitting(self.fitting_library.get_fitting_list()[0])
# Call pertinent methods to display main window.
self.menubar_main()
self.combobox_character()
self.combobox_fitting()
self.tree_modules()
self.fitting_display()
self.stats_display()
def menubar_main(self):
""" Displays and manages the upper menubar. """
# Initial configurations.
menubar = Menu(self.parent)
self.parent.config(menu=menubar)
self.parent.option_add('*tearOff', False)
# First menu contents
fileMenu = Menu(menubar)
fileMenu.add_command(label='New Dropsuit', command=self.new_dropsuit_window)
fileMenu.add_command(label='New Vehicle')
fileMenu.add_command(label='New Character', command=self.add_character_window)
editMenu = Menu(menubar)
editMenu.add_command(label='Edit Character', command=self.edit_character_window)
editMenu.add_command(label='Delete Character', command=self.delete_character_window)
editMenu.add_command(label='Delete Fitting', command=self.delete_fitting_window)
# Add the menus
menubar.add_cascade(label='File', menu=fileMenu)
menubar.add_cascade(label='Edit', menu=editMenu)
def combobox_character(self):
""" Displays and manages the character selection for the main window. """
# Get known characters. API calls or data retrieval here.
character_names = self.character_library.get_character_list()
# Creates the Combobox which has all known characters and automatically
# selects the first character. Also other widgets.
frm_character = Frame(self)
lbl_character = Label(frm_character, text='Character:')
self.cbx_character = ttk.Combobox(frm_character, values=character_names, width=14)
self.cbx_character.set('No Skills')
# Grid management.
frm_character.grid(column=0, row=0)
lbl_character.grid(column=0, row=0, sticky=NW, padx=3, pady=3)
self.cbx_character.grid(column=1, row=0, sticky=NW, padx=3, pady=3)
# Binding.
self.cbx_character.bind('<<ComboboxSelected>>', self.change_character)
def combobox_fitting(self):
""" Displays and manages the current fitting to be displayed on the
main window. """
# Get known fits. API calls or data retrieval here.
fitting_names = self.fitting_library.get_fitting_list()
# Creates nessessary widgets.
frm_fitting = Frame(self)
lbl_fitting = Label(frm_fitting, text='Fitting:')
self.cbx_fitting = ttk.Combobox(frm_fitting, values=fitting_names)
self.cbx_fitting.set(self.current_fit.name)
# Grid management.
frm_fitting.grid(column=1, row=0)
lbl_fitting.grid(column=2, row=0, sticky=NE, padx=3, pady=3)
self.cbx_fitting.grid(column=3, row=0, sticky=NW, padx=3, pady=3)
# Binding.
self.cbx_fitting.bind('<<ComboboxSelected>>', self.change_fitting)
def tree_modules(self):
""" """
frm_modules = Frame(self)
self.tre_modules = ttk.Treeview(frm_modules, height=14, columns=('cpu', 'pg'))
scb_modules = Scrollbar(frm_modules, orient=VERTICAL, command=self.tre_modules.yview)
self.tre_modules.column('#0', width=150, minwidth=150)
self.tre_modules.column('cpu', width=30, minwidth=30)
self.tre_modules.column('pg', width=30, minwidth=30)
self.tre_modules.heading('cpu', text='CPU')
self.tre_modules.heading('pg', text='PG')
for parent in self.weapon_library.get_parents():
self.tre_modules.insert('', 'end', parent, text=parent, tag='ttk')
for child in self.weapon_library.get_children(parent):
self.tre_modules.insert(parent, 'end', child[0], text=child[0], tag='ttk')
self.tre_modules.set(child[0], 'cpu', child[1])
self.tre_modules.set(child[0], 'pg', child[2])
for parent in self.module_library.get_parents():
self.tre_modules.insert('', 'end', parent, text=parent, tag='ttk')
for child in self.module_library.get_children(parent):
self.tre_modules.insert(parent, 'end', child[0], text=child[0], tag='ttk')
self.tre_modules.set(child[0], 'cpu', child[1])
self.tre_modules.set(child[0], 'pg', child[2])
# Grid management.
frm_modules.grid(column=0, row=1)
self.tre_modules.grid(column=0, row=0, sticky=NW, padx=3, pady=3)
scb_modules.grid(column=1, row=0, sticky=NE+S, pady=4)
self.tre_modules.configure(yscrollcommand=scb_modules.set)
# Bindings
self.tre_modules.bind('<Double-1>', self.add_module)
def fitting_display(self):
""" Displays all the current fitting information. """
# Set variables.
fitting_list = StringVar(value=self.current_fit.get_all_modules())
# Creates the widgets needed for this display.
frm_fitting_display = Frame(self, height=300)
self.lbx_fitting = Listbox(frm_fitting_display, listvariable=fitting_list, width=48, height=20, font='TkFixedFont', bg='white')
# Grid management.
frm_fitting_display.grid(column=1, row=1, sticky=W+E+N+S)
self.lbx_fitting.grid(column=0, row=0, sticky=W+E+N+S)
# Bindings
self.lbx_fitting.bind('<Double-1>', self.remove_module)
def stats_display(self):
""" Displays fitting stats. """
# Initialize variables.
cpu_text = '%s/%s' % (self.current_fit.current_cpu, self.current_fit.max_cpu)
pg_text = '%s/%s' % (self.current_fit.current_pg, self.current_fit.max_pg)
cpu_over = self.current_fit.get_cpu_over()
pg_over = self.current_fit.get_pg_over()
# Creates the holding widgets.
nbk_stats = ttk.Notebook(self)
#nbk_stats.grid_propagate(False)
frm_overview = Frame(self)
nbk_stats.add(frm_overview, text='Overview')
# Creates widgets for Dropsuit Type.
lfr_dropsuit_type = ttk.Labelframe(frm_overview, text='Dropsuit Type:')
lbl_dropsuit_type = Label(lfr_dropsuit_type, text=self.current_fit.ds_name).grid(column=0, row=0)
# Creates widgets for Resources.
lfr_resources = ttk.Labelframe(frm_overview, text='Resources')
lbl_cpu1 = Label(lfr_resources, text='CPU:').grid(column=0, row=0, sticky=W)
lbl_cpu2 = Label(lfr_resources, text=cpu_text).grid(column=1, row=0)
lbl_pg1 = Label(lfr_resources, text='PG:').grid(column=2, row=0, sticky=W, padx=10)
lbl_pg2 = Label(lfr_resources, text=pg_text).grid(column=3, row=0)
lbl_cpu3 = Label(lfr_resources, text=cpu_over).grid(column=1, row=1, sticky=E)
lbl_pg3 = Label(lfr_resources, text=pg_over).grid(column=3, row=1, sticky=E)
lfr_resources.columnconfigure(0, minsize=30)
lfr_resources.columnconfigure(1, minsize=90)
lfr_resources.columnconfigure(2, minsize=20)
lfr_resources.columnconfigure(3, minsize=75)
# Creates widgets for main offenses.
lfr_offenses = ttk.Labelframe(frm_overview, text='Main Offense')
lbl_weapon = Label(lfr_offenses, text=self.current_fit.get_primary_weapon_name()).grid(column=0, row=0, columnspan=4)
lbl_dmg1 = Label(lfr_offenses, text='Damage: ').grid(column=0, row=1, sticky=W)
lbl_dmg2 = Label(lfr_offenses, text=self.current_fit.get_primary_stats('damage')).grid(column=1, row=1, sticky=E)
lbl_rof1 = Label(lfr_offenses, text='RoF:').grid(column=2, row=1, sticky=W, padx=10)
lbl_rof2 = Label(lfr_offenses, text=self.current_fit.get_primary_stats('rate_of_fire')).grid(column=3, row=1, sticky=E)
lbl_dps1 = Label(lfr_offenses, text='DPS: ').grid(column=0, row=2, sticky=W)
lbl_dps2 = Label(lfr_offenses, text=self.current_fit.get_primary_dps()).grid(column=1, row=2, sticky=E)
lbl_dpm1 = Label(lfr_offenses, text='DPMag: ').grid(column=2, row=2, sticky=W, padx=10)
lbl_dpm2 = Label(lfr_offenses, text=self.current_fit.get_primary_dpm()).grid(column=3, row=2, sticky=E)
lfr_offenses.columnconfigure(0, minsize=60)
lfr_offenses.columnconfigure(1, minsize=45)
lfr_offenses.columnconfigure(2, minsize=60)
lfr_offenses.columnconfigure(3, minsize=60)
# Creates widgets for Defenses.
lfr_defenses = ttk.Labelframe(frm_overview, text='Defenses')
lbl_shield1 = Label(lfr_defenses, text='Shield HP:').grid(column=0, row=0, sticky=W)
lbl_shield2 = Label(lfr_defenses, text=self.current_fit.get_shield_hp()).grid(column=1, row=0, sticky=E)
lbl_recharge1 = Label(lfr_defenses, text='Recharge:').grid(column=2, row=0, sticky=W, padx=10)
lbl_recharge2 = Label(lfr_defenses, text=self.current_fit.get_shield_recharge()).grid(column=3, row=0, sticky=E)
lbl_armor1 = Label(lfr_defenses, text='Armor HP:').grid(column=0, row=1, sticky=W)
lbl_armor2 = Label(lfr_defenses, text=self.current_fit.get_armor_hp()).grid(column=1, row=1, sticky=E)
lbl_repair1 = Label(lfr_defenses, text='Repair:').grid(column=2, row=1, sticky=W, padx=10)
lbl_repair2 = Label(lfr_defenses, text=self.current_fit.get_armor_repair_rate()).grid(column=3, row=1, sticky=E)
lfr_defenses.columnconfigure(0, minsize=70)
lfr_defenses.columnconfigure(1, minsize=50)
lfr_defenses.columnconfigure(2, minsize=70)
lfr_defenses.columnconfigure(3, minsize=35)
# Creates widgets for Sensors.
lfr_sensors = ttk.Labelframe(frm_overview, text='Sensors')
lbl_prof1 = Label(lfr_sensors, text='Scan Profile: ').grid(column=0, row=0, sticky=W)
lbl_prof2 = Label(lfr_sensors, text=self.current_fit.get_scan_profile()).grid(column=1, row=0, sticky=E)
lbl_prec1 = Label(lfr_sensors, text='Scan Precision: ').grid(column=0, row=1, sticky=W)
lbl_prec2 = Label(lfr_sensors, text=self.current_fit.get_scan_precision()).grid(column=1, row=1, sticky=E)
lbl_radi1 = Label(lfr_sensors, text='Scan Radius: ').grid(column=0, row=2, sticky=W)
lbl_radi2 = Label(lfr_sensors, text=self.current_fit.get_scan_radius()).grid(column=1, row=2, sticky=E)
# Grid management.
nbk_stats.grid(column=2, row=0, rowspan=3, sticky=W+E+N+S, padx=3, pady=3)
lfr_dropsuit_type.grid(column=0, row=0, sticky=EW)
lfr_resources.grid(column=0, row=1, sticky=EW)
lfr_offenses.grid(column=0, row=2, sticky=EW)
lfr_defenses.grid(column=0, row=3, sticky=EW)
lfr_sensors.grid(column=0, row=4, sticky=EW)
frm_overview.columnconfigure(0, minsize=250)
def new_dropsuit_window(self):
""" Handles creating a whole new dropsuit. """
dropsuit_window = DropsuitWindow(self)
def add_character_window(self):
""" """
add_character_window = AddCharacterWindow(self)
def edit_character_window(self):
""" Runs the class which deals with editing or adding a character. """
character_edit_window = CharacterEditWindow(self, self.current_char.name)
def delete_character_window(self):
""" """
delete_character = DeleteCharacterWindow(self)
def delete_fitting_window(self):
""" """
delete_fitting = DeleteFittingWindow(self)
def load_new_dropsuit(self, fitting_name, dropsuit):
""" Called from the DropsuitWindow class. This will load the dropsuit
given by DropsuitWindow into the current fit. """
# Change the fit to hold the selected dropsuit.
self.current_fit = Fitting(fitting_name, self.current_char, dropsuit)
self.fitting_library.save_fitting(self.current_fit)
# Call the fitting_display and stats_display functions.
self.combobox_fitting()
self.fitting_display()
self.stats_display()
def add_module(self, *args):
""" Adds a module to the fitting. """
# Find what is selected.
module_name = self.tre_modules.selection()[0]
# Check to see if its a module or weapon, get the item requested.
if module_name in self.module_library.get_names():
self.current_fit.add_module(module_name)
else:
self.current_fit.add_weapon(module_name)
# Save the changes.
self.fitting_library.save_fitting(self.current_fit)
# Display the change.
self.fitting_display()
self.stats_display()
def remove_module(self, *args):
""" Removes a module from the fitting. """
# Find what is selected.
listbox_index = self.lbx_fitting.curselection()
module_name = self.lbx_fitting.get(listbox_index)
self.current_fit.remove_module(module_name)
# Save the changes.
self.fitting_library.save_fitting(self.current_fit)
# Display the change.
self.fitting_display()
self.stats_display()
def change_character(self, *args):
""" Changes the current characters for this fitting. """
name = self.cbx_character.get()
# Change the character
self.current_char = self.character_library.get_character(name)
self.current_fit.change_character(self.current_char)
# Display the change.
self.fitting_display()
self.stats_display()
def change_fitting(self, *args):
""" Changes the current fitting. """
fitting_name = self.cbx_fitting.get()
# Change the current fitting.
self.current_fit = self.fitting_library.get_fitting(fitting_name)
# Display the change.
self.fitting_display()
self.stats_display()
def update_character(self, character):
""" Called by CharacterEditWindow. This will update a character with
the changed skills. """
# Reload character data. THIS IS A HACK. Add better methods for changing and updating characters.
self.character_library = CharacterLibrary()
self.current_char = self.character_library.get_character(self.current_char.name)
self.current_fit.change_character(self.current_char)
# Display the changes.
self.fitting_display()
self.stats_display()
# Reloads the character dropdown menu. Needed if a new character is added.
self.combobox_character()
def update_fitting(self, fitting):
""" Called by DeleteFittingWindow. """
self.fitting_library = FittingLibrary()
# Selects a fitting and makes it active.
misc_fitting = self.fitting_library.get_fitting_list()[0]
self.current_fit = self.fitting_library.get_fitting(misc_fitting)
# Display the changes.
self.combobox_fitting()
self.fitting_display()
self.stats_display()