From 7125cc5b50571ee4556fbd3236d2c186a6d5bfbb Mon Sep 17 00:00:00 2001
From: GotthardG <51994228+GotthardG@users.noreply.github.com>
Date: Mon, 11 Nov 2024 12:00:23 +0100
Subject: [PATCH] added datacollectionparameters as a class

---
 backend/app/schemas.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/backend/app/schemas.py b/backend/app/schemas.py
index a441edb..3088482 100644
--- a/backend/app/schemas.py
+++ b/backend/app/schemas.py
@@ -3,6 +3,42 @@ from pydantic import BaseModel, EmailStr, constr
 from datetime import date
 
 
+class DataCollectionParameters(BaseModel):
+    priority: Optional[int] = None
+    comments: Optional[str] = None
+    directory: Optional[str] = None
+    proteinname: Optional[str] = None
+    oscillation: Optional[float] = None
+    aperture: Optional[str] = None
+    exposure: Optional[float] = None
+    totalrange: Optional[int] = None
+    transmission: Optional[int] = None
+    dose: Optional[float] = None
+    targetresolution: Optional[float] = None
+    datacollectiontype: Optional[str] = None
+    processingpipeline: Optional[str] = None
+    spacegroupnumber: Optional[int] = None
+    cellparameters: Optional[str] = None
+    rescutkey: Optional[str] = None
+    rescutvalue: Optional[float] = None
+    userresolution: Optional[float] = None
+    pdbid: Optional[str] = None
+    autoprocfull: Optional[bool] = None
+    procfull: Optional[bool] = None
+    adpenabled: Optional[bool] = None
+    noano: Optional[bool] = None
+    ffcscampaign: Optional[bool] = None
+    trustedhigh: Optional[float] = None
+    autoprocextraparams: Optional[str] = None
+    chiphiangles: Optional[float] = None
+
+    class Config:
+        from_attributes = True
+
+class Results(BaseModel):
+    # Define attributes for Results here
+    pass  # Placeholder for now, should be expanded later with actual fields
+
 # Contact Person schemas
 class ContactPersonBase(BaseModel):
     firstname: str
@@ -51,6 +87,8 @@ class AddressUpdate(BaseModel):
 class Sample(BaseModel):
     id: int
     sample_name: str
+    data_collection_parameters: Optional[DataCollectionParameters] = None
+    results: Optional[Results] = None
 
     class Config:
         from_attributes = True