Add files via upload

PCRIInstall-v1p4.py

@@ -0,0 +1,138 @@
+#-----------------------------------------------------------------------------
+# Title:        PrimeCuber-RI Installer
+#
+# Author:    David Gilday
+#
+# Unofficial adaptation of PrimeCuber for Robot Inventor Kit: Marin (HW) & Mladen Dokmanovic (SW)
+#
+# Copyright:    (C) 2021 David Gilday
+#
+# Website:    http://MindCuber.com
+#
+# Version:    v1p4
+#
+# Modified:    $Date: 2021-05-21 15:57:52 +0000 (Fri, 21 May 2021) $
+#
+# Revision:    $Revision: 7865 $
+#
+# Usage:
+#
+#This software may be used for any non-commercial purpose providing
+#that the original author is acknowledged.
+#
+# Disclaimer:
+#
+#This software is provided 'as is' without warranty of any kind, either
+#express or implied, including, but not limited to, the implied warranties
+#of fitness for a purpose, or the warranty of non-infringement.
+#
+#-----------------------------------------------------------------------------
+# Purpose:    PrimeCuber-RI-v1p4 software installer
+#-----------------------------------------------------------------------------
+#
+# Note:
+#This program must be run once to install modules and data files
+#that will allow the main program, PrimeCuber-RI-v1p4, to run
+#
+#-----------------------------------------------------------------------------
+
+from spike import LightMatrix, Speaker
+lm = LightMatrix()
+lm.show_image('SQUARE')
+sp = Speaker()
+sp.beep(72)
+
+import os, umachine, ubinascii
+
+version = "v1p4"
+
+# Remove any files from older versions
+for fn in os.listdir("/"):
+    if len(fn) > 10:
+        ver = None
+        if fn[-3:] == ".py" and (
+            fn[:-7] == "pccolors_" or
+            fn[:-7] == "pcsolver_" or
+            fn[:-7] == "primecuber_"
+            ):
+            ver = fn[-7:-3]
+        elif fn[-4:] == ".bin" and (
+            fn[:-8] == "pcmtab1_" or
+            fn[:-8] == "pcmtab4_"
+            ):
+            ver = fn[-8:-4]
+# MD: Delete files even if the version is the same
+        if ver != None: # and ver < version:
+            print("DELETING: "+fn)
+            os.unlink(fn)
+
+def file_exists(fn):
+    try:
+        ok = os.stat(fn) != None
+    except:
+        ok = False
+    return ok
+
+# Install PrimeCuber-RI v1p4 files
+prj = "/projects/"
+found = 0
+with open(prj+".slots","r") as f:
+    slots = eval(f.read())
+for s in slots:
+    base = prj+str(slots[s]['id'])
+    # Filename used by latest hub OS
+    fn = base+"/__init__.py"
+    if not file_exists(fn):
+        # Try filename used by older versions of hub OS
+        fn = base+".py"
+    if file_exists(fn):
+        with open(fn) as f:
+            for i in range(3):
+                l = f.readline()
+                if l == "#PrimeCuberRI_FILES_v1p4#\n":
+                    print("SLOT: "+str(s)+" "+fn+" "+str(os.stat(fn)[6])+"B")
+                    print("Installing...")
+                    found += 1
+                    of = None
+                    b64 = False
+                    n = 0
+                    for l in f:
+                        if l[:5] == "#FILE":
+                            ofn = l[5:-1]
+                            b64 = ofn[-4:] == ".bin"
+                            sp.beep(67)
+                            of = open(ofn, 'wb')
+                        elif l[:8] == "#ENDFILE":
+                            of.close()
+                            of = None
+                            print("SAVED: "+ofn+" "+str(os.stat(ofn)[6])+"B")
+                        elif of != None:
+                            if b64:
+                                if l[0:5] != "#====":
+                                    of.write(ubinascii.a2b_base64(l[1:-1]))
+                            else:
+                                of.write(l)
+                            n += 1
+                            if n % 50 == 0:
+                                lm.show_image('CLOCK'+str(1+(int(n/50)%12)))
+                    if of != None:
+                        # Missing end of file
+                        of.close()
+                        print("ERROR: end file marker expected")
+                        print("DELETING: "+ofn)
+                        ofn.unlink()
+os.sync()
+if found > 0:
+    sp.beep(72)
+    msg = "PrimeCuber-RI v1p4 files installed"
+    print("FINISHED "+msg)
+    lm.write(msg)
+    umachine.reset()
+else:
+    msg = "ERROR: no files found to install"
+    print(msg)
+    lm.write(msg)
+
+raise SystemExit
+
+# END