Initial Commit

Aeonss · web-flow · commit ce847be61877 · 2022-09-09T17:45:03.000-04:00
diff --git a/WebtoonReader.py b/WebtoonReader.py
@@ -0,0 +1,283 @@
+# Makes sure modules are installed
+try:
+    import tkinter as tk 
+    from tkinter import filedialog
+    from PIL import ImageTk, Image
+except ModuleNotFoundError:
+    print("Downloading packages...")
+    os.system('pip install tkinter')
+    os.system('pip install pillow')
+    print("Please run the program again.")
+    
+
+import os, json, re
+from CustomScroller import ImageScroller
+
+SETTINGS_FILE = os.path.join(os.getcwd(), "webtoonreader_settings.json")
+
+# To do
+# Add bookmarks of where you left off on a manga
+
+class WebtoonReader:
+    def __init__(self):
+        # Create settings file in the home directory if it does not exist
+        if not os.path.isfile(SETTINGS_FILE):
+            with open(SETTINGS_FILE, "a") as f:
+                json.dump({"library" : os.getcwd(), "width" : 720, "height" : 800, "scroll_speed" : 3, "recent_chapter" : "", "recent_chapter_index" : ""}, f)
+                f.close()
+        
+
+        # Window
+        self.window = tk.Tk()
+        self.window.resizable(False, False)
+        
+        self.width = get_json('width')
+        self.height = get_json('height')
+        self.scroll_speed = get_json('scroll_speed')
+        
+        
+        # Center Window
+        x = int(self.window.winfo_screenwidth()/2 - self.width/2)
+        y = int(self.window.winfo_screenheight()/2 - self.height/2)
+        self.window.geometry("+{}+{}".format(x, y))
+        
+        # ImageScroller
+        chapter_path = get_json('recent_chapter')
+        self.frame = ImageScroller(self.window, path=chapter_path, scrollbarwidth=15, width=self.width, height=self.height, speed=self.scroll_speed)
+        self.frame.pack()
+        manga = os.path.basename(os.path.dirname(chapter_path))
+        self.window.title("[WebtoonReader] - " + manga + ": " + os.path.basename(chapter_path))
+        
+        # Menubar
+        menubar = tk.Menu(self.window)
+        settingsmenu = tk.Menu(menubar, tearoff=0) 
+        settingsmenu.add_command(label="Load Library", command=self.get_library)
+        settingsmenu.add_command(label="Load Manga", command=self.load_manga)
+        settingsmenu.add_command(label="Load Chapter", command=lambda: self.create_chapter(None))
+        settingsmenu.add_command(label="Settings", command=self.set_settings)
+        settingsmenu.add_command(label="Help")
+
+        menubar.add_cascade(label="Settings", menu=settingsmenu)
+        menubar.add_cascade(label="Previous Chapter", command=self.prev_chapter)
+        menubar.add_cascade(label="Next Chapter", command=self.next_chapter)
+        
+        # Keybind shortcuts
+        self.window.bind("<Left>", self.key_prev_chapter)
+        self.window.bind("<a>", self.key_prev_chapter)
+        self.window.bind("<Right>", self.key_next_chapter)
+        self.window.bind("<d>", self.key_next_chapter)
+
+
+        # Start the window
+        self.window.config(menu=menubar)
+        self.window.mainloop()
+
+
+    # Loads the directory containing all mangas
+    def get_library(self):
+        library_path = tk.filedialog.askdirectory()
+        
+        if library_path != "":
+            update_json('library', library_path)
+            
+    
+    # Loads a manga from where the user last left off
+    def load_manga(self):
+        manga_path = tk.filedialog.askdirectory()
+        
+        if manga_path == "":
+            return
+        
+        chapter_path = get_json(os.path.basename(manga_path))
+        self.create_chapter(chapter_path)
+
+
+    # Loads a chapter to read
+    def create_chapter(self, path):
+        if path != None:
+            chapter_path = path
+        else:
+            chapter_path = tk.filedialog.askdirectory()
+            if chapter_path == "":
+                return
+        
+        
+        manga_path = os.path.dirname(chapter_path)
+        chapter_list = abslistdir(manga_path)
+        chapter_index = -1
+        
+        # Finds the index of the current manga in the directory
+        for index, chapter in enumerate(chapter_list):
+            if chapter == chapter_path:
+                chapter_index = index
+                break
+        
+        # Updates the image scroller
+        self.frame.destroy()
+        self.frame = ImageScroller(self.window, path=chapter_path, scrollbarwidth=15, width=self.width, height=self.height, speed=self.scroll_speed)
+        self.frame.pack()
+        
+        # Updates settings json
+        update_json('recent_chapter', chapter_path)
+        update_json('recent_chapter_index', chapter_index)
+        manga = os.path.basename(manga_path)
+        update_json(manga, chapter_path)
+        self.window.title("[WebtoonReader] - " + manga + ": " + os.path.basename(chapter_path))
+
+
+    # Finds the next chapter of the manga
+    def next_chapter(self):
+        chapter_index = get_json('recent_chapter_index')
+        chapter_index += 1
+        
+        # Checks if the current chapter is the last chapter
+        chapter_path = get_json('recent_chapter')
+        manga_path = os.path.dirname(chapter_path)
+        chapter_list = abslistdir(manga_path)
+        
+        # Checks if it is the last chapter
+        if chapter_index >= len(chapter_list):
+            tk.messagebox.showwarning('Warning', 'Last Chapter')
+            return
+        
+        chapter_path = chapter_list[chapter_index]
+        
+        # Updates the image scroller
+        self.frame.destroy()
+        self.frame = ImageScroller(self.window, path=chapter_path, scrollbarwidth=15, width=self.width, height=self.height, speed=self.scroll_speed)
+        self.frame.pack()
+        
+        # Updates the settings json
+        update_json('recent_chapter', chapter_path)
+        update_json('recent_chapter_index', chapter_index)
+        manga = os.path.basename(manga_path)
+        update_json(manga, chapter_path)
+        self.window.title("[WebtoonReader] - " + manga + ": " + os.path.basename(chapter_path))
+    
+    
+    # Finds the previous chapter of the manga
+    def prev_chapter(self):
+        chapter_index = get_json('recent_chapter_index')
+        chapter_index -= 1
+        
+        # Checks if the current chapter is the first
+        chapter_path = get_json('recent_chapter')
+        manga_path = os.path.dirname(chapter_path)
+        chapter_list = abslistdir(manga_path)
+        
+        # Checks if it is the first chapter
+        if chapter_index == -1:
+            tk.messagebox.showwarning('Warning', 'First Chapter')
+            return
+        
+        chapter_path = chapter_list[chapter_index]
+        
+        # Updates the image scroller
+        self.frame.destroy()
+        self.frame = ImageScroller(self.window, path=chapter_path, scrollbarwidth=15, width=self.width, height=self.height, speed=self.scroll_speed)
+        self.frame.pack()
+        
+        # Updates the settings json
+        update_json('recent_chapter', chapter_path)
+        update_json('recent_chapter_index', chapter_index)
+        manga = os.path.basename(manga_path)
+        update_json(manga, chapter_path)
+        self.window.title("[WebtoonReader] - " + manga + ": " + os.path.basename(chapter_path))
+        
+        
+    # Keybind shortcut for next chapter
+    def key_next_chapter(self, e):
+        self.next_chapter()
+    
+    
+    # Keybind shortcut for previous chapter
+    def key_prev_chapter(self, e):
+        self.prev_chapter()
+
+
+    # Settings window
+    def set_settings(self):
+        settings = tk.Toplevel()
+        x = int(self.window.winfo_screenwidth()/2 - 200)
+        y = int(self.window.winfo_screenheight()/2 - 200)
+        settings.geometry("%dx%d+%d+%d" % (400, 400, x, y))
+        
+        width_label = tk.Label(settings, text="Width").pack(pady=10)
+        self.width_slider = tk.Scale(settings, from_=200, to=self.window.winfo_screenwidth(), orient=tk.HORIZONTAL, length=300, resolution=20)
+        self.width_slider.set(self.width)
+        self.width_slider.pack(pady=10)
+        self.width_slider.bind("<ButtonRelease-1>", self.update_width)
+        
+        height_label = tk.Label(settings, text="Height").pack(pady=10)
+        self.height_slider = tk.Scale(settings, from_=200, to=self.window.winfo_screenheight(), orient=tk.HORIZONTAL, length=250, resolution=20)
+        self.height_slider.set(self.height)
+        self.height_slider.pack(pady=10)
+        self.height_slider.bind("<ButtonRelease-1>", self.update_height)
+        
+        speed_label = tk.Label(settings, text="Scroll Speed").pack(pady=10)
+        self.scroll_speed_slider = tk.Scale(settings, from_=1, to=5, orient=tk.HORIZONTAL)
+        self.scroll_speed_slider.set(self.scroll_speed)
+        self.scroll_speed_slider.pack(pady=10)
+        self.scroll_speed_slider.bind("<ButtonRelease-1>", self.update_speed)
+
+        settings.mainloop()
+        
+    # Updates width in settings json
+    def update_width(self, e):
+        self.width = self.width_slider.get()
+        update_json('width', self.width_slider.get())
+    
+    # Updates height in settings json
+    def update_height(self, e):
+        self.height = self.height_slider.get()
+        update_json('height', self.height_slider.get())
+        
+    # Updates scroll speed in settings json
+    def update_speed(self, e):
+        self.scroll_speed = self.scroll_speed_slider.get()
+        update_json('scroll_speed', self.scroll_speed_slider.get())
+
+
+# Update the json value if key exists, otherwise adds to json
+def update_json(key, value):
+    json_file = open(SETTINGS_FILE, "r")
+    data = json.load(json_file)
+    data[key] = value
+    
+    json_file = open(SETTINGS_FILE, "w")
+    json_file.write(json.dumps(data))
+    json_file.close()
+    
+
+# Get json value (with key)
+def get_json(key):
+    json_file = open(SETTINGS_FILE, "r")
+    data = json.load(json_file)
+    for item in data:
+        if key in item:
+            value = data[key]
+            json_file.close()
+            return value
+    json_file.close()
+    return None
+
+
+# Returns a natural sorted list of absolute paths directories
+def abslistdir(path):
+    list = []
+    for root, dirs, files in os.walk(path):
+        for dir in dirs:
+            list.append(os.path.join(root, dir).replace("\\", "/"))
+    list.sort(key=natural_sort_key)
+    return list
+
+
+# Natural sort a list
+def natural_sort_key(list):
+    return [int(text) if text.isdigit() else text.lower()
+        for text in re.split(re.compile('([0-9]+)'), list)]  
+
+
+# Main
+if __name__ == "__main__":
+    WebtoonReader()
