From e4aecbe1bc8599ab4f9a7e97c4a12aebb04f23ff Mon Sep 17 00:00:00 2001
From: Nathan Thorpe <nathan@techstormpc.com>
Date: Thu, 20 Apr 2023 10:00:20 -0700
Subject: [PATCH] add created by field

---
 cirro/api/models/dataset.py   | 4 +++-
 cirro/api/services/dataset.py | 2 ++
 cirro/sdk/dataset.py          | 1 +
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/cirro/api/models/dataset.py b/cirro/api/models/dataset.py
index a9261322..29905bf8 100644
--- a/cirro/api/models/dataset.py
+++ b/cirro/api/models/dataset.py
@@ -42,6 +42,7 @@ class Dataset:
     info: Dict[str, Any]
     params: Dict[str, Any]
     created_at: datetime
+    created_by: str
 
     @classmethod
     def from_record(cls, record):
@@ -57,7 +58,8 @@ def from_record(cls, record):
             record.get('sourceDatasets'),
             safe_load_json(record.get('infoJson')),
             safe_load_json(record.get('paramJson')),
-            parse_json_date(record.get('createdAt'))
+            parse_json_date(record.get('createdAt')),
+            record.get('createdBy')
         )
 
     @property
diff --git a/cirro/api/services/dataset.py b/cirro/api/services/dataset.py
index dcbe4f1b..7e46e50f 100644
--- a/cirro/api/services/dataset.py
+++ b/cirro/api/services/dataset.py
@@ -44,6 +44,7 @@ def find_by_project(self, project_id: str, name: str = None) -> List[Dataset]:
                 infoJson
                 paramJson
                 process
+                createdBy
                 createdAt
                 updatedAt
               }
@@ -79,6 +80,7 @@ def get_from_id(self, _id: str) -> Optional[Dataset]:
                   infoJson
                   process
                   project
+                  createdBy
                   createdAt
                   updatedAt
                 }
diff --git a/cirro/sdk/dataset.py b/cirro/sdk/dataset.py
index 63de8d21..f8f4ee1a 100644
--- a/cirro/sdk/dataset.py
+++ b/cirro/sdk/dataset.py
@@ -30,6 +30,7 @@ def __init__(self, dataset: Dataset, client: DataPortalClient):
         self.info = dataset.info
         self.params = dataset.params
         self.created_at = dataset.created_at
+        self.created_by = dataset.created_by
         self._client = client
 
     def __str__(self):