Merge pull request #293 from dod-ccpo/homepage-redirects

Redirect from /home depending on requests and workspaces

atst/models/request.py

@@ -37,7 +37,7 @@ class Request(Base):
     workspace = relationship("Workspace", uselist=False, backref="request")
 
     user_id = Column(ForeignKey("users.id"), nullable=False)
-    creator = relationship("User")
+    creator = relationship("User", backref="owned_requests")
 
     task_order_id = Column(ForeignKey("task_order.id"))
     task_order = relationship("TaskOrder")

atst/routes/__init__.py

@@ -17,7 +17,25 @@ def root():
 
 @bp.route("/home")
 def home():
-    return redirect(url_for("requests.requests_index"))
+    num_workspaces = len(g.current_user.workspace_roles)
+
+    if num_workspaces == 0:
+        return redirect(url_for("requests.requests_index"))
+    elif num_workspaces == 1:
+        workspace_role = g.current_user.workspace_roles[0]
+        workspace_id = workspace_role.workspace.id
+        is_request_owner = workspace_role.role.name == "owner"
+
+        if is_request_owner:
+            return redirect(
+                url_for("workspaces.workspace_reports", workspace_id=workspace_id)
+            )
+        else:
+            return redirect(
+                url_for("workspaces.workspace_projects", workspace_id=workspace_id)
+            )
+    else:
+        return redirect(url_for("workspaces.workspaces"))
 
 
 @bp.route("/styleguide")