Adding team url to datasource.

ministryofjustice · Nov 15, 2023 · ae1d545 · ae1d545
1 parent 43b6db0
commit ae1d545
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/app/app.py b/app/app.py
@@ -214,15 +214,18 @@ def create_data_source():
                 team_info = team_creation_response.json()
                 print(f"Team Info: %s", team_info)
                 team_id = team_info.get("id")
-                if team_id:
+                team_web_url = team_info.get("webUrl")
+                if team_id and team_web_url:
                     data_source.team_id = team_id
+                    data_source.team_web_url = team_web_url
+
                     db.session.commit()
                     flash(
                         "Team created and associated with data source successfully!",
                         "success",
                     )
                 else:
-                    flash("Failed to extract team ID from the response.", "error")
+                    flash("Failed to extract team ID or web url from the response.", "error")
             else:  # If there was an error
                 flash("Failed to create the team.", "error")
         else:

diff --git a/app/models.py b/app/models.py
@@ -31,6 +31,7 @@ class DataSource(db.Model):
     created_by = db.Column(db.String(255), db.ForeignKey("users.id"))
     aad_group_id = db.Column(db.String(255), unique=True)
     team_id = db.Column(db.String(255), unique=True)
+    team_web_url = db.Column(db.String(255), unique=True)
 
     # Relationships
     permissions = db.relationship(