set premium at first

tgbot-collection · Dec 22, 2023 · d0e2025 · d0e2025
1 parent 613f36b
commit d0e2025
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ytdlbot/premium.py b/ytdlbot/premium.py
@@ -42,6 +42,8 @@ async def hello(client: Client, message: types.Message):
     url = data["url"]
     user_id = data["user_id"]
 
+    redis = Redis()
+    redis.r.hset("premium", user_id, 1)
     tempdir = tempfile.TemporaryDirectory(prefix="ytdl-")
     output = pathlib.Path(tempdir.name, "%(title).70s.%(ext)s").as_posix()
     ydl_opts = {"restrictfilenames": False, "quiet": True, "outtmpl": output, "progress_hooks": [download_hook]}
@@ -96,8 +98,6 @@ async def hello(client: Client, message: types.Message):
         )
 
     tempdir.cleanup()
-    redis = Redis()
-    redis.r.hset("premium", user_id, 1)
 
 
 if __name__ == "__main__":