Adding a proper primary key to db connections

airflow/models.py

@@ -139,7 +139,8 @@ class DatabaseConnection(Base):
     """
     __tablename__ = "db_connection"
 
-    db_id = Column(String(ID_LEN), primary_key=True)
+    id = Column(Integer(), primary_key=True)
+    db_id = Column(String(ID_LEN), unique=True)
     db_type = Column(String(500))
     host = Column(String(500))
     schema = Column(String(500))

airflow/operators/__init__.py

@@ -1,6 +1,7 @@
 from bash_operator import BashOperator
 from mysql_operator import MySqlOperator
 from hive_operator import HiveOperator
+from hive2samba_operator import Hive2SambaOperator
 from sensors import SqlSensor
 from sensors import ExternalTaskSensor
 from sensors import HivePartitionSensor

airflow/www/app.py

@@ -997,6 +997,8 @@ class DatabaseConnectionModelView(LoginMixin, ModelView):
             ('presto', 'Presto',),
             ('mysql', 'MySQL',),
             ('oracle', 'Oracle',),
+            ('samba', 'Samba',),
+            ('ftp', 'FTP',),
         ]
     }
 mv = DatabaseConnectionModelView(