[AIRFLOW-5727] SqoopHook: Build --connect parameter only if port/sche…

…ma are defined (#6397) (cherry picked from commit 4649563)
apache · Dec 16, 2019 · b5d7cfb · b5d7cfb
1 parent 4569ee7
commit b5d7cfb
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/airflow/contrib/hooks/sqoop_hook.py b/airflow/contrib/hooks/sqoop_hook.py
@@ -147,11 +147,12 @@ def _prepare_command(self, export=False):
         if self.hcatalog_table:
             connection_cmd += ["--hcatalog-table", self.hcatalog_table]
 
-        connection_cmd += ["--connect", "{}:{}/{}".format(
-            self.conn.host,
-            self.conn.port,
-            self.conn.schema
-        )]
+        connect_str = self.conn.host
+        if self.conn.port:
+            connect_str += ":{}".format(self.conn.port)
+        if self.conn.schema:
+            connect_str += "/{}".format(self.conn.schema)
+        connection_cmd += ["--connect", connect_str]
 
         return connection_cmd