diff --git a/fuzzer/datatypes.py b/fuzzer/datatypes.py index cac8113..1f1981b 100644 --- a/fuzzer/datatypes.py +++ b/fuzzer/datatypes.py @@ -25,6 +25,9 @@ class Order: invoice_amount: int order_date: str # Assuming timestamp as a string for simplicity + def dict(self): + return {k: str(v) for k, v in asdict(self).items()} + @dataclass class Ingredient: @@ -37,6 +40,9 @@ class Ingredient: calories: int protein: int + def dict(self): + return {k: str(v) for k, v in asdict(self).items()} + @dataclass class Supplier: @@ -49,6 +55,9 @@ class Supplier: house_number: str postal_code: int + def dict(self): + return {k: str(v) for k, v in asdict(self).items()} + @dataclass class SupplierContainsIngredients: @@ -57,9 +66,16 @@ class SupplierContainsIngredients: delivery_cost: int delivery_time: int + def dict(self): + return {k: str(v) for k, v in asdict(self).items()} + @dataclass class OrderContainsIngredients: order: Order ingredient: Ingredient quantity: int + + def dict(self): + return {k: str(v) for k, v in asdict(self).items()} +