SNL ID added to mpcomplete endpoint

app.py

@@ -277,7 +277,7 @@
     mpcomplete_store = MongoURIStore(
         uri=f"mongodb+srv://{db_uri}",
         database="mp_consumers",
-        key="structure",
+        key="snl_id",
         collection_name="mpcomplete",
     )
 

src/mp_api/mpcomplete/models.py

@@ -7,6 +7,10 @@ class MPCompleteDoc(BaseModel):
     Defines data for MPComplete structure submissions
     """
 
+    snl_id: str = Field(
+        None, title="SNL ID", description="ID for the submission.",
+    )
+
     structure: Structure = Field(
         None,
         title="Submitted structure",

src/mp_api/mpcomplete/query_operator.py

@@ -3,6 +3,7 @@
 from pymatgen.core.structure import Structure
 from mp_api.core.utils import STORE_PARAMS
 from mp_api.core.query_operator import QueryOperator
+from uuid import uuid4
 
 
 class MPCompletePostQuery(QueryOperator):
@@ -20,8 +21,10 @@ def query(
         self.public_name = public_name
         self.public_email = public_email
         self.comment = comment
+        self.snl_id = str(uuid4())
 
         crit = {
+            "snl_id": self.snl_id,
             "structure": structure,
             "public_email": public_email,
             "public_name": public_name,
@@ -34,6 +37,7 @@ def post_process(self, written):
 
         d = [
             {
+                "snl_id": self.snl_id,
                 "structure": self.structure,
                 "public_email": self.public_email,
                 "public_name": self.public_name,

src/mp_api/thermo/client.py

@@ -1,11 +1,6 @@
-from typing import List, Optional, Tuple
-from collections import defaultdict
-
 from mp_api.core.client import BaseRester
 from mp_api.thermo.models import ThermoDoc
 
-import warnings
-
 
 class ThermoRester(BaseRester):