First Commit

Firmware/updates.py

@@ -0,0 +1,59 @@
+from CONFIG.OTA_CONFIG import OTA_HOST, PROJECT_NAME, FILENAMES
+from CONFIG.MQTT_CONFIG import MQTT_CLIENT_ID
+import uos
+import urequests
+import micropython as mp
+import gc
+
+
+
+def update_file_replace(msg_string):
+    print(f"Starting update process for {msg_string}...")
+    filename = msg_string
+    
+    mp.mem_info(1)
+    gc.collect()
+    
+    try:
+          
+        updated = False
+        print(f"Updating file {filename}")
+        
+        for i,item in enumerate(FILENAMES):
+            print(f"Seeing if {filename} is in {item}")
+
+            if filename in item:
+                file_to_write = item
+                print(f"Found filename! Simple name: {filename} Fullly Qualified: {item}")
+                try:
+                    uos.mkdir('tmp')
+                except:
+                    pass
+                
+                updated = False
+                file_to_write = FILENAMES[i]
+                response = urequests.get(f'{OTA_HOST}/ota_updates/{MQTT_CLIENT_ID}/{filename}', timeout=5)
+                response_text = response.text
+                response.close()
+                #print(f"Found file {filename} with {response_text}")
+                # Get the file we need to write
+                # Write to a tmp file
+                print(f"Going to try to write to tmp/{file_to_write}")
+
+                with open(f'tmp/{filename}', 'w') as source_file:
+                    source_file.write(response_text)
+                    
+                    
+                # Overwrite our onboard file               
+                with open(f'tmp/{filename}', 'r') as source_file, open(file_to_write, 'w') as target_file:
+                    target_file.write(source_file.read())
+                
+                uos.remove(f'tmp/{filename}')
+                    
+                try:
+                    uos.rmdir('tmp')
+                except:
+                    pass
+                break
+    except Exception as e:
+        print(f"Exception updating file! {e}")