Connector verification type (#77)
* added auth columns * Adds API auth migration to connectors * precommit fixes --------- Co-authored-by: Graham Williamson <graham.williamson@socfortress.co>
taylor_socfortress committed
Jul 27, 2023 at 06:05 UTC
1ba789a69c393bbd2b3558cf388fbd6400bc30e5
2 files changed
+41
backend/app/models/models.py
+6
@@ -30,6 +30,9 @@ class ConnectorsAvailable(db.Model):
30
connector_supports: Column[String] = db.Column(db.String(100))
31
connector_configured: Column[Boolean] = db.Column(db.Boolean, default=False)
32
connector_verified: Column[Boolean] = db.Column(db.Boolean, default=False)
33
+ connector_accepts_api_key: Column[Boolean] = db.Column(db.Boolean, default=False)
34
+ connector_accepts_username_password: Column[Boolean] = db.Column(db.Boolean, default=False)
35
+ connector_accepts_file: Column[Boolean] = db.Column(db.Boolean, default=False)
36
37
def __init__(self, connector_name: str, connector_supports: str):
38
"""
@@ -67,6 +70,9 @@ class ConnectorsAvailableSchema(ma.Schema):
70
"connector_supports",
71
"connector_configured",
72
"connector_verified",
73
+ "connector_accepts_api_key",
74
+ "connector_accepts_username_password",
75
+ "connector_accepts_file",
76
)
77
78
backend/migrations/versions/c8c903153246_adds_api_auth_to_connectors.py
new
+35
@@ -0,0 +1,35 @@
1
+"""Adds API auth to connectors
2
+
3
+Revision ID: c8c903153246
4
+Revises: 7b25ec454c0a
5
+Create Date: 2023-07-27 12:17:34.109626
6
+
7
+"""
8
+import sqlalchemy as sa
9
+from alembic import op
10
+
11
+# revision identifiers, used by Alembic.
12
+revision = "c8c903153246"
13
+down_revision = "7b25ec454c0a"
14
+branch_labels = None
15
+depends_on = None
16
+
17
+
18
+def upgrade():
19
+ # ### commands auto generated by Alembic - please adjust! ###
20
+ with op.batch_alter_table("connectors_available", schema=None) as batch_op:
21
+ batch_op.add_column(sa.Column("connector_accepts_api_key", sa.Boolean(), nullable=True))
22
+ batch_op.add_column(sa.Column("connector_accepts_username_password", sa.Boolean(), nullable=True))
23
+ batch_op.add_column(sa.Column("connector_accepts_file", sa.Boolean(), nullable=True))
24
+
25
+ # ### end Alembic commands ###
26
+
27
+
28
+def downgrade():
29
+ # ### commands auto generated by Alembic - please adjust! ###
30
+ with op.batch_alter_table("connectors_available", schema=None) as batch_op:
31
+ batch_op.drop_column("connector_accepts_file")
32
+ batch_op.drop_column("connector_accepts_username_password")
33
+ batch_op.drop_column("connector_accepts_api_key")
34
+
35
+ # ### end Alembic commands ###