0.0.8 add upsert

Author: davidkhala

frame/davidkhala/data/frame/pandas/__init__.py

@@ -0,0 +1,3 @@
+def upsert(df, primary_key:str, record:dict):
+    df.loc[record[primary_key]] = record
+    return df

frame/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "davidkhala.data.frame"
-version = "0.0.7"
+version = "0.0.8"
 description = ""
 authors = [{ name = "David Liu", email = "david-khala@hotmail.com" }]
 requires-python = ">=3.10"

frame/tests/pandas_test.py

@@ -15,6 +15,21 @@ def test_duckDB(self):
         conn.register("airports", df_airports)
         conn.close()
 
+class SyntaxTestCase(unittest.TestCase):
+    def test_upsert(self):
+        from davidkhala.data.frame.pandas import upsert
+        prim_key ='id'
+        df = pandas.DataFrame({
+            prim_key: [1, 2],
+            'name': ['Alice', 'Bob'],
+            'score': [85, 90]
+        }).set_index(prim_key)
+        new_record = {'name': 'Charlie', 'score': 95, prim_key:2}
+        upsert(df, prim_key, new_record)
+
+        self.assertEqual(95, df.at[2, 'score'])
+
+
 
 
 if __name__ == '__main__':