fix: issue NanmiCoder#12

SkipGip · Jul 3, 2023 · e5f4ecd · e5f4ecd
1 parent 0cedd7d
commit e5f4ecd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/media_platform/douyin/login.py b/media_platform/douyin/login.py
@@ -1,6 +1,7 @@
 import sys
 import logging
 import asyncio
+import functools
 
 import aioredis
 from tenacity import (
@@ -104,6 +105,9 @@ async def login_by_qrcode(self):
             sys.exit()
 
         # show login qrcode
+            # utils.show_qrcode(base64_qrcode_img)
+        partial_show_qrcode = functools.partial(utils.show_qrcode, base64_qrcode_img)
+        asyncio.get_running_loop().run_in_executor(executor=None, func=partial_show_qrcode)
         utils.show_qrcode(base64_qrcode_img)
         await asyncio.sleep(2)
 

diff --git a/media_platform/xhs/login.py b/media_platform/xhs/login.py
@@ -1,6 +1,7 @@
 import sys
 import asyncio
 import logging
+import functools
 
 import aioredis
 from tenacity import (
@@ -146,7 +147,10 @@ async def login_by_qrcode(self):
         no_logged_in_session = cookie_dict.get("web_session")
 
         # show login qrcode
-        utils.show_qrcode(base64_qrcode_img)
+        # utils.show_qrcode(base64_qrcode_img)
+        partial_show_qrcode = functools.partial(utils.show_qrcode, base64_qrcode_img)
+        asyncio.get_running_loop().run_in_executor(executor=None, func=partial_show_qrcode)
+
         logging.info(f"waiting for scan code login, remaining time is 20s")
         login_flag: bool = await self.check_login_state(no_logged_in_session)
         if not login_flag: