From 984521f64309406d8cbd2be95dd8fb667a7831ee Mon Sep 17 00:00:00 2001
From: Shekar Siri <sshekarsiri@gmail.com>
Date: Thu, 6 Jul 2023 18:43:59 +0200
Subject: [PATCH] fix(api): feature flag key validation

---
 api/schemas.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/api/schemas.py b/api/schemas.py
index 2d2d9bc79..3606ded4f 100644
--- a/api/schemas.py
+++ b/api/schemas.py
@@ -5,6 +5,7 @@ from pydantic import BaseModel, Field, EmailStr, HttpUrl, root_validator, valida
 from pydantic.types import Json
 
 from chalicelib.utils.TimeUTC import TimeUTC
+import re
 
 
 def attribute_to_camel_case(snake_str):
@@ -1435,7 +1436,7 @@ class FeatureFlagStatus(BaseModel):
 
 class FeatureFlagSchema(BaseModel):
     payload: Optional[str] = Field(default=None)
-    flag_key: str = Field(...)
+    flag_key: str = Field(..., regex=r'^[a-zA-Z0-9\-]+$')
     description: Optional[str] = Field(None)
     flag_type: FeatureFlagType = Field(default=FeatureFlagType.single_variant)
     is_persist: Optional[bool] = Field(default=False)