main_tkinter.py

import ctypes
import tkinter as tk
from tkinter import *
from tkinter import ttk
from tkinter.ttk import *
from tkinter import font as tkfont

import tkinter.messagebox
from tkinter.filedialog import askopenfilename, asksaveasfilename

from PIL import ImageGrab
from PIL import Image
import base64

import project_tk as project_tk
import gui_styles_tk as gui_styles
import file_menu as fm
import about_tkinter as about_tk
import components_tk
import undo_redo_tk
import database_tk
import seat_summary_tk
import toolbar
import time

class MainApplication(tk.Frame):
	def __init__(self, parent, *args, **kwargs):
		tk.Frame.__init__(self, parent, *args, **kwargs)
		user32 = ctypes.windll.user32
		self.screen_width = (user32.GetSystemMetrics(0))
		self.screen_height = (user32.GetSystemMetrics(1))

		self.parent = parent
		
		parent.withdraw()
		splash = Splash(parent)
		self.parent.wait_window(splash)

		if splash.agree:
			parent.deiconify()
		else:
			quit()


		self.icons_folder = r'C:\Users\domhn\Documents\Python\Pycabin_Tkinter\Icons'
		gui_styles.setup_icons(self)
		self.setup_colors()
		self.setup_variables()

		self.parent.title(self.titlebar_text)

		gui_styles.setup_fonts(self)
		
		self.setup_menu()
		self.setup_main_frames()

		#Toolbar
		toolbar.create_toolbar(self)
		
		
		self.setup_seat_summary_page()
	
		
		self.states = undo_redo_tk.Undo_Redo(self)
		self.setup_project_page()
		components_tk.show_frame(self, 'Seats')
		components_tk.show_frame(self, 'Project')
		
		#fix for treeview tags
		self.style = ttk.Style()
		self.style.map('Treeview', foreground=self.fixed_map('foreground'), background=self.fixed_map('background'))

		self.no_of_default_pages = len(self.frames.keys())
		self.done_something = False # if the user has edited something or not

	def setup_variables(self):
		
		self.frames = {}
		self.version = '0.22.0'
		self.save_file = None
		self.cabin_database = r'C:\Users\domhn\Documents\Python\Pycabin_Tkinter\V0.08\test.db'
		self.version_title = f"PYCabin V{self.version}"
		self.titlebar_text = f"PYCabin V{self.version}"
		self.in_fullscreen = False
		

	def update_titlebar(self, mode):

		if not self.save_file:
			text = ''
		else:
			text = self.save_file

		if mode == 'save':
			self.titlebar_text = f"{text} - PYCabin V{self.version}"
		elif mode == 'edit':
			self.titlebar_text = f"{text}* - PYCabin V{self.version}"

		self.parent.title(self.titlebar_text)
		self.done_something = True

	def setup_main_frames(self):
		ttk.Separator(self.parent,orient=HORIZONTAL).grid(row=1, column=0, columnspan=4, sticky='nsew')
		self.rootpane = tk.PanedWindow(self.parent, orient=tk.HORIZONTAL)
		self.rootpane.grid(row=2,column=0, columnspan=4,sticky="nsew")
		self.parent.grid_rowconfigure(2, weight=1)
		self.parent.grid_columnconfigure(3, weight=1)
		self.frame = Frame(self.parent)
		
		self.sidebar_frame = tk.Frame()
		self.sidebar_frame.grid_rowconfigure(1, weight=1)
		self.sidebar_frame.grid_columnconfigure(19, weight=1)
		self.rootpane.add(self.sidebar_frame)
		self.setup_main_treeview()
		self.frame.grid(row=0,column=0, sticky="n")
		
		# the container is where we'll stack a bunch of frames
		self.container = tk.Frame(self.rootpane)
		self.container.grid_columnconfigure(0, weight=1)
		self.rootpane.add(self.container,stretch="always")
		

	def fixed_map(self, option):
		# Fix for setting text colour for Tkinter 8.6.9
		# From: https://core.tcl.tk/tk/info/509cafafae
		#
		# Returns the style map for 'option' with any styles starting with
		# ('!disabled', '!selected', ...) filtered out.

		# style.map() returns an empty list for missing options, so this
		# should be future-safe.
		return [elm for elm in self.style.map('Treeview', query_opt=option) if
			elm[:2] != ('!disabled', '!selected')]
		
	def setup_main_treeview(self):
	
		# tk.Button(self.sidebar_frame, image = self.close_icon2, background='white', relief=FLAT,
		# 	command= lambda action = False: self.open_close_all_nodes(action)).grid(row=0, column=0, sticky='w', padx = 0)
		# tk.Button(self.sidebar_frame, image = self.open_icon2, background='white', relief=FLAT,
		# 	command= lambda action = True: self.open_close_all_nodes(action)).grid(row=0, column=1, sticky='w', padx = 0)
		# tk.Button(self.sidebar_frame, image = self.up_icon2, background='white', relief=FLAT,
		# 	command = self.moveUp).grid(row=0, column=2, sticky='w', padx = 0)
		# tk.Button(self.sidebar_frame, image = self.down_icon2, background='white', relief=FLAT,
		# 	command = self.moveDown).grid(row=0, column=3, sticky='w', padx = 0)

		self.main_treeview = ttk.Treeview(self.sidebar_frame,selectmode='browse', show="tree")
		self.main_treeview.grid(row=1, column=0, columnspan = 20, sticky='nsew')
		
		tree_scrollbar = Scrollbar(self.sidebar_frame, command=self.main_treeview.yview)
		tree_scrollbar.grid(row=1, column=20, sticky='nsew')
		self.main_treeview.config(yscrollcommand=tree_scrollbar.set)
		
		fi = self.folder_icon2
		abi = self.airbus_icon2
		bi = self.boeing_icon2

		# ________ Project ________
		item = self.main_treeview.insert("",'end','Project',text='Project',image = fi)
		
		# ________ Aircraft ________
		# item = self.main_treeview.insert("",'end','Aircraft',text='Aircraft',image = fi)
		# item = self.main_treeview.insert("Aircraft",'end','A320 Aircraft',text=' A320 Family',image = abi)
		# item = self.main_treeview.insert("Aircraft",'end','737 Aircraft',text=' B737 Family',image = bi)

		# ________ Seats ________
		item = self.main_treeview.insert("",'end','Seats',text='Seats',image = fi)
		item = self.main_treeview.insert("Seats",'end','A320 Seats',text=' A320 Family',image = abi)
		# item = self.main_treeview.insert("Seats",'end','737 Seats',text=' 737 Family',image = bi)

		# ________ Monuments ________
		item = self.main_treeview.insert("",'end','Monuments',text='Monuments',image = fi)
		item = self.main_treeview.insert("Monuments",'end','A320 Monuments',text='A320 Family',image = abi)
		item = self.main_treeview.insert("A320 Monuments",'end','A320 Windbreakers',text='Windbreakers',)		

		# ________ LOPA ________
		item = self.main_treeview.insert("",'end','LOPAs',text='LOPAs',image = fi)
		item = self.main_treeview.insert("LOPAs",'end','A320 LOPAs',text=' A320 Family',image = abi)

		# ________ PSU ________
		item = self.main_treeview.insert("",'end','PSUs',text='PSU Layouts',image = fi)
		item = self.main_treeview.insert("PSUs",'end','A320 PSUs',text=' A320 Family',image = abi)

		# ________ OHSC ________
		item = self.main_treeview.insert("",'end','OHSCs',text='OHSC Layouts',image = fi)
		item = self.main_treeview.insert("OHSCs",'end','A320 OHSCs',text=' A320 Family',image = abi)

		# ________ Emergency Equipment ________
		item = self.main_treeview.insert("",'end','Emergency Equipment',text='Emergency Equipment',image = fi)
		item = self.main_treeview.insert("Emergency Equipment",'end','A320 EE',text=' A320 Family',image = abi)

		# ________ EELs ________
		item = self.main_treeview.insert("",'end','EELs',text='EELs',image = fi)
		item = self.main_treeview.insert("EELs",'end','A320 EELs',text=' A320 Family',image = abi)

		# ________ EEL Comparisons ________
		item = self.main_treeview.insert("",'end','EEL Comparisons',text='EEL Comparisons',image = fi)
		item = self.main_treeview.insert("EEL Comparisons",'end','A320 EEL Comparisons',text=' A320 Family',image = abi)

		# ________ Changes ________
		# item = self.main_treeview.insert("",'end','Changes',text='Changes',image = fi)

		
		#self.rootpane.add(self.main_treeview)
		
		self.main_treeview.bind('<<TreeviewSelect>>',lambda event, : self.ProcessOnSingleClick_Main_Tree(event))

		self.popup_menu = tk.Menu(self, tearoff=0)
		self.popup_menu.add_command(label="Delete        ",command= lambda mainapp=self: components_tk.delete_component(mainapp))		
		self.popup_menu.add_command(label="Copy          ",command= lambda mainapp=self: components_tk.copy_component(mainapp))
		
		#self.popup_menu.add_separator()
		
		
		self.main_treeview.bind("<Button-3>", self.popup)

		self.ids_not_allowed = ['Project', 'Aircraft', 'A320 Aircraft', '737 Aircraft', 'Aircraft', 'Seats', 'A320 Seats',
								'737 Seats', 'Monuments', 'A320 Monuments', 'A320 Windbreakers', 'A320 LOPAs', 'Windbreaker'
									'Windbreakers', 'A320 PSUs', 'PSUs', 'OHSCs', 'A320 OHSCs', 'EELs', 'A320 EELs',
									]		
		
		# self.treeview_nodes = {'Aircraft': ['A320 Aircraft', '737 Aircraft'], 'Seats': ['A320 Seats', '737 Seats'], 'Monuments': ['A320 Windbreakers'],
								# 'Windbreakers': ['A320 Windbreakers'],
								# 'LOPAs': ['A320 LOPAs'], 'Changes': ['Changes']}

		self.treeview_nodes = {'Aircraft': ['A320 Aircraft'], 'Seats': ['A320 Seats'], 'Monuments': ['A320 Windbreakers'],
								'Windbreakers': ['A320 Windbreakers'],
								'LOPAs': ['A320 LOPAs'], 'PSUs': ['A320 PSUs'], 'Emergency Equipment': ['A320 EE'],
								'OHSCs': ['A320 OHSCs'], 'EELs': ['A320 EELs'], 'EEL Comparisons': ['A320 EEL Comparisons'],
								'Changes': ['Changes']}
								

	def setup_menu(self):
		menu = tk.Menu(self.master)
		self.master.config(menu=menu)
		#File					  
		file_menu = tk.Menu(menu, tearoff = 0)
		menu.add_cascade(label='File',menu=file_menu)
		
		file_menu.add_command(label = 'New                     Ctrl+N', command = lambda self=self: fm.new_project(mainapp=self))
		file_menu.add_command(label = 'Load                    Ctrl+O', command = lambda self=self: fm.load(mainapp = self))
		file_menu.add_command(label = 'Save                     Ctrl+S', command = lambda self=self: fm.save(mainapp= self))
		file_menu.add_command(label = 'Save As                Ctrl+Shift+S', command = lambda self=self: fm.save_as(mainapp= self))
		#file_menu.add_command(label = 'Save As', command = lambda self=self, mode='save as': fm.save(self, mode))

		#________ EDIT ________				  
		edit_menu = tk.Menu(menu, tearoff = 0)
		menu.add_cascade(label='Edit',menu=edit_menu)
		
		edit_menu.add_command(label = 'Undo                     Ctrl+Z', command = lambda: self.states.undo())
		edit_menu.add_command(label = 'Redo                      Ctrl+Y', command = lambda: self.states.redo())
		

		#________ INSERT ________					  
		insert_menu = tk.Menu(menu, tearoff = 0)
		menu.add_cascade(label='Insert',menu=insert_menu)

		# insert_menu.add_command(label = 'Aircraft', command = lambda self=self, type='Aircraft': components_tk.new_component(self, type))
		#insert_menu.add_command(label = 'Seats - Multiple', command = lambda self=self, type='Seats - Multiple': components_tk.new_component(self, type))
		

		#
		insert_component_menu = tk.Menu(menu, tearoff = 0)
		insert_menu.add_cascade(label = 'Components',menu = insert_component_menu)
		insert_component_menu.add_command(label = 'Seat', command = lambda self=self, type='Seat': components_tk.new_component(self, type))
		insert_component_menu.add_command(label = 'Multiple Seats', command = lambda self=self, type='Seats - Multiple': components_tk.new_component(self, type))
		insert_component_menu.add_command(label = 'Windbreaker', command = lambda self=self, type='Windbreaker': components_tk.new_component(self, type))
		insert_component_menu.add_command(label = 'Emergency Equipment', command = lambda self=self, type='Emergency Equipment': components_tk.new_component(self, type))

		insert_layout_menu = tk.Menu(menu, tearoff = 0)
		insert_menu.add_cascade(label = 'Layouts',menu = insert_layout_menu)

		insert_layout_menu.add_command(label = 'LOPA', command = lambda self=self, type='LOPA': components_tk.new_component(self, type))
		
		insert_layout_menu.add_command(label = 'PSU Layout', command = lambda self=self, type='PSU': components_tk.new_component(self, type))

		insert_layout_menu.add_command(label = 'OHSC Layout', command = lambda self=self, type='OHSC': components_tk.new_component(self, type))

		insert_layout_menu.add_command(label = 'Emergency Equipment Layout', command = lambda self=self, type='EEL': components_tk.new_component(self, type))
		insert_layout_menu.add_command(label = 'EEL Comparison', command = lambda self=self, type='EEL Comparison': components_tk.new_component(self, type))

		# insert_menu.add_command(label = 'Change', command = lambda self=self, type='Change': components_tk.new_component(self, type))
		
		# ________ DATABASE ________
		#db_menu = tk.Menu(menu, tearoff = 0)
		#menu.add_cascade(label='Database',menu=db_menu)

		#db_menu.add_command(label = 'Add Components to Database', command = lambda self=self: database_tk.add_components(self))
		#db_menu.add_command(label = 'Load Components From Database', command = lambda self=self: database_tk.load_components(self))
		
		# ________ ABOUT ________
		about_menu = tk.Menu(menu, tearoff = 0)
		menu.add_cascade(label='About',menu=about_menu)
		
		about_menu.add_command(label = 'About PYCabin', command = lambda self=self: about_tk.about(self))

	def ProcessOnSingleClick_Main_Tree(self, event):
		item_iid = event.widget.selection()[0]
		parent_iid = event.widget.parent(item_iid)
		
		ids_to_ignore = ['A320 Aircraft', '737 Aircraft', 'A320 Seats', '737 Seats',
						'A320 Monuments', 'A320 Windbreakers', 'A320 LOPAs', 'A320 PSUs', 'A320 OHSCs', 'A320 EE',
						'A320 EELs', 'A320 EEL Comparisons']

		if parent_iid:
			if item_iid not in ids_to_ignore:
				components_tk.show_frame(self, event.widget.item(item_iid, 'text'))
		if not parent_iid:
			if (event.widget.item(item_iid, 'text')) == 'Project':
				components_tk.show_frame(self, "Project")
			if (event.widget.item(item_iid, 'text')) == 'Seats':
				components_tk.show_frame(self, "Seats")
				
	def open_close_all_nodes(self, action):
		
		for node in self.treeview_nodes:
			if node in self.main_treeview.get_children():
				self.main_treeview.item(node, open=action)
				
				for node in self.treeview_nodes[node]:
					self.main_treeview.item(node, open=action)

	def moveUp(self):
		leaves = self.main_treeview.selection()
		for i in leaves:
			self.main_treeview.move(i, self.main_treeview.parent(i), self.main_treeview.index(i)-1)

	def moveDown(self):
		leaves = self.main_treeview.selection()
		for i in leaves:
			self.main_treeview.move(i, self.main_treeview.parent(i), self.main_treeview.index(i)+1)
		
	def setup_fonts(self):
	
		self.title_font = tkfont.Font(family='Helvetica', size=18, weight="bold", slant="italic")
		self.title_font.configure(underline=True)

	def setup_colors(self):
	
		self.red_color = 'IndianRed1'
		self.green_color = 'green yellow'
		self.grey_color = 'grey'
		

	def setup_project_page(self):
	
		self.frames['Project'] = project_tk.Project_Page_Tk(self.container, self)
		self.frames['Project'].pack(fill="both", expand=True)
		# self.frames['Project'].grid(row=0, column=0, sticky="NSEW")
		# self.frames['Project'].grid_rowconfigure(0, weight=0)
		# self.frames['Project'].grid_rowconfigure(1, weight=1)
		# self.frames['Project'].grid_columnconfigure(0, weight=1)
		
		self.current_frame = self.frames['Project']
		
	def setup_seat_summary_page(self):
	
		self.frames['Seats'] = seat_summary_tk.Seat_Page_Summary_Tk(self.container, self)
		self.frames['Seats'].pack(fill="both", expand=True)
		
	def popup(self,event):
		try:
			self.popup_menu.selection = self.main_treeview.set(self.main_treeview.identify_row(event.y))
			self.main_treeview.focus(self.main_treeview.identify_row(event.y))
			self.main_treeview.selection_set(self.main_treeview.identify_row(event.y))
			#print(self.main_treeview.selection()[0])
			if self.main_treeview.selection()[0] not in self.ids_not_allowed:			
				self.popup_menu.tk_popup(event.x_root, event.y_root, 0)
		finally:
				self.popup_menu.grab_release()	

	def grab_screen(self):

		img = ImageGrab.grab()
		name = asksaveasfilename()
		img = img.crop((0, 0, 1920, 1010)) 
		img.save(name)

	def fullscreen(self, root):

		if not self.in_fullscreen:
			root.attributes("-fullscreen", True)
			self.in_fullscreen = True
		else:
			root.attributes("-fullscreen", False)
			self.in_fullscreen = False

class Splash(tk.Toplevel):
	def __init__(self, parent):
		tk.Toplevel.__init__(self, parent)
		self.title("Splash")

		## required to make window show before the program gets to the mainloop
		self.update()

		self.agree = False
		
		l = tk.Label(self, text="""
		PYCabin is an app intended to help engineers manage Part 21 DOA projects.

		Note this app is currently in Beta stage and development and testing are 
		both ongoing.
					
		PYCabin makes no promise of warranty, satisfaction, performance, or
		anything else. Understand that your use of this app is completely
		at your own risk. Please select Agree below to continue using this app.""", anchor="e", justify=LEFT)

		l.grid(row=0, column=0, columnspan=4, sticky='nw')

		button1 = ttk.Button(self, text="Agree", 
								command=self.cleanup)
		button2 = ttk.Button(self, text="Disagree",
								command=self.disagree)

		button1.grid(row=1, column=1, columnspan=1, sticky='ne')
		button2.grid(row=1, column=2, columnspan=1, sticky='nw')

		self.geometry("550x200")

	def cleanup(self):

		self.agree = True
		self.destroy()

	def disagree(self):

		self.destroy()



if __name__ == "__main__":
	root = tk.Tk()
	root.resizable(width=tk.TRUE, height=tk.TRUE)
	#MainApplication(root).pack(side="top", fill="both", expand=True)
	MA = MainApplication(root)
	MA.grid(row=1, columnspan=4, sticky='nsew')

	
	root.geometry('{}x{}'.format(MA.screen_width, MA.screen_height))    
	
	root.bind('<Control-n>', lambda event, MA=MA: fm.new_project(event, MA))
	root.bind('<Control-o>', lambda event, MA=MA: fm.load(event, MA))
	root.bind('<Control-s>', lambda event, MA=MA: fm.save(event, MA))
	root.bind('<Control-Shift-KeyPress-S>', lambda event, MA=MA: fm.save_as(event, MA))
	root.bind('<Control-z>', lambda event, MA=MA: MA.states.undo())
	root.bind('<Control-y>', lambda event, MA=MA: MA.states.redo())
	root.bind('<Control-g>', lambda event, MA=MA: MA.grab_screen())
	root.bind('<F12>', lambda event, root=root, MA=MA: MA.fullscreen(root))
	root.state('zoomed')

	# add logo
	icondata= base64.b64decode(MA.pycabin_logo)
	## The temp file is icon.ico
	tempFile= "icon.ico"
	iconfile= open(tempFile,"wb")
	## Extract the icon
	iconfile.write(icondata)
	iconfile.close()
	root.wm_iconbitmap(tempFile)


	root.mainloop()