Allow back script_location in Glue to be None

Fix error introduced in apache#20935 where script location in Glue Job could be None.
potiuk · Apr 29, 2022 · 82ad5d5 · 82ad5d5
1 parent f197030
commit 82ad5d5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/airflow/providers/amazon/aws/operators/glue.py b/airflow/providers/amazon/aws/operators/glue.py
@@ -104,7 +104,9 @@ def execute(self, context: 'Context'):
 
         :return: the id of the current glue job.
         """
-        if not self.script_location.startswith(self.s3_protocol):
+        if self.script_location is None:
+            s3_script_location = None
+        elif self.script_location.startswith(self.s3_protocol):
             s3_hook = S3Hook(aws_conn_id=self.aws_conn_id)
             script_name = os.path.basename(self.script_location)
             s3_hook.load_file(