소스 검색

Better, closing. (#179)

ÁÑÑÍHÌLÅTØR SPÄRK 4 년 전
부모
커밋
d591f522dc
1개의 변경된 파일과 5개의 추가작업 그리고 8개의 파일을 삭제
  1. 5 8
      wbb/__main__.py

+ 5 - 8
wbb/__main__.py

@@ -24,8 +24,9 @@ SOFTWARE.
 import asyncio
 import importlib
 import re
+from contextlib import closing, suppress
 
-import uvloop
+from uvloop import install
 from pyrogram import filters, idle
 from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup
 
@@ -373,12 +374,8 @@ General command are:
 
 
 if __name__ == "__main__":
-    uvloop.install()
-    try:
-        try:
+    install()
+    with closing(loop):
+        with suppress(asyncio.exceptions.CancelledError):
             loop.run_until_complete(start_bot())
-        except asyncio.exceptions.CancelledError:
-            pass
         loop.run_until_complete(asyncio.sleep(3.0))  # task cancel wait
-    finally:
-        loop.close()