Update main.py

wanghaisheng · Jul 26, 2024 · 521ad0f · 521ad0f
1 parent 9ac26fa
commit 521ad0f
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/main.py b/main.py
@@ -631,12 +631,14 @@ def run(env: str = "development", power: int = 16):
         template_ = booster.overload_tasks()
 
         # Replace project README file.
-        if env == "production":
-            with open(SERVER_PATH_README, "w", encoding="utf8") as f:
-                for i in template_:
-                    f.write(i)
-
-            shutil.copyfile(SERVER_PATH_README, os.path.join(SERVER_PATH_DOCS, "index.md"))
+        if render_style=='mkdocs':
+
+            if env == "production":
+                with open(SERVER_PATH_README, "w", encoding="utf8") as f:
+                    for i in template_:
+                        f.write(i)
+
+                shutil.copyfile(SERVER_PATH_README, os.path.join(SERVER_PATH_DOCS, "index.md"))
 
 if __name__ == "__main__":