Add messages client and 400,404 warnings

Author: Jason

mp_api/client/core/client.py

@@ -694,6 +694,11 @@ def _submit_request_and_process(
                 f"REST query timed out on URL {url}. Try again with a smaller request."
             )
 
+        if response.status_code in [400, 404]:
+            warnings.warn(
+                f"The server does not support the request made to {response.url}. This may be due to an outdated mp-api package, or a problem with the query."
+            )
+
         if response.status_code == 200:
             if self.monty_decode:
                 data = json.loads(response.text, cls=MontyDecoder)

mp_api/client/mprester.py

@@ -23,10 +23,7 @@
 from mp_api.client.core import BaseRester, MPRestError
 from mp_api.client.core.settings import MAPIClientSettings
 from mp_api.client.core.utils import validate_ids
-from mp_api.client.routes import (
-    GeneralStoreRester,
-    UserSettingsRester,
-)
+from mp_api.client.routes import GeneralStoreRester, MessagesRester, UserSettingsRester
 from mp_api.client.routes.legacy import LegacyMoleculesRester
 from mp_api.client.routes.materials import (
     AbsorptionRester,
@@ -148,6 +145,7 @@ class MPRester:
     doi: DOIRester
     _user_settings: UserSettingsRester
     _general_store: GeneralStoreRester
+    _messages: MessagesRester
 
     def __init__(
         self,

mp_api/client/routes/__init__.py

@@ -1,2 +1,3 @@
 from ._general_store import GeneralStoreRester
+from ._messages import MessagesRester
 from ._user_settings import UserSettingsRester

mp_api/client/routes/_messages.py

@@ -0,0 +1,53 @@
+from datetime import datetime
+from typing import List
+
+from emmet.core._messages import MessagesDoc, MessageType
+
+from mp_api.client.core import BaseRester
+
+
+class MessagesRester(BaseRester[MessagesDoc]):  # pragma: no cover
+    suffix = "_messages"
+    document_model = MessagesDoc  # type: ignore
+    primary_key = "title"
+    monty_decode = False
+    use_document_model = False
+
+    def set_message(
+        self,
+        title: str,
+        body: str,
+        type: MessageType = MessageType.generic,
+        authors: List[str] = [],
+    ):  # pragma: no cover
+        """Set user settings
+
+        Args:
+            title: Message title
+            body: Message text body
+            type: Message type
+            authors: Message authors
+        Returns:
+            Dictionary with updated message data
+
+
+        Raises:
+            MPRestError.
+        """
+        d = {"title": title, "body": body, "type": type.value, "authors": authors}
+
+        return self._post_resource(body=d).get("data")
+
+    def get_messages(self, last_updated: datetime):  # pragma: no cover
+        """Get user settings.
+
+        Args:
+            last_updated: Datetime to use to query for newer messages
+        Returns:
+            Dictionary with messages data
+
+
+        Raises:
+            MPRestError.
+        """
+        return self._search(last_updated=last_updated)