gr33nhouse: use sd card

python_payload/apps/gr33nhouse/download.py

 from st3m.input import InputState
 from st3m.goose import Optional, List
 from st3m.ui import colours
+from st3m.utils import sd_card_plugged
 import urequests
 import gzip
 from utarfile import TarFile, DIRTYPE
@@ -36,6 +37,18 @@ class DownloadView(BaseView):
         self.response = b""
         self._download_instance = None
 
+    def _get_app_folder(self, tar_size: int) -> Optional[str]:
+        if sd_card_plugged():
+            sd_statvfs = os.statvfs("/sd")
+            if tar_size < sd_statvfs[1] * sd_statvfs[3]:
+                return "/sd/apps/"
+
+        flash_statvfs = os.statvfs("/flash")
+        if tar_size < flash_statvfs[1] * flash_statvfs[3]:
+            return "/flash/apps/"
+
+        return None
+
     def draw(self, ctx: Context) -> None:
         ctx.rgb(0, 0, 0).rectangle(-120, -120, 240, 240).fill()
 
@@ -157,23 +170,31 @@ class DownloadView(BaseView):
                 self._state = 6
                 return
 
+            app_folder = self._get_app_folder(len(tar))
+            if not app_folder:
+                gc.collect()
+                self.response = None
+                self.error_message = f"Not Enough Space\nSD/flash lack:\n{len(tar)}b"
+                self._state = 6
+                return
+
+            if not os.path.exists(app_folder):
+                print(f"making {app_folder}")
+                os.mkdir(app_folder)
+
             for i in t:
                 print(i.name)
 
-                if not os.path.exists("/flash/apps/"):
-                    print("making /flash/apps/")
-                    os.mkdir("/flash/apps/")
-
                 if i.type == DIRTYPE:
                     print("dirtype")
-                    dirname = "/flash/apps/" + i.name
+                    dirname = app_folder + i.name
                     if not os.path.exists(dirname):
                         print("making", dirname)
                         os.mkdir(dirname)
                     else:
                         print("dir", dirname, "exists")
                 else:
-                    filename = "/flash/apps/" + i.name
+                    filename = app_folder + i.name
                     print("writing to", filename)
                     f = t.extractfile(i)
                     with open(filename, "wb") as of:

python_payload/apps/w1f1/__init__.py

@@ -3,7 +3,7 @@ from st3m.input import InputState
 from st3m.goose import Optional
 from st3m.ui.view import ViewManager
 from st3m.settings import SETTINGS_JSON_FILE
-from st3m.utils import save_file_if_changed
+from st3m.utils import save_file_if_changed, sd_card_plugged
 from ctx import Context
 import network
 import leds
@@ -12,7 +12,6 @@ import json
 import math
 from .k3yboard import TextInputModel, KeyboardView
 from .helpers import (
-    sd_card_plugged,
     set_direction_leds,
     copy_across_devices,
     mark_unknown_characters,

python_payload/apps/w1f1/helpers.py

-import os
 import leds
 
 
-def sd_card_plugged() -> bool:
-    try:
-        os.listdir("/sd")
-        return True
-    except OSError:
-        # OSError: [Errno 19] ENODEV
-        return False
-
-
 def copy_across_devices(src: str, dst: str):
     """
     copy a file across devices (flash->SD etc).

python_payload/st3m/utils.py

@@ -137,4 +137,13 @@ def save_file_if_changed(filename: str, desired_contents: str) -> bool:
     return save
 
 
+def sd_card_plugged() -> bool:
+    try:
+        os.listdir("/sd")
+        return True
+    except OSError:
+        # OSError: [Errno 19] ENODEV
+        return False
+
+
 tau = math.pi * 2