From 8c5d1e2d790a2a2d73150fe898ebf98fec398971 Mon Sep 17 00:00:00 2001
From: BasioMeusPuga <disgruntled.mob@gmail.com>
Date: Fri, 17 Nov 2017 01:07:44 +0530
Subject: [PATCH] Streamline sorter

---
 sorter.py | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/sorter.py b/sorter.py
index f0a2c0e..e289db7 100644
--- a/sorter.py
+++ b/sorter.py
@@ -92,16 +92,14 @@ class BookSorter:
             return
 
         # ___________SORTING TAKES PLACE HERE___________
-        try:
-            file_extension = os.path.splitext(filename)[1][1:]
-            if file_extension == 'epub':
-                book_ref = ParseEPUB(filename, self.temp_dir, file_md5)
-            if file_extension == 'cbz':
-                book_ref = ParseCBZ(filename, self.temp_dir, file_md5)
-            if file_extension == 'cbr':
-                book_ref = ParseCBR(filename, self.temp_dir, file_md5)
-        except IndexError:
-            return
+        sorter = {
+            'epub': ParseEPUB,
+            'cbz': ParseCBZ,
+            'cbr': ParseCBR
+        }
+
+        file_extension = os.path.splitext(filename)[1][1:]
+        book_ref = sorter[file_extension](filename, self.temp_dir, file_md5)
 
         # Everything following this is standard
         # None values are accounted for here