From e65cdb346fa50475e839169e6768a391487f545a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Karen=20Bj=C3=B8rndalen?= <karen.e.bjorndalen@uit.no> Date: Wed, 28 Apr 2021 11:52:08 +0200 Subject: [PATCH] Make Parameter.status, Parameter.valid_from, Parameter.valid_to and Parameter.description optional Also a little tidying --- setra_client/models.py | 31 +++++++++++++++---------------- tests/fixtures/params.json | 8 ++++---- 2 files changed, 19 insertions(+), 20 deletions(-) diff --git a/setra_client/models.py b/setra_client/models.py index 4cddeb3..ccda600 100644 --- a/setra_client/models.py +++ b/setra_client/models.py @@ -1,9 +1,8 @@ """Models used by the client""" import datetime import json -import typing from enum import Enum -from typing import Optional, TypeVar, List +from typing import Optional, List import pydantic @@ -69,7 +68,7 @@ class Voucher(BaseModel): voucherdate: datetime.datetime exref: Optional[str] voucherno: int - transactions: typing.List[Transaction] + transactions: List[Transaction] class Batch(BaseModel): @@ -79,11 +78,11 @@ class Batch(BaseModel): """ client: str batchid: str - period: typing.Optional[str] + period: Optional[str] interface: str - vouchertype: typing.Optional[str] - batch_progress: typing.Optional[str] - vouchers: typing.Optional[typing.List[Voucher]] + vouchertype: Optional[str] + batch_progress: Optional[str] + vouchers: Optional[List[Voucher]] class ErrorTransaction(BaseModel): @@ -114,9 +113,9 @@ class ErrorBatch(BaseModel): class BatchErrors(BaseModel): """Model for the /batch_error/<id> endpoint""" - batch_errors: typing.List[ErrorBatch] - voucher_errors: typing.List[ErrorVoucher] - transaction_errors: typing.List[ErrorTransaction] + batch_errors: List[ErrorBatch] + voucher_errors: List[ErrorVoucher] + transaction_errors: List[ErrorTransaction] class CompleteTransaction(BaseModel): @@ -152,7 +151,7 @@ class CompleteVoucher(BaseModel): voucherno_ubw: Optional[int] voucherno_ubw_wflow: Optional[int] vouchertype: Optional[str] - transactions: typing.List[CompleteTransaction] + transactions: List[CompleteTransaction] class CompleteBatch(BaseModel): @@ -176,7 +175,7 @@ class CompleteBatch(BaseModel): getresult_logg: Optional[str] getresult_report: Optional[str] batch_progress: str - vouchers: typing.List[CompleteVoucher] + vouchers: List[CompleteVoucher] class Parameter(BaseModel): @@ -187,7 +186,7 @@ class Parameter(BaseModel): mandatory: str validation: str value: str - status: str - valid_from: datetime.date - valid_to: datetime.date - description: str + status: Optional[str] + valid_from: Optional[datetime.date] + valid_to: Optional[datetime.date] + description: Optional[str] diff --git a/tests/fixtures/params.json b/tests/fixtures/params.json index bfe643c..7ea0b95 100644 --- a/tests/fixtures/params.json +++ b/tests/fixtures/params.json @@ -20,10 +20,10 @@ "mandatory": "j", "validation": "j", "value": "wat", - "status": "anerikke", - "valid_from": "2020-01-01", - "valid_to": "2020-01-01", - "description": "Min parameter b" + "status": null, + "valid_from": null, + "valid_to": null, + "description": null }, { "client": "72", -- GitLab