From 0cd6a136c0bba445cd1142e78dd27090d4c2fa5e Mon Sep 17 00:00:00 2001
From: Rayan fernandes <37145078+Rayanfer32@users.noreply.github.com>
Date: Sun, 6 Jun 2021 17:01:00 +0530
Subject: [PATCH] =?UTF-8?q?=E2=99=BB=EF=B8=8F=20Created=20common=20routes?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/routes.py | 29 +++++++++++------------------
 1 file changed, 11 insertions(+), 18 deletions(-)

diff --git a/app/routes.py b/app/routes.py
index 7ab9e2c..e0b028d 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -5,10 +5,8 @@ from telethon.tl.types import Channel, Chat, User
 
 from .config import index_settings
 
-
 log = logging.getLogger(__name__)
 
-
 async def setup_routes(app, handler):
     h = handler
     client = h.client
@@ -24,6 +22,15 @@ async def setup_routes(app, handler):
         web.post("/login", h.login_post, name="login_handle"),
         web.get("/logout", h.logout_get, name="logout"),
     ]
+    def get_common_routes(p):
+        return [
+                    web.get(p, h.index),
+                    web.get(p + r"/logo", h.logo),
+                    web.get(p + r"/{id:\d+}/view", h.info),
+                    web.get(p + r"/{id:\d+}/download", h.download_get),
+                    web.head(p + r"/{id:\d+}/download", h.download_head),
+                    web.get(p + r"/{id:\d+}/thumbnail", h.thumbnail_get),
+                ]
     if index_all:
         # print(await client.get_dialogs())
         # dialogs = await client.get_dialogs()
@@ -48,14 +55,7 @@ async def setup_routes(app, handler):
             alias_id = h.generate_alias_id(chat)
             p = "/{chat:" + alias_id + "}"
             routes.extend(
-                [
-                    web.get(p, h.index),
-                    web.get(p + r"/logo", h.logo),
-                    web.get(p + r"/{id:\d+}/view", h.info),
-                    web.get(p + r"/{id:\d+}/download", h.download_get),
-                    web.head(p + r"/{id:\d+}/download", h.download_head),
-                    web.get(p + r"/{id:\d+}/thumbnail", h.thumbnail_get),
-                ]
+               get_common_routes(p)
             )
             log.debug(f"Index added for {chat.id} at /{alias_id}")
 
@@ -65,14 +65,7 @@ async def setup_routes(app, handler):
             alias_id = h.generate_alias_id(chat)
             p = "/{chat:" + alias_id + "}"
             routes.extend(
-                [
-                    web.get(p, h.index),
-                    web.get(p + r"/logo", h.logo),
-                    web.get(p + r"/{id:\d+}/view", h.info),
-                    web.get(p + r"/{id:\d+}/download", h.download_get),
-                    web.head(p + r"/{id:\d+}/download", h.download_head),
-                    web.get(p + r"/{id:\d+}/thumbnail", h.thumbnail_get),
-                ]
+                get_common_routes(p) # returns list() of common routes 
             )
             log.debug(f"Index added for {chat.id} at /{alias_id}")
     routes.append(web.view(r"/{wildcard:.*}", h.wildcard))