gui.py

import tkinter as tk
from tkinter import ttk 
from PIL import ImageTk, Image
import FactorioCalc
import IconExtractor
import io
from zipfile import ZipFile
import re
import json 

def setup(modfp):
    global micons 
    global bicons
    global modlist
    micons = IconExtractor.GetmodIcons(modfp)
    bicons = IconExtractor.GetBaseIcons(basefp)
    modlistjson = AssMachines = json.load(open(fr"{modfp}\mod-list.json"))
    modlist = []
    for i in modlistjson["mods"]:
        if i["enabled"] == True:
            modlist.append(i["name"])

def show_image(tar):
    imagefile = findicon(tar)
    image = ImageTk.PhotoImage(data=io.BytesIO(imagefile).read())
    parent.config(image=image)
    parent.image = image

def findicon(tar):
    if "space-exploration" in modlist:
        if tar == "se-rocket-science-pack":
            tar = "orange"
        if tar == "se-iron-ingot-to-plate":
            tar = "iron-plate" 
        if tar.startswith("se-"):
            tar = tar.replace("se-","")
    if "aai-industry" in modlist:
        if tar == "motor":
            tar = "single-cylinder-engine"
        if tar == "electric-motor":
            tar = "small-electric-motor"
        if tar == "electronic-circuit-stone":
            tar = "electronic-circuit"
        if tar == "engine-unit":
            tar = "multi-cylinder-engine"
    print(f"Getting icon: {tar}")
    for item in bicons:
        if item == (f"{tar}.png"):
            with open(f"{bicons[item]}", "rb") as image:
                img = Image.open(f"{bicons[item]}")
                image = img.crop((0,0,64,64))
                image.save("test.png")
                with open("test.png", "rb") as image:
                    f = image.read()
                    print(f"icon found:{bicons[item]}")
                    b = bytearray(f)
                return b
    for mod in micons:
        for item in micons[mod]:
            if item.filename.endswith(f"/{tar}.png"):
                with ZipFile(mod) as myzip:
                    with myzip.open(item) as myfile:
                        print(f"icon found:{item}")
                        image = myfile.read()
                        return image
    print(f"Icon not found: {tar}")
    with open(f"placeholder.png", "rb") as image:
                img = Image.open(f"placeholder.png")
                image = img.crop((0,0,64,64))
                image.save("test.png")
                with open("test.png", "rb") as image:
                    f = image.read()
                    b = bytearray(f)
                    return b

def TreePop(item):
    global ids
    ipms = ipm.get()
    tree.delete(*tree.get_children())
    ids = []
    global storedimgs
    recipes = FactorioCalc.GetRecipes(filepath)
    for name in recipes:
        if item in name:
            if name.startswith("se-"):
                item = name
                break
        if item == name:
            item = name
            break
    print(recipes[item]['ingredients'])
    jsn = FactorioCalc.HumanFriendly(item)
    imagefile = findicon(item)
    image =ImageTk.PhotoImage(data=io.BytesIO(imagefile).read())
    storedimgs.append(image)
    ratio = float(ipms)/float(recipes[item]['products'][0]['amount'])
    if recipes[item]["category"] == "smelting":
        tree.insert('', tk.END,jsn, open=False,image= image, values=(f"{jsn}",ipms,0,FactorioCalc.NumOfSmel(item,filepath,ipms,FactorioCalc.JsonFriendly(furnaceDD.get()))))
    else:
        tree.insert('', tk.END,jsn, open=False,image= image, values=(f"{jsn}",ipms,FactorioCalc.NumOfAss(item,filepath,ipms,FactorioCalc.JsonFriendly(AssemblersDD.get()))))
    items = FactorioCalc.GetSubIngredient(item,recipes)
    print(f"ITEMS: {items}")
    for ingr in items:
        imagefile = findicon(ingr['name'])
        image =ImageTk.PhotoImage(data=io.BytesIO(imagefile).read())
        storedimgs.append(image)
        idn = f"{ingr['name']}0"
        ids.append(idn)
        print(ingr)
        tree.insert(f'{jsn}', tk.END,idn, open=False,image= image, values=(f"{ingr['name']}",ingr['amount']*ratio,FactorioCalc.NumOfAss(ingr['name'],filepath,ingr['amount']*ratio,FactorioCalc.JsonFriendly(AssemblersDD.get()))))
        if "space-exploration" in modlist:
            if ingr['name'] == "electronic-circuit":
                subitems = FactorioCalc.GetSubIngredient('electronic-circuit-stone',recipes)
            else:
                subitems = FactorioCalc.GetSubIngredient(ingr['name'],recipes)
        else:
            subitems = FactorioCalc.GetSubIngredient(ingr['name'],recipes)
        print(f"SUBITEM: {subitems}")
        subingri(subitems,idn,ratio)

def subingri(subitems,parent,ipm):
    recipes = FactorioCalc.GetRecipes(filepath)
    ipms = ipm
    for subingr in subitems:
        imagefile = findicon(subingr['name'])
        image =ImageTk.PhotoImage(data=io.BytesIO(imagefile).read())
        storedimgs.append(image)
        high = 0
        for idno in ids:
            x = re.search(f"{subingr['name']}",idno)
            if x:
                if int(idno[-1]) > high:
                    high = int(idno[-1])
            idn = f"{subingr['name']}{high+1}"
        ids.append(idn)
        if subingr['name'].endswith("ore") or subingr['name'] == "stone" or subingr['name'] == "coal":
            tree.insert(parent, tk.END,idn, open=False,image= image, values=(f"{subingr['name']}",subingr['amount']*ipm,FactorioCalc.NumOfAss(subingr['name'],filepath,ipms,FactorioCalc.JsonFriendly(AssemblersDD.get()))))
        else:
            if recipes[subingr['name']]["category"] == "smelting" or recipes[subingr['name']]["category"] == "kiln":
                tree.insert(parent, tk.END,idn, open=False,image= image, values=(f"{subingr['name']}",subingr['amount']*ipm,0,FactorioCalc.NumOfSmel(subingr['name'],filepath,ipm,FactorioCalc.JsonFriendly(furnaceDD.get()))))
            else:
                tree.insert(parent, tk.END,idn, open=False,image= image, values=(f"{subingr['name']}",subingr['amount']*ipm,FactorioCalc.NumOfAss(subingr['name'],filepath,subingr['amount']*ipm,FactorioCalc.JsonFriendly(AssemblersDD.get())),0))
        sub = FactorioCalc.GetSubIngredient(subingr['name'],recipes)
        if sub != None:
            if "space-exploration" in modlist:
                if subingr['name'] == "electronic-circuit":
                    sub = FactorioCalc.GetSubIngredient('electronic-circuit-stone',recipes)
                else:
                    sub = FactorioCalc.GetSubIngredient(subingr['name'],recipes)
            else:
                sub = FactorioCalc.GetSubIngredient(subingr['name'],recipes)
            subingri(sub,idn,ipm)

filepath = r"C:\Users\rudyb\AppData\Roaming\Factorio\script-output\recipe-lister"
modfp= r"C:\Users\rudyb\AppData\Roaming\Factorio\mods"
basefp = r"C:\Program Files (x86)\Steam\steamapps\common\Factorio\data\base\graphics\icons"
setup(modfp)

root = tk.Tk()
root.title("Demo")
root.geometry("1000x900")

frame = tk.Frame(root)
frame.pack()

frame3 = tk.Frame(root)
frame3.pack()

button = tk.Button(frame, text="QUIT", fg="red", command=quit)
button.pack(side=tk.LEFT)

#dropdown menu
recipes = FactorioCalc.GetRecipes(filepath)
temp = []
storedimgs = []
for item in recipes:
    temp.append(FactorioCalc.HumanFriendly(item))
recipes = temp
recipesDD = ttk.Combobox(frame3,values=recipes, state='normal')
Assemblers = FactorioCalc.GetAssMachines(filepath)
temp = []
for item in Assemblers:
    temp.append(FactorioCalc.HumanFriendly(item))
Assemblers = temp
AssemblersDD = ttk.Combobox(frame3,values=Assemblers, state='normal')
AssemblersDD.set("assembling machine 1")
furnace = FactorioCalc.GetFurnace(filepath)
temp = []
for item in furnace:
    temp.append(FactorioCalc.HumanFriendly(item))
furnace = temp
furnaceDD = ttk.Combobox(frame3, values=furnace, state='normal')
furnaceDD.set("stone furnace")
recipesDD.pack(side=tk.LEFT)
AssemblersDD.pack(side=tk.LEFT)
furnaceDD.pack(side=tk.LEFT)

#Buttons
image = tk.Button(frame, text="image", command=lambda: show_image(FactorioCalc.JsonFriendly(recipesDD.get())))
image.pack(side=tk.LEFT)
calculater = tk.Button(frame, text="calculate", command=lambda: show_image(FactorioCalc.JsonFriendly(recipesDD.get())))
calculater.pack(side=tk.LEFT)
treeB = tk.Button(frame,text="TreeB", command= lambda:TreePop(FactorioCalc.JsonFriendly(recipesDD.get())))
treeB.pack(side=tk.LEFT)
#IPM input
entry_float = tk.DoubleVar()
entry_float.set(60.0)
ipm = ttk.Entry(frame3,textvariable=entry_float)
ipm.pack(side=tk.LEFT)

# label to show the image
frame2 = tk.Frame(root)
frame2.pack()
parent = tk.Label(frame2)
parent.pack()
#treeview
style = ttk.Style()
style.configure('Treeview',rowheight=64)
coloumns = ('#0','#1','#2','#3')
tree = ttk.Treeview(frame2,columns=coloumns, height=900)
tree.heading('#0', text='icon', anchor=tk.W)
tree.heading('#1', text='Item', anchor=tk.W)
tree.heading('#2', text='Ratios', anchor=tk.W)
tree.heading('#3', text='Assemblers', anchor=tk.W)
tree.heading('#4', text='Smelters', anchor=tk.W)
tree.pack()

root.mainloop()