Add short-hand "bulk_upsert"

Author: Photonios

psqlextra/manager.py

@@ -253,6 +253,21 @@ def upsert_and_get(self, conflict_target: List, fields: Dict):
         self.on_conflict(conflict_target, ConflictAction.UPDATE)
         return self.insert_and_get(**fields)
 
+    def bulk_upsert(self, conflict_target: List, rows: List[Dict]):
+        """Creates a set of new records or updates the existing
+        ones with the specified data.
+
+        Arguments:
+            conflict_target:
+                Fields to pass into the ON CONFLICT clause.
+
+            rows:
+                Rows to upsert.
+        """
+
+        self.on_conflict(conflict_target, ConflictAction.UPDATE)
+        return self.bulk_insert(rows)
+
     def _build_insert_compiler(self, rows: List[Dict]):
         """Builds the SQL compiler for a insert query.
 
@@ -484,6 +499,20 @@ def upsert_and_get(self, conflict_target: List, fields: Dict):
 
         return self.get_queryset().upsert_and_get(conflict_target, fields)
 
+    def bulk_upsert(self, conflict_target: List, rows: List[Dict]):
+        """Creates a set of new records or updates the existing
+        ones with the specified data.
+
+        Arguments:
+            conflict_target:
+                Fields to pass into the ON CONFLICT clause.
+
+            rows:
+                Rows to upsert.
+        """
+
+        return self.get_queryset().bulk_upsert(conflict_target, fields)
+
     @staticmethod
     def _on_model_save(sender, **kwargs):
         """When a model gets created or updated."""