mss-python 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

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]]

request/methods.py → 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/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

{request → 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.dist-info → mss_python-0.1.3.dist-info}/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,11 @@
+mss_python/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mss_python/client.py,sha256=DEMAGATE08MW42Ldtepk95RV6funmyyuFbN9XFhiwPk,3202
+mss_python/request/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mss_python/request/method_names.py,sha256=DP22Tra-F86FP-6771q7bDwuX6ojPCidFPzztl4UThI,1030
+mss_python/request/methods.py,sha256=pu_U6z8ANktuv0r3P6MmLKW66fwRhiDCp261vNxqrYE,664
+mss_python/request/types_mss.py,sha256=Xq9YYA5-f_j2Gt7bCcHSC6qEzHs0hswpoXm2IGB9SdA,22354
+mss_python-0.1.3.dist-info/LICENSE,sha256=SxLvAIuAvi2G5v6pRCopG3gAH9SLL_-xACJ0QBz_Blw,1074
+mss_python-0.1.3.dist-info/METADATA,sha256=QHC1KmbqxFdqD8QY-9HHEkc692274XYUHh-AjhUQ1X0,1642
+mss_python-0.1.3.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
+mss_python-0.1.3.dist-info/top_level.txt,sha256=-5Wm4NpowVgq9ImH61vhAiBQFoE21fg4r-7Pn7mOYyE,11
+mss_python-0.1.3.dist-info/RECORD,,

mss_python-0.1.3.dist-info/top_level.txt

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

mss_python-0.1.1.dist-info/RECORD

@@ -1,8 +0,0 @@
-request/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-request/methods.py,sha256=ZJfqxUTxGbWwos4eIswntJflZU-ESsMsk1esbzuMOp0,1656
-request/types_mss.py,sha256=fDz9sAkjm3VRtVOuE-zZVBkVQXg_UW5mG1MT5kYVtNA,22344
-mss_python-0.1.1.dist-info/LICENSE,sha256=SxLvAIuAvi2G5v6pRCopG3gAH9SLL_-xACJ0QBz_Blw,1074
-mss_python-0.1.1.dist-info/METADATA,sha256=uN3QRyZLQ0eIDSijzLmVF2R6JKkZO8WA1Y96YVufvAY,1642
-mss_python-0.1.1.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
-mss_python-0.1.1.dist-info/top_level.txt,sha256=JDL91wFJSRflnNxOORJn3CuoEnpFVsBerOlmjdGdTEk,8
-mss_python-0.1.1.dist-info/RECORD,,

mss_python-0.1.1.dist-info/top_level.txt

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