Merge pull request #149 from dfrommi/swagger-yaml

support swagger spec in yaml format

Author: eliangcs

http_prompt/cli.py

@@ -1,6 +1,7 @@
 from __future__ import unicode_literals
 
 import json
+import yaml
 import os
 import re
 import sys
@@ -113,9 +114,12 @@ def cli(spec, env, url, http_options):
             try:
                 spec = json.loads(content)
             except json.JSONDecodeError:
-                click.secho("Warning: Specification file '%s' is not JSON" %
-                            spec, err=True, fg='red')
-                spec = None
+                try:
+                    spec = yaml.load(content)
+                except yaml.YAMLError:
+                    click.secho("Warning: Specification file '%s' is neither valid JSON nor YAML" %
+                                spec, err=True, fg='red')
+                    spec = None
         finally:
             f.close()
 

requirements.txt

@@ -4,3 +4,4 @@ parsimonious>=0.6.2
 prompt-toolkit>=1.0.0,<2.0.0
 Pygments>=2.1.0
 six>=1.10.0
+PyYAML>=3.0