From b2f6a6c2e08a24806b0203d5805ec727a2e7615b Mon Sep 17 00:00:00 2001 From: Gao Date: Fri, 9 Jun 2023 19:34:37 +0200 Subject: [PATCH] not finished --- DataContainer/MongoDB.py | 11 ++++++++++- testMongoDB.ipynb | 11 ++++++++++- 2 files changed, 20 insertions(+), 2 deletions(-) diff --git a/DataContainer/MongoDB.py b/DataContainer/MongoDB.py index 72a1fa6..08c484b 100644 --- a/DataContainer/MongoDB.py +++ b/DataContainer/MongoDB.py @@ -125,7 +125,16 @@ class MongoDB: self.mongoDB['global'].update_one(filter, {"$set": data_label}, upsert=False) def add_data(self, shotNum, data, date=None, name=None, engine='normal'): - pass + if not date is None: + self.set_date(date) + + if engine == 'normal': + self._add_data_normal(shotNum=shotNum, data=data) + elif engine == 'xarray': + if isinstance(dataSet, type(xr.Dataset())): + self._add_data_xarray_dataSet(shotNum=shotNum, data=data, name=name) + else: + self._add_data_xarray_dataArray(shotNum=shotNum, data=data, name=name) def read_data(self, shotNum, data=None): pass \ No newline at end of file diff --git a/testMongoDB.ipynb b/testMongoDB.ipynb index 5aae7f7..2264b97 100644 --- a/testMongoDB.ipynb +++ b/testMongoDB.ipynb @@ -2088,7 +2088,16 @@ " self.mongoDB['global'].update_one(filter, {\"$set\": data_label}, upsert=False)\n", " \n", " def add_data(self, shotNum, data, date=None, name=None, engine='normal'):\n", - " pass\n", + " if not date is None:\n", + " self.set_date(date)\n", + " \n", + " if engine == 'normal':\n", + " self._add_data_normal(shotNum=shotNum, data=data)\n", + " elif engine == 'xarray':\n", + " if isinstance(dataSet, type(xr.Dataset())):\n", + " self._add_data_xarray_dataSet(shotNum=shotNum, data=data, name=name)\n", + " else:\n", + " self._add_data_xarray_dataArray(shotNum=shotNum, data=data, name=name)\n", " \n", " def read_data(self, shotNum, data=None):\n", " pass\n",