From 50231a5154c23237f13e79b467a707197d8303fb Mon Sep 17 00:00:00 2001
From: Mike Lang <mikelang3000@gmail.com>
Date: Fri, 3 Sep 2021 12:26:19 +1000
Subject: [PATCH] py3 fixes for cutter

---
 cutter/cutter/main.py            | 19 +++++++++++--------
 cutter/cutter/upload_backends.py |  4 ++--
 2 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/cutter/cutter/main.py b/cutter/cutter/main.py
index 47ee46c..9815fc8 100644
--- a/cutter/cutter/main.py
+++ b/cutter/cutter/main.py
@@ -226,7 +226,7 @@ class Cutter(object):
 		""").format(
 			sql.SQL(", ").join(sql.Identifier(key) for key in CUT_JOB_PARAMS)
 		)
-		result = query(self.conn, built_query, name=self.name, upload_locations=self.upload_locations.keys())
+		result = query(self.conn, built_query, name=self.name, upload_locations=list(self.upload_locations.keys()))
 		return result.fetchall()
 
 	# No need to instrument this function, just use get_best_segments() stats
@@ -313,9 +313,7 @@ class Cutter(object):
 
 		# This flag tracks whether we've told requests to finalize the upload,
 		# and serves to detect whether errors from the request call are recoverable.
-		# Wrapping it in a one-element list is a hack that lets us modify it from within
-		# a closure (as py2 lacks the nonlocal keyword).
-		finalize_begun = [False]
+		finalize_begun = False
 
 		# This exception indicates a job we thought was ours somehow disappeared
 		# while we were still trying to cut it. This most likely represents a logic error
@@ -342,7 +340,7 @@ class Cutter(object):
 			if result.rowcount != 1:
 				# If we hadn't yet set finalizing, then this means an operator cancelled the job
 				# while we were cutting it. This isn't a problem.
-				if not finalize_begun[0]:
+				if not finalize_begun:
 					raise JobCancelled()
 				raise JobConsistencyError("No job with id {} and uploader {} when setting: {}".format(
 					job.id, self.name, ", ".join("{} = {!r}".format(k, v) for k, v in kwargs.items())
@@ -353,6 +351,11 @@ class Cutter(object):
 			# do things in between the data being finished and finalizing the request.
 			# This is also where we do the main error handling.
 
+			# Tell python to use the finalize_begun variable from the enclosing scope,
+			# instead of creating a new (shadowing) variable which is the default when
+			# you do "variable = value".
+			nonlocal finalize_begun
+
 			try:
 				for chunk in cut:
 					yield chunk
@@ -367,7 +370,7 @@ class Cutter(object):
 
 			self.logger.debug("Setting job to finalizing")
 			set_row(state='FINALIZING')
-			finalize_begun[0] = True
+			finalize_begun = True
 
 			# Now we return from this generator, and any unknown errors between now and returning
 			# from the upload backend are not recoverable.
@@ -393,7 +396,7 @@ class Cutter(object):
 				if isinstance(ex, requests.HTTPError):
 					ex = "{}: {}".format(ex, ex.response.content)
 
-				if not finalize_begun[0]:
+				if not finalize_begun:
 					# error before finalizing, assume it's a network issue / retryable.
 					self.logger.exception("Retryable error when uploading job {}".format(format_job(job)))
 					raise UploadError("Unhandled error in upload: {}".format(ex), retryable=True)
@@ -552,7 +555,7 @@ class TranscodeChecker(object):
 			UPDATE events
 			SET state = 'DONE', upload_time = %s
 			WHERE id = ANY (%s::uuid[]) AND state = 'TRANSCODING'
-		""", datetime.datetime.utcnow(), ids.keys())
+		""", datetime.datetime.utcnow(), list(ids.keys()))
 		return result.rowcount
 
 
diff --git a/cutter/cutter/upload_backends.py b/cutter/cutter/upload_backends.py
index 96b558a..346eb58 100644
--- a/cutter/cutter/upload_backends.py
+++ b/cutter/cutter/upload_backends.py
@@ -53,7 +53,7 @@ class UploadBackend(object):
 	Should have a method upload_video(title, description, tags, data).
 	Title, description and tags may have backend-specific meaning.
 	Tags is a list of string.
-	Data is an iterator of strings.
+	Data is an iterator of bytes.
 	It should return (video_id, video_link).
 
 	If the video must undergo additional processing before it's available
@@ -241,7 +241,7 @@ class Local(UploadBackend):
 		filepath = os.path.join(self.path, filename)
 		try:
 			if self.write_info:
-				with open(os.path.join(self.path, '{}-{}.json'.format(safe_title, video_id)), 'wb') as f:
+				with open(os.path.join(self.path, '{}-{}.json'.format(safe_title, video_id)), 'w') as f:
 					f.write(json.dumps({
 						'title': title,
 						'description': description,