From 2f7097f9f78b00a75115699134e174cd4357b91d Mon Sep 17 00:00:00 2001
From: Manuel Kamper <kamp@noreply.kmpr.at>
Date: Sat, 1 Jul 2023 05:43:03 +0000
Subject: [PATCH] =?UTF-8?q?Dateien=20hochladen=20nach=20=E2=80=9E=E2=80=9C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 OTA.py | 94 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 OTA.py

diff --git a/OTA.py b/OTA.py
new file mode 100644
index 0000000..7ba0cf3
--- /dev/null
+++ b/OTA.py
@@ -0,0 +1,94 @@
+import machine
+import ubinascii
+import uos
+import urequests
+
+def check_version(host, project, auth=None, timeout=5) -> (bool, str):
+    current_version = ''
+    try:
+        if 'version' in uos.listdir():
+            with open('version', 'r') as current_version_file:
+                current_version = current_version_file.readline().strip()
+
+        if auth:
+            response = urequests.get(f'{host}/{project}/version', headers={'Authorization': f'Basic {auth}'}, timeout=timeout)
+        else:
+            response = urequests.get(f'{host}/{project}/version', timeout=timeout)
+        response_status_code = response.status_code
+        response_text = response.text
+        response.close()
+        if response_status_code != 200:
+            print(f'Remote version file {host}/{project}/version not found')
+            return False, current_version
+        remote_version = response_text.strip()
+        return current_version != remote_version, remote_version
+    except Exception as ex:
+        print(f'Something went wrong: {ex}')
+        return False, current_version
+
+
+def generate_auth(user=None, passwd=None) -> str | None:
+    if not user and not passwd:
+        return None
+    if (user and not passwd) or (passwd and not user):
+        raise ValueError('Either only user or pass given. None or both are required.')
+    auth_bytes = ubinascii.b2a_base64(f'{user}:{passwd}'.encode())
+    return auth_bytes.decode().strip()
+
+
+def ota_update(host, project, filenames, use_version_prefix=True, user=None, passwd=None, hard_reset_device=True, soft_reset_device=False, timeout=5) -> None:
+    all_files_found = True
+    auth = generate_auth(user, passwd)
+    prefix_or_path_separator = '_' if use_version_prefix else '/'
+    try:
+        version_changed, remote_version = check_version(host, project, auth=auth, timeout=timeout)
+        if version_changed:
+            try:
+                uos.mkdir('tmp')
+            except:
+                pass
+            for filename in filenames:
+                if auth:
+                    response = urequests.get(f'{host}/{project}/{remote_version}{prefix_or_path_separator}{filename}', headers={'Authorization': f'Basic {auth}'}, timeout=timeout)
+                else:
+                    response = urequests.get(f'{host}/{project}/{remote_version}{prefix_or_path_separator}{filename}', timeout=timeout)
+                response_status_code = response.status_code
+                response_text = response.text
+                response.close()
+                if response_status_code != 200:
+                    print(f'Remote source file {host}/{project}/{remote_version}{prefix_or_path_separator}{filename} not found')
+                    all_files_found = False
+                    continue
+                with open(f'tmp/{filename}', 'w') as source_file:
+                    source_file.write(response_text)
+            if all_files_found:
+                for filename in filenames:
+                    with open(f'tmp/{filename}', 'r') as source_file, open(filename, 'w') as target_file:
+                        target_file.write(source_file.read())
+                    uos.remove(f'tmp/{filename}')
+                try:
+                    uos.rmdir('tmp')
+                except:
+                    pass
+                with open('version', 'w') as current_version_file:
+                    current_version_file.write(remote_version)
+                if soft_reset_device:
+                    print('Soft-resetting device...')
+                    machine.soft_reset()
+                if hard_reset_device:
+                    print('Hard-resetting device...')
+                    machine.reset()
+    except Exception as ex:
+        print(f'Something went wrong: {ex}')
+
+
+def check_for_ota_update(host, project, user=None, passwd=None, timeout=5, soft_reset_device=False):
+    auth = generate_auth(user, passwd)
+    version_changed, remote_version = check_version(host, project, auth=auth, timeout=timeout)
+    if version_changed:
+        if soft_reset_device:
+            print(f'Found new version {remote_version}, soft-resetting device...')
+            machine.soft_reset()
+        else:
+            print(f'Found new version {remote_version}, hard-resetting device...')
+            machine.reset()
\ No newline at end of file