mss-python 0.1.1__tar.gz → 0.1.3__tar.gz

{mss_python-0.1.1 → mss_python-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mss-python
-Version: 0.1.1
+Version: 0.1.3
 Summary: MSS API wrapper for Python projects
 Home-page: https://github.com/bookingsuedtirol/mss-python
 Requires-Python: >=3.7, <4

{mss_python-0.1.1 → mss_python-0.1.3}/setup.py

@@ -36,7 +36,7 @@ setup(
     # For a discussion on single-sourcing the version across setup.py and the
     # project code, see
     # https://packaging.python.org/guides/single-sourcing-package-version/
-    version="0.1.1",  # Required
+    version="0.1.3",  # Required
     # This is a one-line description or tagline of what your project does. This
     # corresponds to the "Summary" metadata field:
     # https://packaging.python.org/specifications/core-metadata/#summary

mss_python-0.1.3/src/mss_python/client.py

@@ -0,0 +1,111 @@
+import requests
+import xmltodict
+import xml.etree.ElementTree as ET
+
+from .request.methods import Method
+from .request.types_mss import Root, BaseType, Credentials, Search, MethodName, Literal
+
+
+# maybe recursion should be used instead of doing if checks?
+def change_structure(root: Root, child: BaseType):
+    if child.tag == "root":
+        return child
+
+    if child.tag == "request":
+        root.request = child
+
+    elif child.tag == "search":
+        root.request.search = child
+
+    elif child.tag == "order":
+        root.request.order = child
+
+    elif child.tag == "options":
+        root.request.options = child
+
+    elif child.tag == "logging":
+        root.request.logging = child
+
+    elif child.tag == "header":
+        root.header = child
+
+    return root
+
+
+class Client:
+    def __init__(
+        self, cred: Credentials, lang: list, elements: list | None = None
+    ) -> None:
+        self.cred = cred
+        self.lang = Search(lang)
+
+    def request(
+        self,
+        url: str,
+        method_name: MethodName,
+        data: BaseType | None = None,
+        _print: Literal[True, False] = False,
+    ) -> dict:
+        """
+        Makes a post request to the given URL with the specified data and method name.
+
+        data: ElementTree with the data required in the XML
+        url: Service URL
+        method: Method to be executed by the service
+        """
+
+        method = Method(method_name)
+
+        root = method.get_base_xml(self.cred, self.lang)
+
+        if data is None:
+            root = change_structure(root, data)
+
+        if _print is True:
+            print(ET.tostring(root.request.to_xml(), encoding="unicode"))
+
+        response = requests.post(
+            url,
+            data=ET.tostring(root.to_xml(), encoding="unicode"),
+            headers={"Content-Type": "application/xml"},
+        )
+
+        responseRoot = xmltodict.parse(response.content)
+        error = responseRoot["root"]["header"]["error"]
+        if int(error["code"]) > 0:
+            raise Exception(
+                f'MSS returned an error: code {error["code"]}, message: "{error["message"]}"'
+            )
+
+        return refine_response(responseRoot["root"], method_name)
+
+
+def refine_response(resp: dict, meth: MethodName) -> dict:
+    if meth == MethodName.GetHotelList:
+        return refine_getHotelList(resp)
+
+    return resp
+
+
+def refine_getHotelList(resp: dict) -> dict:
+    if resp["result"] is None:
+        return resp
+    elif "hotel" not in resp["result"] or resp["result"]["hotel"] is None:
+        return resp
+    elif resp["result"]["hotel"] is list:
+        resp["result"]["hotel"] = [resp["result"]["hotel"]]
+
+    for hotel in resp["result"]["hotel"]:
+        ensure_list_value(hotel, "pictures", "picture")
+        ensure_list_value(hotel, "gallery", "picture")
+        ensure_list_value(hotel, "feature_view", "feature")
+        ensure_list_value(hotel, "pos", "id_pos")
+
+    return resp
+
+
+def ensure_list_value(hotel: dict, parent_fld: str, child_fld: str) -> None:
+    if parent_fld not in hotel:
+        hotel[parent_fld] = {child_fld: []}
+    elif hotel[parent_fld][child_fld] is list:
+        hotel[parent_fld][child_fld] = [hotel[parent_fld][child_fld]]

mss_python-0.1.1/src/request/methods.py → mss_python-0.1.3/src/mss_python/request/method_names.py

@@ -1,6 +1,4 @@
-from __future__ import annotations
-import xml.etree.ElementTree as ET
-from request.types_mss import *  # might need change
+from enum import Enum
 
 
 class MethodName(Enum):
@@ -28,22 +26,3 @@ class MethodName(Enum):
     GetHotelPictureGroups = "getHotelPictureGroups"
     GetOptionalServices = "getOptionalServices"
     ValidateCoupon = "validateCoupon"
-
-
-class Method:
-    def __init__(self, name: MethodName):
-        self.name = name
-
-    def get_base_xml(self, cred: Credentials, lang: Search) -> Root:
-        """
-        Returns an XML (ElementTree) with the mandatory elements needed for the request.
-
-        cred: XML Credentials Element (username, password, source)
-
-        lang: XML Search Element with the language element defined
-        """
-
-        header = Header(cred, self.name)
-        req = Request(lang)
-        root = Root(header, req)
-        return root

mss_python-0.1.3/src/mss_python/request/methods.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+from .method_names import MethodName
+from .types_mss import Credentials, Search, Root, Header, Request
+
+
+class Method:
+    def __init__(self, name: MethodName):
+        self.name = name
+
+    def get_base_xml(self, cred: Credentials, lang: Search) -> Root:
+        """
+        Returns an XML (ElementTree) with the mandatory elements needed for the request.
+
+        cred: XML Credentials Element (username, password, source)
+
+        lang: XML Search Element with the language element defined
+        """
+
+        header = Header(cred, self.name)
+        req = Request(lang)
+        root = Root(header, req)
+        return root

{mss_python-0.1.1/src → mss_python-0.1.3/src/mss_python}/request/types_mss.py

@@ -4,7 +4,8 @@ from enum import Enum, IntFlag
 from typing import Literal
 import xml.etree.ElementTree as ET
 from abc import ABC
-from methods import MethodName
+
+from .method_names import MethodName
 
 
 def get_child_xml(value, field, root):
@@ -35,8 +36,8 @@ class BaseType(ABC):
         root = ET.Element(self.tag)
         # root.append(self.search.to_xml())
 
-        for field, value in vars(self).items():
-            get_child_xml(value, field, root)
+        for field_, value in vars(self).items():
+            get_child_xml(value, field_, root)
 
         return root
 
@@ -624,7 +625,7 @@ class SpecialTheme(Enum):
     ThemeIDBicycleTouring = 86
     ThemeIDEBike = 87
 
-    ##These values were theme_bit
+    ## These values were theme_bit
     # Hiking = 1
     # Cycling = 2
     # Family = 4

{mss_python-0.1.1 → mss_python-0.1.3}/src/mss_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mss-python
-Version: 0.1.1
+Version: 0.1.3
 Summary: MSS API wrapper for Python projects
 Home-page: https://github.com/bookingsuedtirol/mss-python
 Requires-Python: >=3.7, <4

mss_python-0.1.3/src/mss_python.egg-info/SOURCES.txt

@@ -0,0 +1,13 @@
+LICENSE
+README.md
+setup.py
+src/mss_python/__init__.py
+src/mss_python/client.py
+src/mss_python.egg-info/PKG-INFO
+src/mss_python.egg-info/SOURCES.txt
+src/mss_python.egg-info/dependency_links.txt
+src/mss_python.egg-info/top_level.txt
+src/mss_python/request/__init__.py
+src/mss_python/request/method_names.py
+src/mss_python/request/methods.py
+src/mss_python/request/types_mss.py

mss_python-0.1.3/src/mss_python.egg-info/top_level.txt

@@ -0,0 +1 @@
+mss_python

mss_python-0.1.1/src/mss_python.egg-info/SOURCES.txt

@@ -1,10 +0,0 @@
-LICENSE
-README.md
-setup.py
-src/mss_python.egg-info/PKG-INFO
-src/mss_python.egg-info/SOURCES.txt
-src/mss_python.egg-info/dependency_links.txt
-src/mss_python.egg-info/top_level.txt
-src/request/__init__.py
-src/request/methods.py
-src/request/types_mss.py

mss_python-0.1.1/src/mss_python.egg-info/top_level.txt

@@ -1 +0,0 @@
-request