Fix people usage (#393)

Author: DavidMStraub

gramps_webapi/api/resources/objects.py

@@ -37,6 +37,7 @@
     get_db_handle,
     get_search_indexer,
     get_tree_from_jwt,
+    update_usage_people,
 )
 from . import ProtectedResource
 from .util import add_object, fix_object_dict, transaction_to_json, validate_object_dict
@@ -82,6 +83,8 @@ def post(self) -> Response:
                 except ValueError:
                     abort_with_message(400, "Error while adding object")
             trans_dict = transaction_to_json(trans)
+        if number_new_people:
+            update_usage_people()
         # update search index
         tree = get_tree_from_jwt()
         indexer: SearchIndexer = get_search_indexer(tree)

gramps_webapi/api/resources/transactions.py

@@ -40,6 +40,7 @@
     get_db_handle,
     get_search_indexer,
     get_tree_from_jwt,
+    update_usage_people,
     use_args,
 )
 from . import ProtectedResource
@@ -98,6 +99,8 @@ def post(self, args) -> Response:
                 except (KeyError, UnicodeDecodeError, json.JSONDecodeError, TypeError):
                     abort_with_message(400, "Error while processing transaction")
             trans_dict = transaction_to_json(trans)
+        if num_people_new:
+            update_usage_people()
         # update search index
         tree = get_tree_from_jwt()
         indexer: SearchIndexer = get_search_indexer(tree)

gramps_webapi/api/tasks.py

@@ -40,6 +40,7 @@
     get_db_outside_request,
     get_search_indexer,
     send_email,
+    update_usage_people,
 )
 
 
@@ -135,6 +136,7 @@ def import_file(tree: str, file_name: str, extension: str, delete: bool = True):
         extension=extension.lower(),
         delete=delete,
     )
+    update_usage_people(tree=tree)
     _search_reindex_incremental(tree)
 
 
@@ -206,7 +208,6 @@ def export_media(tree: str, view_private: bool) -> Dict[str, Union[str, int]]:
 @shared_task()
 def import_media_archive(tree: str, file_name: str, delete: bool = True):
     """Import a media archive."""
-    # check_quota_people(to_add=object_counts["people"], tree=tree)
     db_handle = get_db_outside_request(tree=tree, view_private=True, readonly=True)
     result = run_import_media_archive(
         tree=tree,