jan/openreplay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(chalice): strict validation for names

Browse source
This commit is contained in:
Taha Yassine Kraiem 2023-11-15 18:05:19 +01:00
parent 4784c1548d
commit c9d0c0f2b2
1 changed files with 19 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
api/schemas.py
View file

@ -7,6 +7,8 @@ from pydantic.types import Json
from chalicelib.utils.TimeUTC import TimeUTC from chalicelib.utils.TimeUTC import TimeUTC
import re import re
NAME_PATTERN = r"^[a-z,A-Z,0-9,\-,é,è,à,ç, ,|,&,\/,\\,_,.,#]*$"
def attribute_to_camel_case(snake_str): def attribute_to_camel_case(snake_str):
components = snake_str.split("_") components = snake_str.split("_")
@ -32,16 +34,16 @@ class UserLoginSchema(_Grecaptcha):
class UserSignupSchema(UserLoginSchema): class UserSignupSchema(UserLoginSchema):
fullname: str = Field(...) fullname: str = Field(..., regex=NAME_PATTERN)
organizationName: str = Field(...) organizationName: str = Field(..., regex=NAME_PATTERN)
class Config: class Config:
alias_generator = attribute_to_camel_case alias_generator = attribute_to_camel_case
class EditAccountSchema(BaseModel): class EditAccountSchema(BaseModel):
name: Optional[str] = Field(None) name: Optional[str] = Field(None, regex=NAME_PATTERN)
tenantName: Optional[str] = Field(None) tenantName: Optional[str] = Field(None, regex=NAME_PATTERN)
opt_out: Optional[bool] = Field(None) opt_out: Optional[bool] = Field(None)
_transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace) _transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace)
@ -63,16 +65,16 @@ class EditUserPasswordSchema(BaseModel):
class UpdateTenantSchema(BaseModel): class UpdateTenantSchema(BaseModel):
name: Optional[str] = Field(None) name: Optional[str] = Field(None, regex=NAME_PATTERN)
opt_out: Optional[bool] = Field(None) opt_out: Optional[bool] = Field(None)
tenant_name: Optional[str] = Field(None) tenant_name: Optional[str] = Field(None, regex=NAME_PATTERN)
class Config: class Config:
alias_generator = attribute_to_camel_case alias_generator = attribute_to_camel_case
class CreateProjectSchema(BaseModel): class CreateProjectSchema(BaseModel):
name: str = Field(default="my first project") name: str = Field(default="my first project", regex=NAME_PATTERN)
_transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace) _transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace)
@ -88,14 +90,14 @@ class CurrentContext(CurrentAPIContext):
class AddCollaborationSchema(BaseModel): class AddCollaborationSchema(BaseModel):
name: str = Field(...) name: str = Field(..., regex=NAME_PATTERN)
url: HttpUrl = Field(...) url: HttpUrl = Field(...)
_transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace) _transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace)
_transform_url = validator('url', pre=True, allow_reuse=True)(remove_whitespace) _transform_url = validator('url', pre=True, allow_reuse=True)(remove_whitespace)
class EditCollaborationSchema(AddCollaborationSchema): class EditCollaborationSchema(AddCollaborationSchema):
name: Optional[str] = Field(None) name: Optional[str] = Field(None, regex=NAME_PATTERN)
class CreateNotificationSchema(BaseModel): class CreateNotificationSchema(BaseModel):
@ -138,13 +140,13 @@ class CreateEditWebhookSchema(BaseModel):
webhookId: Optional[int] = Field(None) webhookId: Optional[int] = Field(None)
endpoint: str = Field(...) endpoint: str = Field(...)
authHeader: Optional[str] = Field(None) authHeader: Optional[str] = Field(None)
name: Optional[str] = Field(...) name: Optional[str] = Field(..., regex=NAME_PATTERN)
_transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace) _transform_name = validator('name', pre=True, allow_reuse=True)(remove_whitespace)
class CreateMemberSchema(BaseModel): class CreateMemberSchema(BaseModel):
userId: Optional[int] = Field(None) userId: Optional[int] = Field(None)
name: str = Field(...) name: str = Field(..., regex=NAME_PATTERN)
email: EmailStr = Field(...) email: EmailStr = Field(...)
admin: bool = Field(False) admin: bool = Field(False)
@ -153,7 +155,7 @@ class CreateMemberSchema(BaseModel):
class EditMemberSchema(BaseModel): class EditMemberSchema(BaseModel):
name: str = Field(...) name: str = Field(..., regex=NAME_PATTERN)
email: EmailStr = Field(...) email: EmailStr = Field(...)
admin: bool = Field(False) admin: bool = Field(False)
@ -328,7 +330,7 @@ class AlertDetectionMethod(str, Enum):
class AlertSchema(BaseModel): class AlertSchema(BaseModel):
name: str = Field(...) name: str = Field(..., regex=NAME_PATTERN)
detection_method: AlertDetectionMethod = Field(...) detection_method: AlertDetectionMethod = Field(...)
change: Optional[AlertDetectionType] = Field(default=AlertDetectionType.change) change: Optional[AlertDetectionType] = Field(default=AlertDetectionType.change)
description: Optional[str] = Field(None) description: Optional[str] = Field(None)
@ -1036,11 +1038,11 @@ class __CardSchema(BaseModel):
class CardSchema(__CardSchema, CardChartSchema): class CardSchema(__CardSchema, CardChartSchema):
view_type: Union[MetricTimeseriesViewType, \ view_type: Union[MetricTimeseriesViewType, \
MetricTableViewType, MetricOtherViewType] = Field(...) MetricTableViewType, MetricOtherViewType] = Field(...)
metric_type: MetricType = Field(...) metric_type: MetricType = Field(...)
metric_of: Union[MetricOfTimeseries, MetricOfTable, MetricOfErrors, \ metric_of: Union[MetricOfTimeseries, MetricOfTable, MetricOfErrors, \
MetricOfPerformance, MetricOfResources, MetricOfWebVitals, \ MetricOfPerformance, MetricOfResources, MetricOfWebVitals, \
MetricOfClickMap] = Field(default=MetricOfTable.user_id) MetricOfClickMap] = Field(default=MetricOfTable.user_id)
metric_value: List[IssueType] = Field(default=[]) metric_value: List[IssueType] = Field(default=[])
is_template: bool = Field(default=False) is_template: bool = Field(default=False)
@ -1212,7 +1214,7 @@ class LiveSessionSearchFilterSchema(BaseModel):
type: LiveFilterType = Field(...) type: LiveFilterType = Field(...)
source: Optional[str] = Field(default=None) source: Optional[str] = Field(default=None)
operator: Literal[SearchEventOperator._is, \ operator: Literal[SearchEventOperator._is, \
SearchEventOperator._contains] = Field(default=SearchEventOperator._contains) SearchEventOperator._contains] = Field(default=SearchEventOperator._contains)
transform = root_validator(pre=True, allow_reuse=True)(transform_old_FilterType) transform = root_validator(pre=True, allow_reuse=True)(transform_old_FilterType)