Handle if uploading throws an error

ow · Apr 24, 2023 · 8f0d359 · 8f0d359
1 parent fa0a0b9
commit 8f0d359
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/art.py b/art.py
@@ -83,11 +83,15 @@
 				if remote_filename is None:
 						logging.info('Uploading new image: ' + str(file))
 
-						if file.endswith('.jpg'):
-								remote_filename = tv.art().upload(data, file_type='JPEG', matte="none")
-						elif file.endswith('.png'):
-								remote_filename = tv.art().upload(data, file_type='PNG', matte="none")
-
+						try:
+							if file.endswith('.jpg'):
+									remote_filename = tv.art().upload(data, file_type='JPEG', matte="none")
+							elif file.endswith('.png'):
+									remote_filename = tv.art().upload(data, file_type='PNG', matte="none")
+						except Exception as e:
+							logging.error('There was an error: ' + str(e))
+							sys.exit()
+
 						# Add the filename to the list of uploaded filenames
 						uploaded_files.append({'file': file, 'remote_filename': remote_filename})