Deletion of multiple records (#172)

* Deletion of multiple records

* PR comments

Author: JWittmeyer

business_objects/comments.py

@@ -4,7 +4,7 @@
 from . import general, organization
 from .. import enums
 from ..session import session
-from typing import Dict, List, Any, Optional, Union
+from typing import Dict, List, Any, Optional, Union, Iterable
 from ..util import prevent_sql_injection
 
 
@@ -305,6 +305,21 @@ def remove(comment_id: str, with_commit: bool = False) -> None:
     general.flush_or_commit(with_commit)
 
 
+def delete_by_type_and_xfkey(
+    project_id: str,
+    xfkeys: Iterable[str],
+    xftype: enums.CommentCategory,
+    with_commit: bool = False,
+) -> None:
+
+    session.query(CommentData).filter(
+        CommentData.project_id == project_id,
+        CommentData.xfkey.in_(xfkeys),
+        CommentData.xftype == xftype.value,
+    ).delete()
+    general.flush_or_commit(with_commit)
+
+
 def get_unique_comments_keys_for(
     xftype: enums.CommentCategory, project_id: Optional[str] = None
 ) -> List[str]:

business_objects/record.py

@@ -773,6 +773,18 @@ def delete(project_id: str, record_id: str, with_commit: bool = False) -> None:
     general.flush_or_commit(with_commit)
 
 
+def delete_many(
+    project_id: str, record_ids: Iterable[str], with_commit: bool = False
+) -> int:
+    res = (
+        session.query(Record)
+        .filter(Record.project_id == project_id, Record.id.in_(record_ids))
+        .delete()
+    )
+    general.flush_or_commit(with_commit)
+    return res
+
+
 def delete_all(project_id: str, with_commit: bool = False) -> None:
     session.query(Record).filter(Record.project_id == project_id).delete()
     general.flush_or_commit(with_commit)