about summary refs log tree commit diff
diff options
context:
space:
mode:
authorNakidai <plaza521@inbox.ru>2024-09-08 18:39:35 +0300
committerNakidai <plaza521@inbox.ru>2024-09-08 18:39:35 +0300
commitcc8a91ae054816e16ed22441ba12eb47728a1b83 (patch)
tree772bf5e476670bc446a10eac70114df1ca21553a
parent144b0f7e1ab21a9eaab40d43f3baf5f00a26f196 (diff)
downloadpetthecord-cc8a91ae054816e16ed22441ba12eb47728a1b83.tar.gz
petthecord-cc8a91ae054816e16ed22441ba12eb47728a1b83.zip
Add redirect on root
-rw-r--r--src/petthecord/server.py15
1 files changed, 12 insertions, 3 deletions
diff --git a/src/petthecord/server.py b/src/petthecord/server.py
index 8c3d8a2..35cbca7 100644
--- a/src/petthecord/server.py
+++ b/src/petthecord/server.py
@@ -1,6 +1,7 @@
 from io import BytesIO
+from typing import NoReturn
 
-from aiohttp.web import Application, StreamResponse, get, Request, Response
+from aiohttp.web import Application, StreamResponse, get, HTTPFound, Request, Response
 from discord import NotFound, HTTPException
 from discord.ext import commands
 from petpetgif import petpet
@@ -11,9 +12,17 @@ class Server(Application):
         self.client = client
         super().__init__()
 
-        self.add_routes([get("/{uid}", self.root)])
+        self.add_routes(
+            [
+                get("/{uid}", self.petpet),
+                get("/", self.root)
+            ]
+        )
 
-    async def root(self, request: Request) -> StreamResponse:
+    async def root(self, _: Request) -> NoReturn:
+        raise HTTPFound("https://github.com/nakidai/petthecord")
+
+    async def petpet(self, request: Request) -> StreamResponse:
         try:
             uid = int(request.match_info["uid"][:request.match_info["uid"].find('.')])
         except ValueError: