Feature/nat 328 site names should be validated

gso/api/v1/imports.py

@@ -6,16 +6,21 @@ from fastapi import Depends, HTTPException, status
 from fastapi.routing import APIRouter
 from orchestrator.security import opa_security_default
 from orchestrator.services import processes
-from orchestrator.services import subscriptions as wfo_subscriptions
 from pydantic import BaseModel, root_validator, validator
-from sqlalchemy.exc import MultipleResultsFound
+from pydantic.fields import ModelField
 
 from gso.products.product_blocks.iptrunk import IptrunkType, PhyPortCapacity
 from gso.products.product_blocks.router import RouterRole, RouterVendor
 from gso.products.product_blocks.site import SiteTier
 from gso.services import subscriptions
 from gso.services.crm import CustomerNotFoundError, get_customer_by_name
-from gso.utils.helpers import LAGMember
+from gso.utils.helpers import (
+    LAGMember,
+    validate_country_code,
+    validate_ipv4_or_ipv6,
+    validate_site_fields_is_unique,
+    validate_site_name,
+)
 
 router = APIRouter(prefix="/imports", tags=["Imports"], dependencies=[Depends(opa_security_default)])
 
@@ -38,6 +43,32 @@ class SiteImportModel(BaseModel):
     site_ts_address: str
     customer: str
 
+    @validator("site_ts_address", allow_reuse=True)
+    def validate_ts_address(cls, site_ts_address: str) -> str:
+        validate_site_fields_is_unique("site_ts_address", site_ts_address)
+        validate_ipv4_or_ipv6(site_ts_address)
+        return site_ts_address
+
+    @validator("site_country_code", allow_reuse=True)
+    def country_code_must_exist(cls, country_code: str) -> str:
+        validate_country_code(country_code)
+        return country_code
+
+    @validator("site_internal_id", "site_bgp_community_id", allow_reuse=True)
+    def validate_unique_fields(cls, value: str, field: ModelField) -> str | int:
+        return validate_site_fields_is_unique(field.name, value)
+
+    @validator("site_name", allow_reuse=True)
+    def site_name_must_be_valid(cls, site_name: str) -> str:
+        """Validate the site name.
+
+        The site name must consist of three uppercase letters (A-Z) followed
+        by an optional single digit (0-9).
+        """
+        validate_site_fields_is_unique("site_name", site_name)
+        validate_site_name(site_name)
+        return site_name
+
 
 class RouterImportModel(BaseModel):
     customer: str
@@ -151,15 +182,6 @@ def import_site(site: SiteImportModel) -> dict[str, Any]:
 
     :raises HTTPException: If the site already exists or if there's an error in the process.
     """
-    try:
-        subscription = wfo_subscriptions.retrieve_subscription_by_subscription_instance_value(
-            resource_type="site_name", value=site.site_name, sub_status=("provisioning", "active")
-        )
-        if subscription:
-            raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Site already exists.")
-    except MultipleResultsFound:
-        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Multiple subscriptions found.")
-
     pid = _start_process("import_site", site.dict())
     return {"detail": "Site added successfully.", "pid": pid}