Added Main Tinder Functionality

Author: zachsamuels

Pipfile

@@ -8,6 +8,9 @@ flake8 = "*"
 pytest = "*"
 
 [packages]
+pillow = "*"
+pygame = "*"
+aiohttp = "*"
 
 [requires]
 python_version = "3.7"

Pipfile.lock

@@ -0,0 +1,4 @@
+from .mainwindow import Tinder
+
+if __name__ == "__main__":
+    Tinder().start()

src/__main__.py

@@ -0,0 +1,103 @@
+import tkinter as tk
+from PIL import Image, ImageTk
+import io
+from random import randint
+import asyncio
+import time
+import aiohttp
+import os
+from pygame import mixer
+
+
+class Tinder:
+    def __init__(self):
+        mixer.init()
+        self.dir = os.path.dirname(os.path.realpath(__file__))
+        self.root = tk.Tk()
+        self.root.title("Cat Tinder")
+        self.root.geometry("400x500")
+        self.root.configure(background='grey')
+        self.root.protocol("WM_DELETE_WINDOW", self.on_closing)
+        #self.root.bind('<Motion>', self.motion)
+        self.loop = asyncio.get_event_loop()
+        self.session = None
+        self.images = list()
+        
+        
+    def start(self):
+        self.loop.run_until_complete(self.get_cache())
+        self.new_image()
+
+    async def get_cache(self):
+        print("refreshing image cache")
+        t = time.time()
+        if not self.session:
+            headers = {'x-api-key': 'd12e9702-f791-4d13-9ffd-0edeed9cecaf'}
+            self.session = aiohttp.ClientSession(headers=headers)
+        for _ in range(10):
+            if randint(0,10) == 9:
+                image_number = randint(1,10)
+                im = Image.open(os.path.join(self.dir, os.path.join("res", os.path.join("images", f"{image_number}.jpg"))))
+                im = im.resize((400, 400), Image.NEAREST)
+                image = ImageTk.PhotoImage(im)
+                self.images.append([image, True])
+            else:
+                async with self.session.get('https://api.thecatapi.com/v1/images/search') as res:
+                    data = await res.json()
+                    url = data[0]['url']
+                async with self.session.get(url) as res:
+                    image_bytes = await res.read()
+                    im = Image.open(io.BytesIO(image_bytes))
+                    im = im.resize((400, 400), Image.NEAREST)
+                    image = ImageTk.PhotoImage(im)
+                    self.images.append([image, False])
+        print(time.time()-t)
+
+
+    def all_children(self):
+        children = self.root.winfo_children()
+        for item in children:
+            if item.winfo_children():
+                children.extend(item.winfo_children())
+        return children
+
+
+    def new_image(self):
+        widget_list = self.all_children()
+        for item in widget_list:
+            item.pack_forget()
+        self.frame = tk.Frame(self.root, bg = "black")
+        try:
+            image, jumpscare = self.images.pop(0)
+        except IndexError:
+            self.loop.run_until_complete(self.get_cache())
+            image, jumpscare = self.images.pop(0)
+        label = tk.Label(self.frame, image=image)
+        label.pack(side = tk.TOP, fill = "both", expand = "yes")
+        like = tk.Button(self.frame, text="Like", background="green", command = self.new_image)
+        like.pack(side = tk.RIGHT)
+        dislike = tk.Button(self.frame, text="Dislike", background="red", command = self.new_image)
+        dislike.pack(side = tk.LEFT)
+        if jumpscare:
+            mixer.music.load(os.path.join(self.dir, os.path.join("res", os.path.join("sounds", "jumpscare.mp3"))))
+            mixer.music.play()
+        self.frame.pack()
+        self.root.mainloop()
+
+
+    def on_closing(self):
+        x = randint(0, 1520)
+        y = randint(0, 580)
+        self.root.geometry(f"+{x}+{y}")
+
+
+    def motion(self,event):
+        frame_x, frame_y = self.frame.winfo_x(), self.frame.winfo_y()
+        x, y = event.x, event.y
+        if x > 300 and y < 15:
+            move_x = (400 - x) + frame_x
+            move_y = (15 - y) - frame_y
+            self.root.geometry(f"400x500+{move_x}+{move_y}")
+
+if __name__ == "__main__":
+    Tinder().start()