From 5ee10b0d38d0b343a3026b228d3821ecbf9870b4 Mon Sep 17 00:00:00 2001
From: Fabio Manganiello <info@fabiomanganiello.com>
Date: Tue, 14 Jun 2022 00:44:01 +0200
Subject: [PATCH] If img_dir is not found under content_dir then treat
 content_dir itself as img_dir

---
 madblog/app.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/madblog/app.py b/madblog/app.py
index 95f467c..0a88f49 100644
--- a/madblog/app.py
+++ b/madblog/app.py
@@ -24,11 +24,13 @@ class BlogApp(Flask):
         if not os.path.isdir(self.pages_dir):
             # If the `markdown` subfolder does not exist, then the whole
             # `config.content_dir` is treated as the root for markdown files.
-            self.pages_dir = os.path.join(config.content_dir)
+            self.pages_dir = config.content_dir
 
         img_dir = os.path.join(config.content_dir, 'img')
         if os.path.isdir(img_dir):
             self.img_dir = os.path.abspath(img_dir)
+        else:
+            self.img_dir = config.content_dir
 
         css_dir = os.path.join(config.content_dir, 'css')
         if os.path.isdir(css_dir):