apps: software: Add support for user-defined applications

Signed-off-by: Daniel Thompson <daniel@redfelineninja.org.uk>

1 files changed, 15 insertions, 0 deletions

diff --git a/wasp/apps/software.py b/wasp/apps/software.py
index 91bde38..c51036b 100644
--- a/wasp/apps/software.py
+++ b/wasp/apps/software.py
@@ -16,6 +16,7 @@ whilst still allowing users to activate so many awesome applications!
 
 import wasp
 import icons
+import os
 
 
 class SoftwareApp():
@@ -50,6 +51,20 @@ class SoftwareApp():
         db.append(('timer', factory('Timer')))
         db.append(('weather', factory('Weather')))
 
+        # Handle user-loaded applications
+        try:
+            for app in os.listdir('apps'):
+                name = None
+                if app.endswith('.py'):
+                    name = app[:-3]
+                if app.endswith('.mpy'):
+                    name = app[:-4]
+                if name:
+                    db.append((name, factory(name)))
+        except OSError:
+            # apps does not exist...
+            pass
+
         # Get the initial state for the checkboxes
         for _, checkbox in db:
             label = checkbox.label.replace(' ', '')