If a fragment fails to upload, keep trying until it succeeds

DavidBerdik · Jun 23, 2019 · 5e75389 · 5e75389
1 parent 7db0470
commit 5e75389
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/InfiniDrive.py b/InfiniDrive.py
@@ -81,7 +81,13 @@ def print_ascii_logo():
 		doc.save(mem_doc)
 
 		# Upload Word document to Google Drive
-		driveAPI.store_doc(driveConnect, dirId, str(docNum) + ".docx", mem_doc)
+		while True:
+			try:
+				driveAPI.store_doc(driveConnect, dirId, str(docNum) + ".docx", mem_doc)
+			except:
+				print('Fragment ' + str(docNum) + ' failed to upload. Retrying.')
+				continue
+			break
 
 		# Increment docNum for next Word document and read next chunk of data.
 		docNum = docNum + 1