semiworking program, still very alpha

1 files changed, 9 insertions, 4 deletions

diff --git a/src/pyssg/discovery.py b/src/pyssg/discovery.py
index 39c8bb1..606c1d0 100644
--- a/src/pyssg/discovery.py
+++ b/src/pyssg/discovery.py
@@ -11,7 +11,7 @@ def get_file_list(extensions: list[str], exclude: list[str]=None) -> list[str]:
 
         for f in files:
             if f.endswith(tuple(extensions)):
-                out.append(os.path.join(root, f))
+                out.append(os.path.join(root, f).replace(cwd, '')[1:])
 
     return out
 
@@ -27,14 +27,19 @@ def get_dir_structure(exclude: list[str]=None) -> list[str]:
         for d in dirs:
             if root in out:
                 out.remove(root)
-            out.append(os.path.join(root, d).replace(cwd, ''))
+            out.append(os.path.join(root, d))
+
+    return [o.replace(cwd, '')[1:] for o in out]
 
-    return out
 
+def get_all_files(src: str) -> tuple[list[str], list[str], list[str]]:
+    iwd = os.getcwd()
+    os.chdir(src)
 
-def get_all_files():
     md_files = get_file_list(['.md', '.markdown'], ['templates'])
     html_files = get_file_list(['.html'], ['templates'])
     dirs = get_dir_structure(['templates'])
 
+    os.chdir(iwd)
+
     return (dirs, md_files, html_files)