mbu-dev-shared-components 4.0.3__tar.gz → 4.2.0__tar.gz

{mbu_dev_shared_components-4.0.3 → mbu_dev_shared_components-4.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mbu_dev_shared_components
-Version: 4.0.3
+Version: 4.2.0
 Summary: Shared components to use in RPA projects
 Author-email: MBU <rpa@mbu.aarhus.dk>
 License-Expression: MIT
@@ -12,6 +12,7 @@ License-File: LICENSE
 Requires-Dist: cryptography>=43.0.0
 Requires-Dist: pyodbc>=5.1.0
 Requires-Dist: python-dateutil==2.9.*
+Requires-Dist: dotenv
 Provides-Extra: getorganized
 Requires-Dist: requests_ntlm>=1.2.0; extra == "getorganized"
 Provides-Extra: google
@@ -31,6 +32,8 @@ Provides-Extra: sap
 Provides-Extra: solteqtand
 Requires-Dist: uiautomation; extra == "solteqtand"
 Requires-Dist: pyodbc>=5.1.0; extra == "solteqtand"
+Requires-Dist: psutil; extra == "solteqtand"
+Requires-Dist: docx2pdf; extra == "solteqtand"
 Provides-Extra: utils
 Requires-Dist: python-dateutil==2.9.*; extra == "utils"
 Requires-Dist: cryptography>=43.0.0; extra == "utils"

{mbu_dev_shared_components-4.0.3 → mbu_dev_shared_components-4.2.0}/mbu_dev_shared_components/database/utility.py

@@ -1,15 +1,17 @@
 """This module handles general database connection and calls"""
 
-import os
 import json
-from typing import Dict, Union, Tuple, Any
-from dateutil import parser
+import os
+from typing import Any, Dict, Tuple, Union
+
 import pyodbc
-from dotenv import load_dotenv 
+from dateutil import parser
+from dotenv import load_dotenv
 
 
 class Utility:
     """Base class handling general utilities"""
+
     def connect_to_db(self, autocommit=True, db_env="PROD") -> pyodbc.Connection:
         """Establish connection to sql database
 
@@ -22,17 +24,24 @@ class Utility:
         rpa_conn = pyodbc.connect(rpa_conn_string, autocommit=autocommit)
         return rpa_conn
 
-    def execute_query(self, query: str, params: list = None) -> pyodbc.Cursor:
+    def execute_query(
+        self, query: str, params: list = None, return_dict: bool = False
+    ) -> list | None:
         """Execute SQL query with pyodbc"""
         params = [] if not params else params
-        is_select = query.strip().upper().startswith('SELECT')
+        is_select = query.strip().upper().startswith("SELECT")
         try:
             res = self.cursor.execute(query, params)
             if is_select:
-                res = self.cursor.fetchall()
-                if len(res) == 0:
+                rows = self.cursor.fetchall()
+                if len(rows) == 0:
                     print("No results from query")
                     return None
+                if return_dict:
+                    columns = [column[0] for column in self.cursor.description]
+                    res = [dict(zip(columns, row)) for row in rows]
+                else:
+                    res = rows
                 return res
             else:
                 return None
@@ -44,15 +53,19 @@ class Utility:
     def fetch_env(self, db_env):
         """Get env variable based on context, PROD or TEST"""
         if db_env.upper() == "PROD":
-            connection_env = "DBCONNECTIONSTRINGPROD" 
+            connection_env = "DBCONNECTIONSTRINGPROD"
             return connection_env
         if db_env.upper() == "TEST":
-            connection_env = "DBCONNECTIONSTRINGDEV" 
+            connection_env = "DBCONNECTIONSTRINGDEV"
             return connection_env
 
-        raise ValueError(f"arg db_env is {db_env.upper()} but should be 'PROD' or 'TEST'")
+        raise ValueError(
+            f"arg db_env is {db_env.upper()} but should be 'PROD' or 'TEST'"
+        )
 
-    def execute_stored_procedure(self, stored_procedure: str, params: Dict[str, Tuple[type, Any]] | None = None) -> Dict[str, Union[bool, str, Any]]:
+    def execute_stored_procedure(
+        self, stored_procedure: str, params: Dict[str, Tuple[type, Any]] | None = None
+    ) -> Dict[str, Union[bool, str, Any]]:
         """
         Executes a stored procedure with the given parameters.
 
@@ -76,12 +89,12 @@ class Utility:
             "int": int,
             "float": float,
             "datetime": parser.isoparse,
-            "json": lambda x: json.dumps(x, ensure_ascii=False)
+            "json": lambda x: json.dumps(x, ensure_ascii=False),
         }
 
         try:
             if params:
-                param_placeholders = ', '.join([f"@{key} = ?" for key in params.keys()])
+                param_placeholders = ", ".join([f"@{key} = ?" for key in params.keys()])
                 param_values = []
 
                 for key, value in params.items():
@@ -92,7 +105,9 @@ class Utility:
                         else:
                             param_values.append(actual_value)
                     else:
-                        raise ValueError("Each parameter value must be a tuple of (type, actual_value).")
+                        raise ValueError(
+                            "Each parameter value must be a tuple of (type, actual_value)."
+                        )
 
                 sql = f"EXEC {stored_procedure} {param_placeholders}"
                 rows_updated = self.cursor.execute(sql, tuple(param_values))

mbu_dev_shared_components-4.2.0/mbu_dev_shared_components/solteqtand/application/aftalebog.py

@@ -0,0 +1,209 @@
+from datetime import datetime
+
+import uiautomation as auto
+
+from .handler_base import HandlerBase
+
+
+class AftalebogHandler(HandlerBase):
+    """Functions within aftalebog"""
+
+    def get_appointments_aftalebog(
+        self, close_after: bool = False, headers_to_keep: list | None = None
+    ) -> dict:
+        """Function to retrive data on appointments in view in aftalebog"""
+
+        # Get list control
+        list_box = self.wait_for_control(
+            control_type=auto.GroupControl,
+            search_params={"AutomationId": "GroupBoxView"},
+            search_depth=5,
+        )
+
+        appointment_list = self.find_element_by_property(
+            control=list_box, control_type=50008
+        )
+
+        # Extract headers
+        appointment_headers = [
+            header.Name
+            for header in appointment_list.GetFirstChildControl().GetChildren()
+        ]
+
+        # Extract ListItem controls
+        appointment_ctrls = [
+            ctrl for ctrl in appointment_list.GetChildren() if ctrl.ControlType == 50007
+        ]
+
+        # Package data in dictionary
+        # Keep only selected headers if any selected.
+        if not headers_to_keep:
+            headers_to_keep = appointment_headers
+
+        appointment_data = {
+            j: {
+                k: v.Name
+                for k, v in zip(appointment_headers, ctrl.GetChildren())
+                if k in headers_to_keep
+            }
+            for j, ctrl in enumerate(appointment_ctrls)
+        }
+
+        if close_after:
+            # Should maybe be in a method of its own?
+            list_box.SendKeys("{Control}{F4}")
+            self.wait_for_control_to_disappear(
+                control_type=auto.WindowControl,
+                search_params={"AutomationId": "FormBooking"},
+            )
+
+        return appointment_data
+
+    def set_date_in_aftalebog(self, from_date: datetime, to_date: datetime) -> None:
+        """Set to and from dates in aftalebog oversigt"""
+        import locale
+
+        dt_picker_from = self.wait_for_control(
+            control_type=auto.PaneControl,
+            search_params={"AutomationId": "DateTimePickerFromDate"},
+            search_depth=7,
+        )
+
+        from_keys = (
+            f"{from_date.day}"
+            + "{right}"
+            + f"{from_date.month}"
+            + "{right}"
+            + f"{from_date.year}"
+        )
+
+        dt_picker_from.SendKeys(from_keys)
+
+        try:
+            from_date.strftime(format="%d. %B %Y") == dt_picker_from.Name
+        except Exception:
+            # Should maybe try a number of times until it hits right or ends in systemerror
+            # End with raise error where resulting dates are printed
+            print("Dates after insert not matching input")
+            print(
+                (
+                    f"'From' input: {from_date.strftime(format='%d. %B %Y')} "
+                    + f"Current value: {dt_picker_from.Name}"
+                )
+            )
+
+        dt_picker_to = self.wait_for_control(
+            control_type=auto.PaneControl,
+            search_params={"AutomationId": "DateTimePickerToDate"},
+            search_depth=7,
+        )
+
+        to_keys = (
+            f"{to_date.day}"
+            + "{right}"
+            + f"{to_date.month}"
+            + "{right}"
+            + f"{to_date.year}"
+        )
+
+        dt_picker_to.SendKeys(to_keys)
+
+        locale.setlocale(locale.LC_TIME, "da_dk.utf-8")
+
+        try:
+            to_date.strftime(format="%d. %B %Y") == dt_picker_to.Name
+        except Exception:
+            print("Dates after insert not matching input")
+            print(
+                (
+                    f"'To' input: {to_date.strftime(format='%d. %B %Y')} "
+                    + f"Current value: {dt_picker_to.Name}"
+                )
+            )
+
+    def pick_appointment_types_aftalebog(self, appointment_types: str | list):
+        """Set one or more appointment types in aftalebog oversigt"""
+
+        if isinstance(appointment_types, str):
+            appointment_types = [appointment_types]
+
+        # deselect all
+        slct_none = self.wait_for_control(
+            control_type=auto.PaneControl,
+            search_params={"AutomationId": "ButtonToggleStatusList"},
+            search_depth=7,
+        )
+        slct_none.SetFocus()
+        # If possible to select none click once, otherwise click twice
+        try:
+            assert slct_none.Name == "Vælge ingen"
+            slct_none.SendKeys("{Enter}")
+        except AssertionError:
+            slct_none.SendKeys("{Enter}{Enter}")
+
+        # Getting status controls
+        status_list = self.wait_for_control(
+            control_type=auto.ListControl,
+            search_params={"AutomationId": "CheckedListBoxStatus"},
+            search_depth=7,
+        )
+        status_ctrls = [
+            _child
+            for _child in status_list.GetChildren()
+            if _child.ControlType == 50002
+        ]
+        status_names = [
+            _child.Name
+            for _child in status_list.GetChildren()
+            if _child.ControlType == 50002
+        ]
+
+        # Toggle all selected appointment types
+        for a_type in appointment_types:
+            slct_idx = status_names.index(a_type)
+            status_ctrls[slct_idx].GetTogglePattern().Toggle()
+
+    def pick_clinic_aftalebog(self, clinic: str):
+        """Set clinic in aftalebog oversigt"""
+
+        ## UNFINISHED
+        # Press clinic button
+        clinic_button = self.wait_for_control(
+            control_type=auto.PaneControl,
+            search_params={"AutomationId": "ButtonClinic"},
+            search_depth=8,
+        )
+        clinic_button.SetFocus()
+        clinic_button.SendKeys("{Enter}")
+
+        # Wait for popup window
+        find_clinic = self.wait_for_control(
+            control_type=auto.WindowControl,
+            search_params={"AutomationId": "FormFindClinics"},
+            search_depth=2,
+        )
+        # Get list and select clinic
+        clinic_list = self.find_element_by_property(
+            control=find_clinic, automation_id="ListClinics"
+        )
+        clinic_ctrls = [
+            _child
+            for _child in clinic_list.GetChildren()
+            if _child.ControlType == 50007
+        ]
+        clinic_names = [
+            _child.Name
+            for _child in clinic_list.GetChildren()
+            if _child.ControlType == 50007
+        ]
+        try:
+            slct_idx = clinic_names.index(clinic)
+        except Exception as e:
+            print(e)
+            print(f"Chosen clinic: {clinic}")
+            print("Possibilities: ")
+            print(" \n".join(clinic_names[::-1]))
+        # Search for the clinic if it is in the list (to get in focus)
+        find_clinic.SendKeys(clinic)
+        clinic_ctrls[slct_idx].SetFocus()
+        clinic_ctrls[slct_idx].SendKeys("{Enter}")

{mbu_dev_shared_components-4.0.3 → mbu_dev_shared_components-4.2.0}/mbu_dev_shared_components/solteqtand/application/app_handler.py

@@ -1,15 +1,18 @@
 """This module provides the main application handler for Solteq Tand, integrating various components."""
+
 import os
+
 import uiautomation as auto
 
+from .aftalebog import AftalebogHandler
+from .appointment import AppointmentHandler
 from .base_ui import BaseUI
-from .patient import PatientHandler
+from .clinic import ClinicHandler
 from .document import DocumentHandler
-from .appointment import AppointmentHandler
 from .edi_portal import EDIHandler
-from .clinic import ClinicHandler
 from .event import EventHandler
 from .journal_note import JournalNoteHandler
+from .patient import PatientHandler
 
 
 class SolteqTandApp(
@@ -20,7 +23,8 @@ class SolteqTandApp(
     EDIHandler,
     ClinicHandler,
     EventHandler,
-    JournalNoteHandler
+    JournalNoteHandler,
+    AftalebogHandler,
 ):
     """
     Main application handler for Solteq Tand, integrating various components.
@@ -62,9 +66,9 @@ class SolteqTandApp(
         """
         self.app_window = self.wait_for_control(
             auto.WindowControl,
-            {'AutomationId': 'FormLogin'},
+            {"AutomationId": "FormLogin"},
             search_depth=3,
-            timeout=60
+            timeout=60,
         )
         self.app_window.SetFocus()
 
@@ -76,13 +80,13 @@ class SolteqTandApp(
 
         login_button = self.app_window.PaneControl(AutomationId="ButtonLogin")
         login_button.SetFocus()
-        login_button.SendKeys('{ENTER}')
+        login_button.SendKeys("{ENTER}")
 
         self.app_window = self.wait_for_control(
             auto.WindowControl,
-            {'AutomationId': 'FormFront'},
+            {"AutomationId": "FormFront"},
             search_depth=2,
-            timeout=60
+            timeout=60,
         )
 
     def open_sub_tab(self, sub_tab_name: str):
@@ -97,7 +101,7 @@ class SolteqTandApp(
 
         if not is_sub_tab_selected:
             sub_tab_button.SetFocus()
-            sub_tab_button.SendKeys('{ENTER}')
+            sub_tab_button.SendKeys("{ENTER}")
 
     def open_tab(self, tab_name: str):
         """
@@ -122,13 +126,13 @@ class SolteqTandApp(
         tab_button = self.find_element_by_property(
             control=self.app_window,
             control_type=auto.ControlType.TabItemControl,
-            name=tab_name_modified
+            name=tab_name_modified,
         )
         is_tab_selected = tab_button.GetPattern(10010).IsSelected
 
         if not is_tab_selected:
             tab_button.SetFocus()
-            tab_button.SendKeys('{ENTER}')
+            tab_button.SendKeys("{ENTER}")
 
     def open_from_main_menu(self, menu_item: str) -> None:
         """
@@ -137,20 +141,16 @@ class SolteqTandApp(
         # Find hyperlink
         menu_link = self.wait_for_control(
             control_type=auto.HyperlinkControl,
-            search_params={
-                'Name': menu_item
-            },
-            search_depth=5
+            search_params={"Name": menu_item},
+            search_depth=5,
         )
 
         menu_link.GetInvokePattern().Invoke()
 
         self.app_window = self.wait_for_control(
             control_type=auto.WindowControl,
-            search_params={
-                "AutomationId": "FormBooking"
-            },
-            search_depth=2
+            search_params={"AutomationId": "FormBooking"},
+            search_depth=2,
         )
 
     def close_solteq_tand(self):

{mbu_dev_shared_components-4.0.3 → mbu_dev_shared_components-4.2.0}/mbu_dev_shared_components.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mbu_dev_shared_components
-Version: 4.0.3
+Version: 4.2.0
 Summary: Shared components to use in RPA projects
 Author-email: MBU <rpa@mbu.aarhus.dk>
 License-Expression: MIT
@@ -12,6 +12,7 @@ License-File: LICENSE
 Requires-Dist: cryptography>=43.0.0
 Requires-Dist: pyodbc>=5.1.0
 Requires-Dist: python-dateutil==2.9.*
+Requires-Dist: dotenv
 Provides-Extra: getorganized
 Requires-Dist: requests_ntlm>=1.2.0; extra == "getorganized"
 Provides-Extra: google
@@ -31,6 +32,8 @@ Provides-Extra: sap
 Provides-Extra: solteqtand
 Requires-Dist: uiautomation; extra == "solteqtand"
 Requires-Dist: pyodbc>=5.1.0; extra == "solteqtand"
+Requires-Dist: psutil; extra == "solteqtand"
+Requires-Dist: docx2pdf; extra == "solteqtand"
 Provides-Extra: utils
 Requires-Dist: python-dateutil==2.9.*; extra == "utils"
 Requires-Dist: cryptography>=43.0.0; extra == "utils"

{mbu_dev_shared_components-4.0.3 → mbu_dev_shared_components-4.2.0}/mbu_dev_shared_components.egg-info/SOURCES.txt

@@ -37,6 +37,7 @@ mbu_dev_shared_components/sap/__init__.py
 mbu_dev_shared_components/sap/create_invoice.py
 mbu_dev_shared_components/solteqtand/__init__.py
 mbu_dev_shared_components/solteqtand/application/__init__.py
+mbu_dev_shared_components/solteqtand/application/aftalebog.py
 mbu_dev_shared_components/solteqtand/application/app_handler.py
 mbu_dev_shared_components/solteqtand/application/appointment.py
 mbu_dev_shared_components/solteqtand/application/base_ui.py

{mbu_dev_shared_components-4.0.3 → mbu_dev_shared_components-4.2.0}/mbu_dev_shared_components.egg-info/requires.txt

@@ -1,6 +1,7 @@
 cryptography>=43.0.0
 pyodbc>=5.1.0
 python-dateutil==2.9.*
+dotenv
 
 [dev]
 pylint
@@ -34,6 +35,8 @@ numpy
 [solteqtand]
 uiautomation
 pyodbc>=5.1.0
+psutil
+docx2pdf
 
 [utils]
 python-dateutil==2.9.*

{mbu_dev_shared_components-4.0.3 → mbu_dev_shared_components-4.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mbu_dev_shared_components"
-version = "4.0.3"
+version = "4.2.0"
 authors = [
   { name="MBU", email="rpa@mbu.aarhus.dk" },
 ]
@@ -20,6 +20,7 @@ dependencies = [
   "cryptography >= 43.0.0",
   "pyodbc >= 5.1.0",
   "python-dateutil == 2.9.*",
+  "dotenv",
 ]
 
 [project.optional-dependencies]
@@ -49,6 +50,8 @@ sap = [
 solteqtand = [
   "uiautomation",
   "pyodbc >= 5.1.0",
+  "psutil",
+  "docx2pdf",
 ]
 utils = [
   "python-dateutil == 2.9.*",