Handle input dir/feed

1 files changed, 4 insertions, 3 deletions

diff --git a/main.py b/main.py
index 0622b5e..85293a0 100644
--- a/main.py
+++ b/main.py
@@ -36,7 +36,7 @@ def get_html(PathList):
             HtmlList.append(path)
     return HtmlList
 
-if not os.path.isdir("input"):
+if not os.path.isdir(dir):
     print("Nothing to do")
     exit(1)
 
@@ -52,8 +52,9 @@ if not os.path.isdir("output/markdown"):
 if not os.path.isdir("output/gemini"):
     os.makedirs("output/gemini")
 
-if os.path.isfile("input/feed.gmi"):
-    shutil.copy("input/feed.gmi", "output/gemini/feed.gmi")
+feedname = os.path.join(dir, "feed.gmi")
+if os.path.isfile(feedname):
+    shutil.copy(feedname, "output/gemini/feed.gmi")
 
 print(convert_html_to_md(HtmlList))
 print(convert_md_to_gemini())