class Application(tk.Frame):
def __init__(self, master=None):
super().__init__(master)
self.config = json.load(open("../config.json"))
self.master = master
self.pack()
self.create_widgets()
self.last_info = ""
signal.signal(signal.SIGINT, self.signal_handler)
self.cache = []
def create_widgets(self):
self.master.title("Overworld Reader")
self.connect_button = tk.Button(self, text="Connect", fg="green", command=self.connect)
self.connect_button.grid(column=0,row=1)
self.displays = []
self.quit = tk.Button(self, text="Disconnect", fg="red", command=self.disconnect)
self.quit.grid(column=1,row=1)
def connect(self):
print("Connecting to: ", self.config["IP"])
self.SWSHReader = SWSHReader(self.config["IP"])
self.update()
def disconnect(self):
print("Disconnecting")
self.after_cancel(self.after_token)
self.SWSHReader.close(False)
self.SWSHReader = None
def signal_handler(self, signal, frame):
self.disconnect()
sys.exit(0)
def update(self):
self.SWSHReader.KCoordinates = KCoordinates(self.SWSHReader.readKCoordinatesBlock())
pkms = self.SWSHReader.KCoordinates.ReadOwPokemonFromBlock()
info = []
infocheck = []
for pkm in pkms:
info.append(pkm)
infocheck.append(pkm.toString())
if infocheck != self.last_info:
i = 0
for display in self.displays:
for widget in display:
widget.grid_forget()
self.displays = []
for pkm in info:
self.displays.append([])
self.displays[i].append(tk.Text(self,height=5))
self.displays[i][0].grid(column=2+(5 if i%2 else 0), row=2+int((i*3)/2), columnspan=1, rowspan=2)
self.displays[i].append(tk.Label(self))
self.displays[i][1].grid(column=0+(5 if i%2 else 0), row=2+int((i*3)/2), columnspan=2, rowspan=2)
self.displays[i].append(tk.Label(self))
self.displays[i][2].grid(column=3+(5 if i%2 else 0), row=2+int((i*3)/2), columnspan=2, rowspan=2)
s1 = pb.SpriteResource('pokemon', pkm.species, shiny=pkm.getShinyType((pkm.sid<<16) | pkm.tid, pkm.PID)).img_data
try:
s2 = urllib.request.urlopen(f"https://www.serebii.net/swordshield/ribbons/{pkm.Ribbons[pkm.mark].lower() if pkm.mark != 255 else ''}mark.png").read()
im2 = Image.open(io.BytesIO(s2))
image2 = ImageTk.PhotoImage(im2)
self.cache.append(image2)
self.displays[i][2].config(image=image2)
except Exception as e:
pass
im = Image.open(io.BytesIO(s1))
image = ImageTk.PhotoImage(im)
self.cache.append(image)
self.displays[i][1].config(image=image)
self.last_info = pkm.toString()
self.displays[i][0].delete(1.0, tk.END)
self.displays[i][0].insert(1.0, pkm.toString())
i += 1
self.last_info = infocheck
self.after_token = self.after(1000, self.update)
# Connect to Switch
config = json.load(open("../config.json"))
r = SWSHReader(config["IP"])
def signal_handler(signal, frame): #CTRL+C handler
print("Stop request")
r.close()
signal.signal(signal.SIGINT, signal_handler)
# Give the KCoords class access to the switch
r.KCoordinates.r = r
last_info = []
while True:
# Refresh KCoords block
r.KCoordinates = KCoordinates(r.readKCoordinatesBlock())
pkms = r.KCoordinates.ReadOwPokemonFromBlock()
info = []
for pkm in pkms:
info.append(pkm.toString())
if info != last_info:
last_info = info
for pkm in info:
print(pkm)
print("-------------------------------")
r.pause(0.3)
r.close()