Add PyYAML warning in interactive_console.py (fix #167)

tumblr · Nov 15, 2023 · 2aa5c5d · 2aa5c5d
1 parent c5dbeaa
commit 2aa5c5d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/interactive_console.py b/interactive_console.py
@@ -5,10 +5,15 @@
 from builtins import input
 
 import pytumblr
-import yaml
 import os
 import code
 from requests_oauthlib import OAuth1Session
+try:
+    import yaml
+except ImportError:
+    print('You need PyYAML to run interactive console\npip install pyyaml')
+    import sys
+    sys.exit(-1)
 
 
 def new_oauth(yaml_path):