Merge pull request #1400 from dod-ccpo/portfolio-data-as-dict

class method to export portfolio data as dictionary as an arg to stat…

atst/jobs.py

@@ -180,7 +180,7 @@ def do_work(fn, task, csp, **kwargs):
 def do_provision_portfolio(csp: CloudProviderInterface, portfolio_id=None):
     portfolio = Portfolios.get_for_update(portfolio_id)
     fsm = Portfolios.get_or_create_state_machine(portfolio)
-    fsm.trigger_next_transition()
+    fsm.trigger_next_transition(csp_data=portfolio.to_dictionary())
 
 
 @celery.task(bind=True, base=RecordFailure)

atst/models/portfolio.py

@@ -1,11 +1,16 @@
+import re
+from string import ascii_lowercase, digits
+from random import choices
+from itertools import chain
+
 from sqlalchemy import Column, String
 from sqlalchemy.orm import relationship
 from sqlalchemy.types import ARRAY
-from itertools import chain
 
 from atst.models.base import Base
 import atst.models.types as types
 import atst.models.mixins as mixins
+from atst.models.task_order import TaskOrder
 from atst.models.portfolio_role import PortfolioRole, Status as PortfolioRoleStatus
 from atst.domain.permission_sets import PermissionSets
 from atst.utils import first_or_none
@@ -159,6 +164,51 @@ class Portfolio(
     def application_id(self):
         return None
 
+    def to_dictionary(self):
+        ppoc = self.owner
+        user_id = f"{ppoc.first_name[0]}{ppoc.last_name}".lower()
+
+        domain_name = re.sub("[^0-9a-zA-Z]+", "", self.name).lower() + "".join(
+            choices(ascii_lowercase + digits, k=4)
+        )
+        portfolio_data = {
+            "user_id": user_id,
+            "password": "",
+            "domain_name": domain_name,
+            "first_name": ppoc.first_name,
+            "last_name": ppoc.last_name,
+            "country_code": "US",
+            "password_recovery_email_address": ppoc.email,
+            "address": {  # TODO: TBD if we're sourcing this from data or config
+                "company_name": "",
+                "address_line_1": "",
+                "city": "",
+                "region": "",
+                "country": "",
+                "postal_code": "",
+            },
+            "billing_profile_display_name": "ATAT Billing Profile",
+        }
+
+        try:
+            initial_task_order: TaskOrder = self.task_orders[0]
+            initial_clin = initial_task_order.sorted_clins[0]
+            portfolio_data.update(
+                {
+                    "initial_clin_amount": initial_clin.obligated_amount,
+                    "initial_clin_start_date": initial_clin.start_date.strftime(
+                        "%Y/%m/%d"
+                    ),
+                    "initial_clin_end_date": initial_clin.end_date.strftime("%Y/%m/%d"),
+                    "initial_clin_type": initial_clin.number,
+                    "initial_task_order_id": initial_task_order.number,
+                }
+            )
+        except IndexError:
+            pass
+
+        return portfolio_data
+
     def __repr__(self):
         return "<Portfolio(name='{}', user_count='{}', id='{}')>".format(
             self.name, self.user_count, self.id

atst/models/portfolio_state_machine.py

@@ -99,6 +99,8 @@ class PortfolioStateMachine(
     def trigger_next_transition(self, **kwargs):
         state_obj = self.machine.get_state(self.state)
 
+        kwargs["csp_data"] = kwargs.get("csp_data", {})
+
         if state_obj.is_system:
             if self.current_state in (FSMStates.UNSTARTED, FSMStates.STARTING):
                 # call the first trigger availabe for these two system states