From e955e9f574a83f59296e79ad7dae331921b35888 Mon Sep 17 00:00:00 2001 From: Hakan Calim <hakan.calim@fau.de> Date: Wed, 1 Nov 2023 21:18:06 +0100 Subject: [PATCH] NAT-328: updated validation site name by an utils method --- gso/workflows/site/create_site.py | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/gso/workflows/site/create_site.py b/gso/workflows/site/create_site.py index 625b458d..ae2e11c3 100644 --- a/gso/workflows/site/create_site.py +++ b/gso/workflows/site/create_site.py @@ -1,5 +1,3 @@ -import re - from orchestrator.forms import FormPage from orchestrator.targets import Target from orchestrator.types import FormGenerator, State, SubscriptionLifecycle, UUIDstr @@ -13,7 +11,12 @@ from gso.products.product_blocks import site as site_pb from gso.products.product_blocks.site import LatitudeCoordinate, LongitudeCoordinate from gso.products.product_types import site from gso.services.crm import customer_selector -from gso.utils.helpers import validate_country_code, validate_ipv4_or_ipv6, validate_site_fields_is_unique +from gso.utils.helpers import ( + validate_country_code, + validate_ipv4_or_ipv6, + validate_site_fields_is_unique, + validate_site_name, +) def initial_input_form_generator(product_name: str) -> FormGenerator: # noqa: C901 @@ -56,14 +59,8 @@ def initial_input_form_generator(product_name: str) -> FormGenerator: # noqa: C by an optional single digit (0-9). """ validate_site_fields_is_unique("site_name", site_name) - pattern = re.compile(r"^[A-Z]{3}[0-9]?$") - if pattern.match(site_name): - return site_name - else: - raise ValueError( - "Enter a valid site name. It must consist of three uppercase letters (A-Z) followed by an optional" - " single digit (0-9)." - ) + validate_site_name(site_name) + return site_name user_input = yield CreateSiteForm -- GitLab