1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
from pydantic import Field
from pydantic.main import BaseModel
from pymatgen.core.structure import Structure
from enum import Enum
from typing import Optional
class MPCompleteDoc(BaseModel):
"""
Defines data for MPComplete structure submissions
"""
structure: Optional[Structure] = Field(
None,
description="Structure submitted by the user.",
)
public_name: Optional[str] = Field(
None,
description="Public name of submitter.",
)
public_email: Optional[str] = Field(
None,
description="Public email of submitter.",
)
class MPCompleteDataStatus(Enum):
"""
Submission status for MPComplete data
"""
submitted = "SUBMITTED"
pending = "PENDING"
running = "RUNNING"
error = "ERROR"
complete = "COMPLETE"
|
