From 66b304d34c6cca60dc45ec3ce975850c2de94389 Mon Sep 17 00:00:00 2001 From: Jeroen Oudshoorn Date: Fri, 13 Oct 2023 09:45:01 +0200 Subject: [PATCH] v2.4.7 Signed-off-by: Jeroen Oudshoorn --- pwnagotchi/plugins/default/gdrivesync.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/pwnagotchi/plugins/default/gdrivesync.py b/pwnagotchi/plugins/default/gdrivesync.py index 16c206ac..ca34ca93 100644 --- a/pwnagotchi/plugins/default/gdrivesync.py +++ b/pwnagotchi/plugins/default/gdrivesync.py @@ -71,7 +71,15 @@ class GdriveSync(plugins.Plugin): backup_folder_name = 'PwnagotchiBackups' backup_folder_id = self.get_folder_id_by_name(self.drive, backup_folder_name) - # Get the list of files in the folder + if not backup_folder_id: + # If the folder doesn't exist, create it + folder = self.drive.CreateFile( + {'title': backup_folder_name, 'mimeType': 'application/vnd.google-apps.folder'}) + folder.Upload() + backup_folder_id = folder['id'] + print(f"Created folder '{backup_folder_name}' with ID: {backup_folder_id}") + + # Continue with the rest of the code using backup_folder_id file_list = self.drive.ListFile({'q': f"'{backup_folder_id}' in parents and trashed=false"}).GetList() if not file_list: