create generic copy_js_files util

Author: Archmonger

setup.py

@@ -99,6 +99,18 @@
 # ----------------------------------------------------------------------------
 # Build Javascript
 # ----------------------------------------------------------------------------
+def copy_js_files(source_dir: Path, destination: Path) -> None:
+    if destination.exists():
+        shutil.rmtree(destination)
+    destination.mkdir()
+
+    for file in source_dir.iterdir():
+        if file.is_file():
+            shutil.copy(file, destination / file.name)
+        else:
+            copy_js_files(file, destination / file.name)
+
+
 def build_javascript_first(build_cls: type):
     class Command(build_cls):
         def run(self):
@@ -133,18 +145,12 @@ def run(self):
             log.info("Copying @pyscript/core distribution")
             pyscript_dist = js_dir / "node_modules" / "@pyscript" / "core" / "dist"
             pyscript_static_dir = static_dir / "pyscript"
-            if not pyscript_static_dir.exists():
-                pyscript_static_dir.mkdir()
-            for file in pyscript_dist.iterdir():
-                shutil.copy(file, pyscript_static_dir / file.name)
+            copy_js_files(pyscript_dist, pyscript_static_dir)
 
             log.info("Copying Morphdom distribution")
             morphdom_dist = js_dir / "node_modules" / "morphdom" / "dist"
             morphdom_static_dir = static_dir / "morphdom"
-            if not morphdom_static_dir.exists():
-                morphdom_static_dir.mkdir()
-            for file in morphdom_dist.iterdir():
-                shutil.copy(file, morphdom_static_dir / file.name)
+            copy_js_files(morphdom_dist, morphdom_static_dir)
 
             log.info("Successfully built Javascript")
             super().run()

tests/test_app/__init__.py

@@ -21,32 +21,39 @@
     == 0
 )
 
-# Make sure the current PyScript distribution is always available
-pyscript_dist = js_dir / "node_modules" / "@pyscript" / "core" / "dist"
-pyscript_static_dir = (
+
+# Make sure the test environment is always using the latest JS
+def copy_js_files(source_dir: Path, destination: Path) -> None:
+    if destination.exists():
+        shutil.rmtree(destination)
+    destination.mkdir()
+
+    for file in source_dir.iterdir():
+        if file.is_file():
+            shutil.copy(file, destination / file.name)
+        else:
+            copy_js_files(file, destination / file.name)
+
+
+# Copy PyScript
+copy_js_files(
+    js_dir / "node_modules" / "@pyscript" / "core" / "dist",
     Path(__file__).parent.parent.parent
     / "src"
     / "reactpy_django"
     / "static"
     / "reactpy_django"
-    / "pyscript"
+    / "pyscript",
 )
-if not pyscript_static_dir.exists():
-    pyscript_static_dir.mkdir()
-for file in pyscript_dist.iterdir():
-    shutil.copy(file, pyscript_static_dir / file.name)
-
-# Make sure the current Morphdom distrubiton is always available
-morphdom_dist = js_dir / "node_modules" / "morphdom" / "dist"
-morphdom_static_dir = (
+
+
+# Copy MorphDOM
+copy_js_files(
+    js_dir / "node_modules" / "morphdom" / "dist",
     Path(__file__).parent.parent.parent
     / "src"
     / "reactpy_django"
     / "static"
     / "reactpy_django"
-    / "morphdom"
+    / "morphdom",
 )
-if not morphdom_static_dir.exists():
-    morphdom_static_dir.mkdir()
-for file in morphdom_dist.iterdir():
-    shutil.copy(file, morphdom_static_dir / file.name)