Skip to content
Snippets Groups Projects
Commit 78261fdb authored by Neda Moeini's avatar Neda Moeini
Browse files

Replace GEANT as a customer in importing site.

parent 12a9336a
No related branches found
No related tags found
1 merge request!60Feature/nat 217 import sites
Pipeline #83855 failed
...@@ -42,6 +42,7 @@ class SiteImport(BaseModel): ...@@ -42,6 +42,7 @@ class SiteImport(BaseModel):
site_internal_id: int site_internal_id: int
site_tier: SiteTier site_tier: SiteTier
site_ts_address: str site_ts_address: str
customer: str
@router.post("/sites", status_code=status.HTTP_201_CREATED, tags=["Import"]) @router.post("/sites", status_code=status.HTTP_201_CREATED, tags=["Import"])
...@@ -70,7 +71,7 @@ def import_site(site: SiteImport) -> Dict[str, Any]: ...@@ -70,7 +71,7 @@ def import_site(site: SiteImport) -> Dict[str, Any]:
raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Multiple subscriptions found.") raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Multiple subscriptions found.")
pid = start_process("import_site", site.dict()) pid = start_process("import_site", site.dict())
return {"detail": "Site import process started.", "pid": pid} return {"detail": "Site added successfully.", "pid": pid}
class RouterImportModel(BaseModel): class RouterImportModel(BaseModel):
......
...@@ -9,13 +9,15 @@ from orchestrator.workflows.steps import resync, set_status, store_process_subsc ...@@ -9,13 +9,15 @@ from orchestrator.workflows.steps import resync, set_status, store_process_subsc
from gso.products.product_blocks.site import SiteTier from gso.products.product_blocks.site import SiteTier
from gso.products.product_types import site from gso.products.product_types import site
from gso.services.crm import get_customer_by_name
from gso.workflows.site.create_site import initialize_subscription from gso.workflows.site.create_site import initialize_subscription
@step("Create subscription") @step("Create subscription")
def create_subscription() -> State: def create_subscription(customer: str) -> State:
customer_id: UUID = get_customer_by_name(customer)["id"]
product_id: UUID = ProductTable.query.filter_by(product_type="Site").first().product_id product_id: UUID = ProductTable.query.filter_by(product_type="Site").first().product_id
subscription = site.SiteInactive.from_product_id(product_id, uuid4()) subscription = site.SiteInactive.from_product_id(product_id, customer_id)
return { return {
"subscription": subscription, "subscription": subscription,
...@@ -38,6 +40,7 @@ def generate_initial_input_form() -> FormGenerator: ...@@ -38,6 +40,7 @@ def generate_initial_input_form() -> FormGenerator:
site_internal_id: int site_internal_id: int
site_tier: SiteTier site_tier: SiteTier
site_ts_address: str site_ts_address: str
customer: str
user_input = yield ImportSite user_input = yield ImportSite
return user_input.dict() return user_input.dict()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment