Use mkdocs.utils.load_config instead of ConfigData

Author: jimporter

mike/commands.py

@@ -22,7 +22,7 @@ def _redirect_template(user_template=None):
 
 
 def _add_redirect_to_commit(commit, template, src, dst,
-                            use_directory_urls=True):
+                            use_directory_urls):
     if os.path.splitext(src)[1] == '.html':
         reldst = os.path.relpath(dst, os.path.dirname(src))
         href = '/'.join(reldst.split(os.path.sep))
@@ -81,15 +81,15 @@ def deploy(cfg, version, title=None, aliases=[], update_aliases=False,
     with git_utils.Commit(branch, message) as commit:
         commit.delete_files([version_str] + list(info.aliases))
 
-        for f in git_utils.walk_real_files(cfg.site_dir):
-            canonical_file = f.copy(destdir, cfg.site_dir)
+        for f in git_utils.walk_real_files(cfg['site_dir']):
+            canonical_file = f.copy(destdir, cfg['site_dir'])
             commit.add_file(canonical_file)
             for d in alias_destdirs:
-                alias_file = f.copy(d, cfg.site_dir)
+                alias_file = f.copy(d, cfg['site_dir'])
                 if redirect:
                     _add_redirect_to_commit(
                         commit, t, alias_file.path, canonical_file.path,
-                        cfg.use_directory_urls
+                        cfg['use_directory_urls']
                     )
                 else:
                     commit.add_file(alias_file)
@@ -172,7 +172,7 @@ def alias(cfg, version, aliases, update_aliases=False, redirect=True,
                 if redirect:
                     _add_redirect_to_commit(
                         commit, t, alias_file.path, canonical_file.path,
-                        cfg.use_directory_urls
+                        cfg['use_directory_urls']
                     )
                 else:
                     commit.add_file(alias_file)

mike/driver.py

@@ -84,20 +84,20 @@ def add_git_arguments(parser, *, commit=True, prefix=True):
 
 def load_mkdocs_config(args, strict=False):
     try:
-        cfg = mkdocs_utils.ConfigData(args.config_file)
+        cfg = mkdocs_utils.load_config(args.config_file)
         if args.branch is None:
-            args.branch = cfg.remote_branch
+            args.branch = cfg['remote_branch']
         if args.remote is None:
-            args.remote = cfg.remote_name
+            args.remote = cfg['remote_name']
         return cfg
-    except OSError:
+    except FileNotFoundError as e:
         if strict:
-            raise RuntimeError('{!r} not found'.format(args.config_file))
+            raise
         if args.branch is None or args.remote is None:
-            raise RuntimeError((
-                '{!r} not found; pass --config-file or set ' +
-                '--remote/--branch explicitly'
-            ).format(args.config_file))
+            raise FileNotFoundError(
+                '{}; pass --config-file or set --remote/--branch explicitly'
+                .format(str(e))
+            )
 
 
 def check_remote_status(args, strict=False):

mike/mkdocs_plugin.py

@@ -9,7 +9,7 @@
 
 try:
     from mkdocs.exceptions import PluginError
-except ImportError:
+except ImportError:  # pragma: no cover
     PluginError = ValueError
 
 

mike/mkdocs_utils.py

@@ -1,22 +1,34 @@
+import mkdocs.config
 import os
 import re
 import subprocess
 import yaml
+from collections.abc import Iterable
 from contextlib import contextmanager
 from tempfile import NamedTemporaryFile
 
 docs_version_var = 'MIKE_DOCS_VERSION'
 
 
-class ConfigData:
-    def __init__(self, config_file):
-        with open(config_file) as f:
-            config = yaml.load(f, Loader=yaml.Loader)
-        self.site_dir = os.path.join(os.path.dirname(config_file),
-                                     config.get('site_dir', 'site'))
-        self.remote_name = config.get('remote_name', 'origin')
-        self.remote_branch = config.get('remote_branch', 'gh-pages')
-        self.use_directory_urls = config.get('use_directory_urls', True)
+def _open_config(config_file=None):
+    if config_file is None:
+        config_file = ['mkdocs.yml', 'mkdocs.yaml']
+    elif not isinstance(config_file, Iterable) or isinstance(config_file, str):
+        config_file = [config_file]
+
+    exc = None
+    for i in config_file:
+        try:
+            return open(i, 'rb')
+        except FileNotFoundError as e:
+            if not exc:
+                exc = e
+    raise exc
+
+
+def load_config(config_file=None, **kwargs):
+    with _open_config(config_file) as f:
+        return mkdocs.config.load_config(f, **kwargs)
 
 
 @contextmanager

test/data/basic_theme/mkdocs.yml

@@ -1,4 +1,5 @@
 site_name: test
+site_url: https://example.invalid/
 theme: mkdocs
 
 nav:

test/data/mkdocs_plugin/mkdocs.yml

@@ -1,5 +1,5 @@
 site_name: test
-site_url: https://example.com/
+site_url: https://example.invalid/
 theme:
   name: mkdocs
 plugins:

test/data/no_directory_urls/mkdocs.yml

@@ -1,4 +1,5 @@
 site_name: test
+site_url: https://example.invalid/
 theme: mkdocs
 use_directory_urls: false
 

test/data/remote/mkdocs.yml

@@ -1,4 +1,5 @@
 site_name: test
+site_url: https://example.invalid/
 theme: mkdocs
 remote_name: myremote
 remote_branch: mybranch

test/data/site_dir/mkdocs.yml

@@ -1,4 +1,5 @@
 site_name: test
+site_url: https://example.invalid/
 theme: mkdocs
 site_dir: built_docs
 

test/integration/test_serve.py

@@ -11,7 +11,7 @@
 
 @unittest.skipIf(platform.system() == 'Windows',
                  "SIGINT doesn't work on windows")
-class TestList(unittest.TestCase):
+class TestServe(unittest.TestCase):
     def setUp(self):
         self.stage = stage_dir('serve')
         git_init()