Post by Uncle Buddy on Dec 25, 2023 23:58:18 GMT -8
<drive>:\treebard\app\python\media.py Last Changed 2024-03-21
# media.py
import tkinter as tk
import sqlite3
from os import remove, path
from PIL import Image, ImageTk
from files import get_current_tree, appwide_db_path, current_drive
from widgets import make_formats_dict, open_message, Combobox
from messages import images_msg
from query_strings import (
update_image_setting_delete_user_images, select_image_setting_user_images,
select_current_image_directory, update_preferences_default_images,
update_image_setting_image_male, update_image_setting_image_female,
update_image_setting_image_unisex, update_image_setting_image_nested_place,
update_image_setting_image_source, select_media_id,
select_media_links_count_image_person, select_current_person_nested_place_source,
select_media_links_count_image_nested_place, insert_media_links_image_person,
select_media_links_count_image_source, select_media_links_main_image_person,
delete_media_links_image_person, select_media_links_main_image_nested_place,
delete_media_links_image_nested_place, insert_image_new, delete_media,
select_media_links_main_image_source, delete_media_links_image_source,
delete_media_links_image, update_media_links_main_image,
select_image_setting_builtin_default_images, insert_media_links_image_nested_place,
insert_media_links_image_source)
import dev_tools as dt
from dev_tools import look, seeline
def get_current_elements(family_tree_id):
conn = sqlite3.connect(appwide_db_path)
cur = conn.cursor()
current_tree, current_dir = get_current_tree(family_tree_id, cur)
cur.execute("ATTACH ? as tree", (current_tree,))
cur.execute(select_current_person_nested_place_source)
current_person_id, current_nested_place, current_source = cur.fetchone()
cur.execute("DETACH tree")
cur.close()
conn.close()
return (
current_tree, current_dir, current_person_id, current_nested_place,
current_source)
class MediaTab(tk.Frame):
def __init__(
self, master, formats, treebard, family_tree, family_tree_id,
main, *args, **kwargs):
tk.Frame.__init__(self, master, *args, **kwargs)
self.master = master
self.formats = formats
self.treebard = treebard
self.family_tree = family_tree
self.family_tree_id = family_tree_id
self.main = main
self.usepicvar = tk.IntVar()
if self.main.use_default_images is False:
self.usepicvar.set(0)
elif self.main.use_default_images:
self.usepicvar.set(1)
self.radframe_children = []
self.imagegot = ""
self.types = ("male", "female", "unisex", "place", "source")
self.make_inputs()
self.get_user_prefs()
def combobox_selected(self):
pictype = self.image_type.var.get()
for k,v in self.user_pics.items():
if k == pictype:
self.user_input.delete(0, "end")
self.user_input.insert(0, v)
break
def make_inputs(self):
instrux = tk.Label(
self,
text="Any .jpg, .png, .gif, or .bmp file on your computer can be linked to any number of person, place, or source elements in your tree. First add the file to the tree, then to link it to an element, tell treebard what type of element it is, and select the file from the project images folder. Images, main images, and default images are explained in the right-click context help menus.",
wraplength=750, justify="left", bd=1, relief="raised")
labframe1 = tk.LabelFrame(
self, text="ADD AN IMAGE TO THE TREE")
addlab = tk.Label(
labframe1, text="Select an image:")
self.add_input = tk.Entry(labframe1, width=64)
browse1 = tk.Button(labframe1, text=" ... ")
browse1.config(
command=lambda browse1=browse1, entry=self.add_input:
self.open_image_explorer(
browse1, entry))
caption_lab = tk.Label(labframe1, text="Caption:")
self.add_caption = tk.Entry(labframe1, width=64)
buttons1 = tk.Frame(labframe1)
self.ok_new_image = tk.Button(
buttons1, text="OK", width=6, command=self.ok_add_img)
frame = tk.Frame(self)
typelab = tk.Label(
frame, text="Changes below will relate to what element type?")
self.image_type = Combobox(
frame, self.formats, family_tree=self.family_tree,
values=self.types, entry_width=6)
labframe2 = tk.LabelFrame(
self, text="LINK OR UNLINK CURRENT ELEMENT AND EXISTING IMAGE")
linklab = tk.Label(
labframe2,
text="Select an image that's already in the tree:")
self.link_input = tk.Entry(labframe2, width=64)
self.browse2 = tk.Button(labframe2, text=" ... ")
self.browse2.config(
command=lambda browse2=self.browse2, entry=self.link_input:
self.open_image_explorer(
browse2, entry))
self.linkvar = tk.IntVar(None, 99)
rads = (
"Link to Current Element", "Unlink from Current Element",
"Unlink from All Elements", "Delete from Tree")
self.radframe = tk.Frame(labframe2)
for idx, text in enumerate(rads):
rad = tk.Radiobutton(
self.radframe, text=text,
variable=self.linkvar, value=idx, anchor="w")
rad.grid(column=idx, row=0)
self.radframe_children.append(rad)
buttons2 = tk.Frame(labframe2)
self.ok_link_pic = tk.Button(
buttons2, text="OK", width=6, command=self.ok_link_image)
labframe3 = tk.LabelFrame(self, text="ADD OR DELETE DEFAULT IMAGES")
radframe1 = tk.Frame(labframe3)
self.use_image = tk.Radiobutton(
radframe1, text="Use default images.", variable=self.usepicvar,
value=1)
self.no_image = tk.Radiobutton(
radframe1, text="Don't use default images.", variable=self.usepicvar,
value=0)
use_or_not = tk.Button(
radframe1, text="OK", command=self.ok_use_default_images)
userlab = tk.Label(
labframe3,
text="Add, delete or replace user-defined default image:")
self.user_input = tk.Entry(labframe3, width=64)
browse3 = tk.Button(
labframe3, text=" ... ")
browse3.config(
command=lambda browse1=browse3, entry=self.user_input:
self.open_image_explorer(
browse3, entry))
buttons3 = tk.Frame(labframe3)
self.ok_default_image = tk.Button(
buttons3, text="OK", command=self.ok_set_default_image, width=6)
self.reset = tk.Button(
buttons3,
text="REVERT TO BUILT-IN DEFAULT IMAGES", command=self.revert)
# children of self
instrux.grid(column=0, row=0, ipadx=9, ipady=6, columnspan=2, pady=(0,6))
labframe1.grid(column=0, row=1, columnspan=3, sticky="ew", pady=(6,0))
frame.grid(column=0, row=2, sticky="w", pady=(12,6))
labframe2.grid(column=0, row=3, columnspan=3, sticky="ew", pady=(6,0))
labframe3.grid(column=0, row=4, columnspan=3, sticky="ew", pady=(6,0))
# children of labframe1
labframe1.columnconfigure(1, weight=1)
addlab.grid(column=0, row=2, sticky="w", padx=(6,0))
self.add_input.grid(column=1, row=2, sticky="ew", padx=(6,0))
browse1.grid(column=2, row=2, sticky="w", padx=6)
caption_lab.grid(column=0, row=3, sticky="w", padx=(6,0))
self.add_caption.grid(column=1, row=3, sticky="ew", padx=(6,0))
buttons1.grid(column=1, row=4, sticky="e", columnspan=2, pady=6, padx=(0,6))
# children of buttons1
self.ok_new_image.grid(column=0, row=0, sticky="e")
# children of frame
typelab.grid(column=0, row=0, sticky="w")
self.image_type.grid(column=1, row=0, sticky="w", padx=6)
# children of labframe2
labframe2.columnconfigure(1, weight=1)
linklab.grid(column=0, row=0, sticky="w", padx=(6,0))
self.link_input.grid(column=1, row=0, sticky="ew", padx=(6,0))
self.browse2.grid(column=2, row=0, sticky="w", padx=6)
self.radframe.grid(column=0, row=1, sticky="w", columnspan=2)
buttons2.grid(column=1, row=2, sticky="e", columnspan=2, pady=6, padx=(0,6))
# children of buttons2
self.ok_link_pic.grid(column=0, row=0, sticky="e")
# children of labframe3
labframe3.columnconfigure(1, weight=1)
radframe1.grid(column=0, row=0, sticky="w", pady=(6,0))
userlab.grid(column=0, row=1, sticky="w", padx=(6,0))
self.user_input.grid(column=1, row=1, sticky="ew", padx=(6,0))
browse3.grid(column=2, row=1, sticky="w", padx=6)
buttons3.grid(column=1, row=2, sticky="e", pady=6, columnspan=2, padx=(0,6))
#children of radframe1
self.use_image.grid(column=0, row=0, sticky="w")
self.no_image.grid(column=1, row=0, sticky="w")
use_or_not.grid(column=2, row=0, sticky="w", padx=(12,0))
# children of buttons3
self.ok_default_image.grid(column=0, row=0, padx=6)
self.reset.grid(column=1, row=0, sticky="e")
self.add_input.focus_set()
def revert(self):
conn = sqlite3.connect(appwide_db_path)
conn.execute("PRAGMA foreign_keys = 1")
cur = conn.cursor()
cur.execute(update_image_setting_delete_user_images)
conn.commit()
cur.close()
conn.close()
def get_user_prefs(self):
conn = sqlite3.connect(appwide_db_path)
cur = conn.cursor()
cur.execute(select_image_setting_user_images)
self.user_pics = dict(zip(self.types, cur.fetchone()))
cur.close()
conn.close()
def open_image_explorer(self, evt_widget, entry):
conn = sqlite3.connect(appwide_db_path)
cur = conn.cursor()
self.current_tree, self.current_dir = get_current_tree(
self.family_tree_id, cur)
cur.execute("ATTACH ? as tree", (self.current_tree,))
if evt_widget == self.browse2:
openpic_dir = f"{current_drive}treebard/data/{self.current_dir}/images"
# openpic_dir = "{}treebard\data\{}\images".format(
# current_drive, self.current_dir)
title = "Select Image to Link or Unlink from an Element"
else:
cur.execute(select_current_image_directory)
result = cur.fetchone()
if result:
openpic_dir = result[0]
title = "Select Image to Import to the Project"
self.imagegot = tk.filedialog.askopenfilename(
initialdir=openpic_dir,
title=title,
defaultextension=".jpg",
filetypes=(
('', '*.jpg'),
('', '*.png'),
('', '*.gif'),
('', '*.bmp'),
('all files', '*.*')))
cur.execute("DETACH tree")
cur.close()
conn.close()
entry.delete(0, "end")
entry.insert(0, self.imagegot)
def ok_use_default_images(self):
conn = sqlite3.connect(appwide_db_path)
conn.execute("PRAGMA foreign_keys = 1")
cur = conn.cursor()
self.current_tree = get_current_tree(self.family_tree_id, cur)[0]
cur.execute("ATTACH ? as tree", (self.current_tree,))
use_default_images = self.usepicvar.get()
if use_default_images == 0:
self.main.use_default_images = False
self.main.top_pic_button.config(image="")
self.main.top_pic_button.image = ""
self.main.top_pic_place_button.config(image="")
self.main.top_pic_place_button.image = ""
self.main.top_pic_source_button.config(image="")
self.main.top_pic_source_button.image = ""
elif use_default_images == 1:
self.main.use_default_images = True
self.main.show_top_pic(cur)
self.main.show_top_pic_nested_place(cur)
self.main.show_top_pic_source(cur)
cur.execute(update_preferences_default_images, (use_default_images,))
conn.commit()
cur.execute("DETACH tree")
cur.close()
conn.close()
def ok_set_default_image(self):
kind = self.image_type.entry.get()
if kind not in self.types:
self.error_no_type()
return
conn = sqlite3.connect(appwide_db_path)
conn.execute("PRAGMA foreign_keys = 1")
cur = conn.cursor()
self.current_tree = get_current_tree(self.family_tree_id, cur)[0]
cur.execute("ATTACH ? as tree", (self.current_tree,))
contents = self.user_input.get()
file = self.imagegot.split("/")[-1]
if len(contents) != 0:
saved = Image.open(self.imagegot)
default_images_path = "{}treebard/data/settings/images/{}".format(
current_drive, file)
# overwrites file by same name if it exists
saved.save(default_images_path)
else:
file = ""
if kind == "male":
cur.execute(update_image_setting_image_male, (file,))
elif kind == "female":
cur.execute(update_image_setting_image_female, (file,))
elif kind == "unisex":
cur.execute(update_image_setting_image_unisex, (file,))
elif kind == "place":
cur.execute(update_image_setting_image_nested_place, (file,))
elif kind == "source":
cur.execute(update_image_setting_image_source, (file,))
conn.commit()
self.get_user_prefs()
self.reset_gui(kind, cur)
if len(file) != 0:
self.redraw_default_image(kind, file)
else:
self.redraw_default_image(kind, file, cur=cur, conn=conn)
self.main.get_default_images(cur)
conn.commit()
cur.execute("DETACH tree")
cur.close()
conn.close()
def ok_add_img(self):
file = self.add_input.get().strip().split("/")[-1]
caption = self.add_caption.get().strip()
if len(file) == 0:
self.error_no_input(self.add_input)
return
elif len(caption) == 0:
self.error_no_input(self.add_caption)
return
conn = sqlite3.connect(appwide_db_path)
conn.execute("PRAGMA foreign_keys = 1")
cur = conn.cursor()
cur.execute("ATTACH ? as tree", (self.current_tree,))
cur.execute(insert_image_new, (file, caption))
conn.commit()
if len(file) != 0:
saved = Image.open(self.imagegot)
tree_images_path = "{}treebard/data/{}/images/{}".format(
current_drive, self.current_dir, file)
# overwrites file by same name if it exists
saved.save(tree_images_path)
cur.execute("DETACH tree")
cur.close()
conn.close()
self.add_input.delete(0, "end")
self.add_caption.delete(0, "end")
def ok_link_image(self):
conn = sqlite3.connect(appwide_db_path)
conn.execute("PRAGMA foreign_keys = 1")
cur = conn.cursor()
cur.execute("ATTACH ? as tree", (self.current_tree,))
cur.execute(select_current_person_nested_place_source)
current_person_id, current_nested_place, current_source = cur.fetchone()
path_string = self.link_input.get().strip()
name = path_string.split("/")[-1]
if len(name) == 0:
self.error_no_input(self.link_input)
cur.execute(select_media_id, (name,))
result = cur.fetchone()
if result:
image_id = result[0]
select = self.linkvar.get()
kind = self.image_type.entry.get()
if kind not in self.types:
self.error_no_type()
return
if select > 3:
self.error_no_radio()
return
elif select == 0: #link to current elem
if kind in self.types[0:3]:
link_query = insert_media_links_image_person
cur.execute(
select_media_links_count_image_person, (current_person_id,))
image_count = cur.fetchone()[0]
element = current_person_id
elif kind == "place":
link_query = insert_media_links_image_nested_place
cur.execute(
select_media_links_count_image_nested_place,
(current_nested_place,))
image_count = cur.fetchone()[0]
element = current_nested_place
elif kind == "source":
link_query = insert_media_links_image_source
cur.execute(
select_media_links_count_image_source, (current_source,))
image_count = cur.fetchone()[0]
element = current_source
else:
print("error message radiobutton not selected")
if image_count != 0:
mainbool = 0
else:
mainbool = 1
cur.execute(link_query, (image_id, element, mainbool))
conn.commit()
self.reset_gui(kind, cur)
elif select == 1: #unlink curr elem
if kind not in self.types:
self.error_no_type()
return
elif kind in (self.types[0:3]):
cur.execute(select_media_links_main_image_person, (current_person_id,))
linked = cur.fetchall()
self.replace_main_image(linked, image_id, cur, conn)
cur.execute(delete_media_links_image_person, (current_person_id, image_id))
conn.commit()
elif kind == "place":
cur.execute(
select_media_links_main_image_nested_place,
(current_nested_place,))
linked = cur.fetchall()
self.replace_main_image(linked, image_id, cur, conn)
cur.execute(
delete_media_links_image_nested_place,
(current_nested_place, image_id))
conn.commit()
elif kind == "source":
cur.execute(select_media_links_main_image_source, (current_source,))
linked = cur.fetchall()
self.replace_main_image(linked, image_id, cur, conn)
cur.execute(delete_media_links_image_source, (current_source, image_id))
conn.commit()
self.reset_gui(kind, cur)
elif select == 2: # unlink all elem
cur.execute(delete_media_links_image, (image_id,))
conn.commit()
self.reset_gui(kind, cur)
elif select == 3: # delete from tree
if kind not in self.types:
self.error_no_type()
return
elif kind in (self.types[0:3]):
cur.execute(select_media_links_main_image_person, (current_person_id,))
linked = cur.fetchall()
self.replace_main_image(linked, image_id, cur, conn)
cur.execute(delete_media_links_image, (image_id,))
conn.commit()
cur.execute(delete_media, (image_id,))
conn.commit()
file = name.rsplit(".", 1)[0]
bare_path = path_string.rsplit("/", 1)[0]
resized = "{}/thumbnails/{}_resized_top_pic.png".format(bare_path, file)
if path.isfile(path_string):
remove(path_string)
if path.isfile(resized):
remove(resized)
elif kind == "place":
cur.execute(
select_media_links_main_image_nested_place,
(current_nested_place,))
linked = cur.fetchall()
self.replace_main_image(linked, image_id, cur, conn)
cur.execute(delete_media_links_image, (image_id,))
conn.commit()
cur.execute(delete_media, (image_id,))
conn.commit()
file = name.rsplit(".", 1)[0]
bare_path = path_string.rsplit("/", 1)[0]
resized = "{}/thumbnails/{}_resized_top_pic.png".format(bare_path, file)
if path.isfile(path_string):
remove(path_string)
if path.isfile(resized):
remove(resized)
elif kind == "source":
cur.execute(select_media_links_main_image_source, (current_source,))
linked = cur.fetchall()
self.replace_main_image(linked, image_id, cur, conn)
cur.execute(delete_media_links_image, (image_id,))
conn.commit()
cur.execute(delete_media, (image_id,))
conn.commit()
file = name.rsplit(".", 1)[0]
bare_path = path_string.rsplit("/", 1)[0]
resized = "{}/thumbnails/{}_resized_top_pic.png".format(bare_path, file)
if path.isfile(path_string):
remove(path_string)
if path.isfile(resized):
remove(resized)
self.reset_gui(kind, cur)
cur.execute("DETACH tree")
cur.close()
conn.close()
def replace_main_image(self, linked, image_id, cur, conn):
if len(linked) > 1:
main_image = linked[0][1]
if main_image == image_id:
new_main_image_link = linked[1][0]
cur.execute(update_media_links_main_image, (new_main_image_link,))
conn.commit()
def error_no_type(self):
def err_done1(combo, msg):
msg1[0].grab_release()
msg1[0].destroy()
combo.entry.focus_set()
msg1 = open_message(
self,
images_msg[0],
"Image Type Missing",
"OK")
msg1[0].grab_set()
msg1[2].config(
command=lambda combo=self.image_type, msg=msg1: err_done1(
combo, msg))
def error_no_radio(self):
def err_done2(rad1, msg):
msg2[0].grab_release()
msg2[0].destroy()
rad1.focus_set()
rad1 = self.radframe_children[0]
msg2 = open_message(
self,
images_msg[1],
"Action Type Missing",
"OK")
msg2[0].grab_set()
msg2[2].config(
command=lambda rad1=rad1, msg=msg2: err_done2(
rad1, msg))
def error_no_input(self, entry):
def err_done3(entry, msg):
msg3[0].grab_release()
msg3[0].destroy()
entry.focus_set()
msg3 = open_message(
self,
images_msg[2],
"Input Missing",
"OK")
msg3[0].grab_set()
msg3[2].config(
command=lambda entry=entry, msg=msg3: err_done3(
entry, msg))
def reset_gui(self, kind, cur):
self.link_input.delete(0, "end")
self.image_type.entry.delete(0, "end")
self.linkvar.set(99)
self.user_input.delete(0, "end")
# self.usepicvar.set(1)
(tree, current_dir, current_person_id, current_nested_place,
current_source) = get_current_elements(self.family_tree_id)
if kind in self.types[0:3]:
self.main.show_top_pic(cur)
elif kind == self.types[3]:
self.main.show_top_pic_nested_place(cur)
elif kind == self.types[4]:
self.main.show_top_pic_source(cur)
def redraw_default_image(self, kind, file, cur=None, conn=None):
if len(file) == 0:
cur.execute(select_image_setting_builtin_default_images)
(male_image, female_image, unisex_image, nested_place_image, source_image) = cur.fetchone()
else:
img_stg = file
if kind in self.types[0:3]:
button = self.main.top_pic_button
if len(file) == 0:
if kind == "male":
img_stg = male_image
elif kind == "female":
img_stg = female_image
elif kind == "unisex":
img_stg = unisex_image
elif kind == "place":
button = self.main.top_pic_place_button
if len(file) == 0:
img_stg = nested_place_image
elif kind == "source":
button = self.main.top_pic_source_button
if len(file) == 0:
img_stg = source_image
new_stg = '{}treebard/data/settings/images/{}'.format(
current_drive, img_stg)
top = Image.open(new_stg)
top = self.main.resize_top_pic(top, new_stg)
img1 = ImageTk.PhotoImage(top, master=self.master)
button.config(image=img1)
button.image = img1