handle 1 record delete and strip @OData rows in get_dataframe returns

Author: Max Wang

src/PowerPlatform/Dataverse/client.py

@@ -13,7 +13,7 @@
 from .core._auth import _AuthManager
 from .core.config import DataverseConfig
 from .data._odata import _ODataClient
-from .utils._pandas import dataframe_to_records
+from .utils._pandas import dataframe_to_records, strip_odata_keys
 
 
 class DataverseClient:
@@ -425,7 +425,7 @@ def get_dataframe(
                 record_id=record_id,
                 select=select,
             )
-            return pd.DataFrame([result])
+            return pd.DataFrame([strip_odata_keys(result)])
 
         def _paged_df() -> Iterable[pd.DataFrame]:
             for batch in self.get(
@@ -437,7 +437,7 @@ def _paged_df() -> Iterable[pd.DataFrame]:
                 expand=expand,
                 page_size=page_size,
             ):
-                yield pd.DataFrame(batch)
+                yield pd.DataFrame([strip_odata_keys(row) for row in batch])
 
         return _paged_df()
 
@@ -563,7 +563,11 @@ def delete_dataframe(
         if not isinstance(ids, pd.Series):
             raise TypeError("ids must be a pandas Series")
 
-        return self.delete(table_schema_name, ids.tolist(), use_bulk_delete=use_bulk_delete)
+        id_list = ids.tolist()
+        if len(id_list) == 1:
+            return self.delete(table_schema_name, id_list[0])
+        else:
+            return self.delete(table_schema_name, id_list, use_bulk_delete=use_bulk_delete)
 
     # SQL via Web API sql parameter
     def query_sql(self, sql: str):

src/PowerPlatform/Dataverse/utils/_pandas.py

@@ -22,3 +22,8 @@ def dataframe_to_records(df: pd.DataFrame) -> List[Dict[str, Any]]:
                 clean[k] = None
         records.append(clean)
     return records
+
+
+def strip_odata_keys(record: Dict[str, Any]) -> Dict[str, Any]:
+    """Remove OData metadata keys (keys containing '@') from a record dict."""
+    return {k: v for k, v in record.items() if "@" not in k}