Browse Source

[extractor/common] Interactive TFA code input

Sergey M․ 10 năm trước cách đây
mục cha
commit
e64b756943
1 tập tin đã thay đổi với 3 bổ sung2 xóa
  1. 3 2
      youtube_dl/extractor/common.py

+ 3 - 2
youtube_dl/extractor/common.py

@@ -15,6 +15,7 @@ import xml.etree.ElementTree
 from ..compat import (
 from ..compat import (
     compat_cookiejar,
     compat_cookiejar,
     compat_cookies,
     compat_cookies,
+    compat_getpass,
     compat_HTTPError,
     compat_HTTPError,
     compat_http_client,
     compat_http_client,
     compat_urllib_error,
     compat_urllib_error,
@@ -610,7 +611,7 @@ class InfoExtractor(object):
 
 
         return (username, password)
         return (username, password)
 
 
-    def _get_tfa_info(self):
+    def _get_tfa_info(self, note='two-factor verification code'):
         """
         """
         Get the two-factor authentication info
         Get the two-factor authentication info
         TODO - asking the user will be required for sms/phone verify
         TODO - asking the user will be required for sms/phone verify
@@ -624,7 +625,7 @@ class InfoExtractor(object):
         if downloader_params.get('twofactor', None) is not None:
         if downloader_params.get('twofactor', None) is not None:
             return downloader_params['twofactor']
             return downloader_params['twofactor']
 
 
-        return None
+        return compat_getpass('Type %s and press [Return]: ' % note)
 
 
     # Helper functions for extracting OpenGraph info
     # Helper functions for extracting OpenGraph info
     @staticmethod
     @staticmethod