KAN-46 add basic filter downloads ✨

download_filters.py

@@ -0,0 +1,29 @@
+"""Download filters from public repository."""
+
+import os
+import shutil
+import requests
+
+
+def download_test_filters(url):
+    """Download filters."""
+
+    r = requests.get(url, allow_redirects=True, timeout=10)
+    with open("filter/test_filters.zip", "wb") as f:
+        f.write(r.content)
+
+    shutil.unpack_archive(
+        "filter/test_filters.zip",
+        "filter/temp",
+        "zip",
+    )
+
+    shutil.copytree(
+        "filter/temp/test_filters",
+        "filter",
+        dirs_exist_ok=True,
+    )
+
+    shutil.rmtree("filter/temp")
+
+    os.remove("filter/test_filters.zip")

main.py

@@ -0,0 +1,21 @@
+"""Project CLI"""
+
+import click
+
+from download_filters import download_test_filters
+
+
+@click.group
+def cli():
+    pass
+
+
+@cli.command()
+def download_filters():
+    """Download filters."""
+    click.echo("Downloading filters...")
+    download_test_filters("https://xspect.s3.eu-central-1.amazonaws.com/filters.zip")
+
+
+if __name__ == "__main__":
+    cli()

tests/conftest.py

@@ -23,7 +23,8 @@ def pytest_sessionstart():
             print("Downloading " + assembly)
 
             r = requests.get(url, allow_redirects=True, timeout=10)
-            open("tests/test_assemblies/" + assembly + ".zip", "wb").write(r.content)
+            with open("tests/test_assemblies/" + assembly + ".zip", "wb") as f:
+                f.write(r.content)
 
             # Unzip and move
             shutil.unpack_archive(