No extra params

Fixes #58 (closed)

geomagio/api/ws/DataApiQuery.py

@@ -4,7 +4,7 @@ import os
 from typing import Dict, List, Union
 
 from obspy import UTCDateTime
-from pydantic import BaseModel, root_validator, validator
+from pydantic import BaseModel, Extra, root_validator, validator
 
 from ... import pydantic_utcdatetime
 from .Element import ELEMENTS
@@ -46,19 +46,20 @@ class DataApiQuery(BaseModel):
     endtime: UTCDateTime = None
     elements: List[str] = DEFAULT_ELEMENTS
     sampling_period: SamplingPeriod = SamplingPeriod.MINUTE
-    data_type: Union[DataType, str] = DataType.VARIATION
+    type: Union[DataType, str] = DataType.VARIATION
     format: OutputFormat = OutputFormat.IAGA2002
 
-    @validator("data_type")
-    def validate_data_type(
-        cls, data_type: Union[DataType, str]
-    ) -> Union[DataType, str]:
-        if data_type not in DataType.values() and len(data_type) != 2:
+    class Config:
+        extra = Extra.forbid
+
+    @validator("type")
+    def validate_data_type(cls, type: Union[DataType, str]) -> Union[DataType, str]:
+        if type not in DataType.values() and len(type) != 2:
             raise ValueError(
-                f"Bad data type value '{data_type}'."
+                f"Bad data type value '{type}'."
                 f" Valid values are: {', '.join(DataType.values())}"
             )
-        return data_type
+        return type
 
     @validator("elements", pre=True, always=True)
     def validate_elements(cls, elements: List[str]) -> List[str]: