MPComplete endpoint now uses new sub data resource

src/mp_api/routes/mpcomplete/models.py

@@ -1,9 +1,10 @@
 from pydantic import Field
+from pydantic.main import BaseModel
 from pymatgen.core.structure import Structure
-from maggma.api.models import UserSubmissionDataModel
+from enum import Enum
 
 
-class MPCompleteDoc(UserSubmissionDataModel):
+class MPCompleteDoc(BaseModel):
     """
     Defines data for MPComplete structure submissions
     """
@@ -21,3 +22,15 @@ class MPCompleteDoc(UserSubmissionDataModel):
     public_email: str = Field(
         None, title="Public email", description="Public email of submitter.",
     )
+
+
+class MPCompleteDataStatus(Enum):
+    """
+    Submission status for MPComplete data
+    """
+
+    submitted = "SUBMITTED"
+    pending = "PENDING"
+    running = "RUNNING"
+    error = "ERROR"
+    complete = "COMPLETE"

src/mp_api/routes/mpcomplete/resources.py

@@ -1,19 +1,22 @@
-from maggma.api.resource import UserSubmissionResource
+from maggma.api.resource import SubmissionResource
 from maggma.api.query_operator import PaginationQuery
-from mp_api.routes.mpcomplete.models import MPCompleteDoc
+from mp_api.routes.mpcomplete.models import MPCompleteDoc, MPCompleteDataStatus
 from mp_api.routes.mpcomplete.query_operator import (
     MPCompletePostQuery,
     MPCompleteGetQuery,
 )
 
 
 def mpcomplete_resource(mpcomplete_store):
-    resource = UserSubmissionResource(
+    resource = SubmissionResource(
         mpcomplete_store,
         MPCompleteDoc,
         post_query_operators=[MPCompletePostQuery()],
         get_query_operators=[MPCompleteGetQuery(), PaginationQuery()],
         tags=["MPComplete"],
+        state_enum=MPCompleteDataStatus,
+        default_state=MPCompleteDataStatus.submitted.value,
+        calculate_submission_id=True,
         include_in_schema=True,
     )