Add --no-upload argument if the OTA file's already uploaded skip the reuploading

* If the file is already uploaded utilize the --no-upload arg to skip the reupload & add the OTA url link to the json file
* If the file doesn't exist it will add NOT_FOUND_ON_DROPBOX in the json url section

Signed-off-by: Arindam Bhattacharjee <abhattacharjee717@gmail.com>

Author: userariii

generate_ota.py

@@ -46,10 +46,28 @@ def sha256sum(filename):
             h.update(chunk)
     return h.hexdigest()
 
+def check_file_exists(dbx, dropbox_path):
+    try:
+        dbx.files_get_metadata(dropbox_path)
+        return True
+    except dropbox.exceptions.ApiError:
+        return False
+
+def get_or_create_shared_link(dbx, dropbox_path):
+    try:
+        links = dbx.sharing_list_shared_links(path=dropbox_path).links
+        if links:
+            return make_direct_download_link(links[0].url)
+        else:
+            link_metadata = dbx.sharing_create_shared_link_with_settings(dropbox_path)
+            return make_direct_download_link(link_metadata.url)
+    except dropbox.exceptions.ApiError as e:
+        print(f"⚠️ Could not get or create shared link for {dropbox_path}: {e}")
+        return None
+
 def upload_to_dropbox(local_path, dropbox_folder, access_token):
     dbx = dropbox.Dropbox(access_token)
     filename = os.path.basename(local_path)
-
     # Upload path: /<codename>/<filename>
     dropbox_path = f"/{dropbox_folder}/{filename}".replace("//", "/")
 
@@ -67,7 +85,6 @@ def upload_to_dropbox(local_path, dropbox_folder, access_token):
             offset = CHUNK_SIZE
             with tqdm(total=file_size, unit='B', unit_scale=True, desc='Uploading') as pbar:
                 pbar.update(CHUNK_SIZE)
-
                 while offset < file_size:
                     chunk = f.read(CHUNK_SIZE)
                     if (file_size - offset) <= CHUNK_SIZE:
@@ -80,29 +97,17 @@ def upload_to_dropbox(local_path, dropbox_folder, access_token):
                         cursor.offset = offset
                         pbar.update(len(chunk))
 
-    try:
-        link_metadata = dbx.sharing_create_shared_link_with_settings(dropbox_path)
-    except dropbox.exceptions.ApiError as e:
-        if e.error.is_shared_link_already_exists():
-            links = dbx.sharing_list_shared_links(path=dropbox_path).links
-            if links:
-                link_metadata = links[0]
-            else:
-                print("⚠️ Failed to retrieve existing shared link.")
-                sys.exit(1)
-        else:
-            print(f"❌ Failed to create shared link: {e}")
-            sys.exit(1)
-
-    direct_link = make_direct_download_link(link_metadata.url)
-    print("✅ Direct Download Link:")
-    print(direct_link)
-    return direct_link
+    return get_or_create_shared_link(dbx, dropbox_path)
 
 def main():
+    args = sys.argv[1:]
+    no_upload = "--no-upload" in args
+    codename_arg = next((arg for arg in args if not arg.startswith("--")), None)
+
     access_token = load_access_token()
+    dbx = dropbox.Dropbox(access_token)
 
-    codename = input("Enter device codename (e.g. PL2, miatoll): ").strip()
+    codename = codename_arg.strip() if codename_arg else input("Enter device codename (e.g. PL2, miatoll): ").strip()
     ota_dir = os.path.join("out", "target", "product", codename)
 
     try:
@@ -115,9 +120,12 @@ def main():
         print(f"❌ No OTA zip found in {ota_dir} matching pattern AndroidOne-*{codename}-*.zip")
         sys.exit(1)
 
-    filename = files[0]
-    file_path = os.path.join(ota_dir, filename)
-    print(f"📦 Found OTA package: {file_path}")
+    ota_filename = files[0]
+    ota_filepath = os.path.join(ota_dir, ota_filename)
+    dropbox_path = f"/{codename}/{ota_filename}"
+    print(f"📦 Found OTA package: {ota_filepath}")
+
+    android_exp_files = [f for f in os.listdir(ota_dir) if f.startswith("AndroidExperience")]
 
     build_prop = os.path.join(ota_dir, "system", "build.prop")
     datetime = "UNKNOWN"
@@ -128,29 +136,51 @@ def main():
                     datetime = line.strip().split("=")[1]
                     break
 
-    file_id = sha256sum(file_path)
-    file_size = os.path.getsize(file_path)
-    version = "15"  # Adjust this if needed
-
-    dropbox_link = upload_to_dropbox(file_path, codename, access_token)
+    version = "15"
+    ota_file_id = sha256sum(ota_filepath)
+    ota_file_size = os.path.getsize(ota_filepath)
 
-    output_dir = os.path.join(SCRIPT_DIR, "devices")
-    os.makedirs(output_dir, exist_ok=True)
-    output_file = os.path.join(output_dir, f"{codename}.json")
+    if no_upload:
+        print("⏩ --no-upload specified, skipping all uploads...")
+        if check_file_exists(dbx, dropbox_path):
+            print(f"ℹ️ OTA file exists on Dropbox: {dropbox_path}")
+            ota_link = get_or_create_shared_link(dbx, dropbox_path) or "LINK_NOT_FOUND"
+        else:
+            print(f"⚠️ OTA file not found on Dropbox at {dropbox_path}")
+            ota_link = "NOT_FOUND_ON_DROPBOX"
+    else:
+        if check_file_exists(dbx, dropbox_path):
+            print(f"✅ File already exists on Dropbox: {dropbox_path}")
+            ota_link = get_or_create_shared_link(dbx, dropbox_path)
+        else:
+            print("⬆️ File not found on Dropbox, proceeding to upload...")
+            ota_link = upload_to_dropbox(ota_filepath, codename, access_token)
+
+        if android_exp_files:
+            recovery_path = os.path.join(ota_dir, "recovery.img")
+            if os.path.isfile(recovery_path):
+                print("🔍 AndroidExperience file detected, uploading recovery.img (no JSON entry)...")
+                upload_to_dropbox(recovery_path, codename, access_token)
+            else:
+                print("⚠️ recovery.img NOT found — skipping.")
 
     ota_json = {
         "response": [
             {
                 "datetime": datetime,
-                "filename": filename,
-                "id": file_id,
-                "size": file_size,
-                "url": dropbox_link,
+                "filename": ota_filename,
+                "id": ota_file_id,
+                "size": ota_file_size,
+                "url": ota_link,
                 "version": version
             }
         ]
     }
 
+    output_dir = os.path.join(SCRIPT_DIR, "devices")
+    os.makedirs(output_dir, exist_ok=True)
+    output_file = os.path.join(output_dir, f"{codename}.json")
+
     with open(output_file, "w") as f:
         json.dump(ota_json, f, indent=2)