mbu-dev-shared-components 0.4.2__tar.gz → 0.4.4__tar.gz

{mbu_dev_shared_components-0.4.2 → mbu_dev_shared_components-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mbu_dev_shared_components
-Version: 0.4.2
+Version: 0.4.4
 Summary: Shared components to use in RPA projects
 Author-email: MBU <rpa@mbu.aarhus.dk>
 Classifier: Programming Language :: Python :: 3

mbu_dev_shared_components-0.4.4/mbu_dev_shared_components/database/constants.py

@@ -0,0 +1,131 @@
+"""This module handles generating and fetching constants and credentials from the database"""
+
+import os
+import pyodbc
+from datetime import datetime
+
+from mbu_dev_shared_components.utils.fernet_encryptor import Encryptor
+from mbu_dev_shared_components.database.utility import connect_to_db, execute_query
+
+
+def add_credential(
+        credential_name: str,
+        username: str,
+        password: str,
+        changed_at: datetime = datetime.now(),
+        db_env: str = "PROD"
+):
+    encryptor = Encryptor()
+    encrypted_password = encryptor.encrypt(password)
+
+    rpa_conn = connect_to_db(db_env=db_env)
+    cursor = rpa_conn.cursor()
+
+    query = """
+        INSERT INTO [RPA].[rpa].[Credentials]
+            ([name]
+            ,[username]
+            ,[password]
+            ,[changed_at])
+        VALUES
+            (?
+            ,?
+            ,?
+            ,?)
+        """
+    params = [credential_name, username, encrypted_password, changed_at]
+    execute_query(query=query, cursor=cursor, params=params)
+
+
+def get_credential(
+        credential_name: str,
+        db_env: str = "PROD"
+    ) -> dict:
+    
+    rpa_conn = connect_to_db(db_env=db_env)
+    cursor = rpa_conn.cursor()
+    encryptor = Encryptor()
+
+    query = """
+        SELECT 
+            Username
+            ,cast(Password as varbinary(max))
+        FROM [RPA].[rpa].[Credentials]
+        WHERE
+            name = ?
+        """
+    
+    params = [credential_name]
+
+    res = execute_query(query=query, cursor=cursor, params=params)
+    if res is not None:
+        res = res[0]
+        username = res[0]
+        encrypted_password = res[1]
+
+        decrypted_password = encryptor.decrypt(encrypted_password)
+
+        return  {
+            "username": username, 
+            "decrypted_password": decrypted_password, 
+            "encrypted_password": encrypted_password
+        } 
+    else:
+        print(f"No credential found with name {credential_name}")
+
+def get_constant(
+        constant_name: str,
+        db_env: str = "PROD"
+    ) -> tuple:
+    
+    rpa_conn = connect_to_db(db_env=db_env)
+    cursor = rpa_conn.cursor()
+    encryptor = Encryptor()
+
+    query = """
+        SELECT 
+            name
+            ,value
+        FROM [RPA].[rpa].[Constants]
+        WHERE
+            name = ?
+        """
+    
+    params = [constant_name]
+
+    res = execute_query(query=query, cursor=cursor, params=params)
+    if res is not None:
+
+        returned_constant = res[0]
+        constant_name = returned_constant[0]
+        value = returned_constant[1]
+
+        return {"constant_name": constant_name, "value": value}
+    else:
+        print(f"No constant found with name: {constant_name}")
+
+
+def add_constant(
+        constant_name: str,
+        value: str,
+        changed_at: datetime = datetime.now(),
+        db_env: str = "PROD"
+):
+    query = """
+        INSERT INTO [RPA].[rpa].[Constants]
+            ([name]
+            ,[value]
+            ,[changed_at])
+        VALUES
+            (?
+            ,?
+            ,?)
+        """
+    
+    rpa_conn = connect_to_db(db_env=db_env)
+    cursor = rpa_conn.cursor()
+
+    params = [constant_name, value, changed_at]
+    execute_query(query=query, cursor=cursor, params=params)
+
+

mbu_dev_shared_components-0.4.4/mbu_dev_shared_components/database/logging.py

@@ -0,0 +1,86 @@
+"""This module handles logging in the RPA database"""
+
+from datetime import datetime
+import time
+import os
+import socket
+
+from mbu_dev_shared_components.database.utility import execute_query, connect_to_db, fetch_env
+from mbu_dev_shared_components.utils.db_stored_procedure_executor import execute_stored_procedure
+
+def log_event(
+        log_db: str,
+        level: str, 
+        message: str, 
+        context: str, 
+        db_env="TEST"
+    ):
+    created_at = datetime.now()
+    """Logs the inputted parameters in """
+    query = f"""
+        INSERT INTO RPA.{log_db}
+            ([level]
+            ,[message]
+            ,[created_at]
+            ,[context])
+        VALUES
+            (?
+            ,?
+            ,?
+            ,?)
+        """
+    rpa_conn = connect_to_db(db_env=db_env)
+    cursor = rpa_conn.cursor()
+
+    params = [level, message, created_at, context]
+    execute_query(query=query, cursor=cursor, params=params)
+
+def _send_heartbeat(
+        servicename,
+        status,
+        details,
+        db_env,
+):
+    conn_env = fetch_env(db_env=db_env)
+    conn_str = os.getenv(conn_env)
+    hostname = socket.gethostname()
+    params = {
+        "ServiceName": (str, servicename), 
+        "Status": (str, status), 
+        "HostName": (str, hostname), 
+        "Details": (str, details)
+    }
+    result = execute_stored_procedure(
+        connection_string=conn_str,
+        stored_procedure='rpa.sp_UpdateHeartbeat',
+        params=params)
+    if result["success"] is not True:
+        print(result["error_message"])
+    
+def log_heartbeat(
+        stop: str|bool,
+        servicename: str,
+        heartbeat_interval: int,
+        details: str = "",
+        db_env: str = "PROD"
+    ):
+    if isinstance(stop,str):
+        stop = stop == "True"
+    if not isinstance(heartbeat_interval, int):
+        heartbeat_interval = int(heartbeat_interval)
+    while not stop:
+        status = "RUNNING"
+        _send_heartbeat(
+            servicename,
+            status,
+            details,
+            db_env,
+        )
+        time.sleep(heartbeat_interval)
+    status = "STOPPED"
+    _send_heartbeat(
+        servicename,
+        status,
+        details,
+        db_env,
+    )

mbu_dev_shared_components-0.4.4/mbu_dev_shared_components/database/utility.py

@@ -0,0 +1,44 @@
+"""This module handles general database connection and calls"""
+
+import os
+import pyodbc
+from datetime import datetime
+
+def connect_to_db(autocommit=True, db_env="PROD") -> pyodbc.Connection:
+    """Establish connection to sql database
+
+    Returns:
+        rpa_conn (pyodbc.Connection): The connection object to the SQL database.
+    """
+    connection_env = fetch_env(db_env)
+    rpa_conn_string = os.getenv(connection_env)
+    rpa_conn = pyodbc.connect(rpa_conn_string, autocommit=autocommit)
+    return rpa_conn
+
+def execute_query(query: str, cursor: pyodbc.Cursor, params: list) -> pyodbc.Cursor:
+    is_select = query.strip().upper().startswith('SELECT')
+    try:
+        res = cursor.execute(query, params)
+        if is_select:
+            res = cursor.fetchall()
+            if len(res) == 0:
+                print("No results from query")
+                return None
+            return res
+        else:
+            return None
+    except pyodbc.Error as e:
+        print(e)
+    finally:
+        cursor.close()
+
+
+def fetch_env(db_env):
+    if db_env.upper() == "PROD":
+        connection_env = "DbConnectionString"
+        return connection_env
+    if db_env.upper() == "TEST":
+        connection_env = "DbConnectionStringTest"
+        return connection_env
+    
+    raise ValueError(f"arg db_env is {db_env.upper()} but should be 'PROD' or 'TEST'")

{mbu_dev_shared_components-0.4.2 → mbu_dev_shared_components-0.4.4}/mbu_dev_shared_components/getorganized/objects.py

@@ -73,7 +73,14 @@ class CaseDataJson:
         }
 
         if case_title is not None:
-            # Optionally add case_title to your search criteria
+            # Change the value of the existing case category to "Standard"
+            for field in search_case_folder_data["FieldProperties"]:
+                if field["InternalName"] == "ows_CaseCategory":
+                    field["Value"] = "Standard"
+
+                    break
+
+            # Optionally add the case title property
             search_case_folder_data["FieldProperties"].append({
                 "InternalName": "ows_CaseTitle",
                 "Value": case_title,

{mbu_dev_shared_components-0.4.2 → mbu_dev_shared_components-0.4.4}/mbu_dev_shared_components/msoffice365/sharepoint_api/files.py

@@ -35,6 +35,8 @@ import os
 from office365.runtime.auth.user_credential import UserCredential
 from office365.sharepoint.client_context import ClientContext
 
+from office365.sharepoint.files.file import File
+
 
 class Sharepoint:
     """
@@ -124,6 +126,22 @@ class Sharepoint:
                 return None
         return None
 
+    def fetch_file_using_open_binary(self, file_name: str, folder_name: str) -> Optional[bytes]:
+        """
+        Downloads a file using the open_binary method from SharePoint.
+        """
+        if self.ctx:
+            try:
+                file_url = f"/teams/{self.site_name}/{self.document_library}/{folder_name}/{file_name}"
+                file_content = File.open_binary(self.ctx, file_url)
+                return file_content.content
+            except Exception:
+                import traceback
+                print("Failed to download file:")
+                traceback.print_exc()
+                return None
+        return None
+
     def _write_file(self, folder_destination: str, file_name: str, file_content: bytes):
         """
         Saves the binary content of a file to a specified local destination.
@@ -211,3 +229,23 @@ class Sharepoint:
                     self.upload_file(folder_name, file_path, file_name)
                 except Exception as e:
                     print(f"Failed to upload file '{file_path}': {e}")
+
+    def upload_file_from_bytes(self, binary_content: bytes, file_name: str, folder_name: str):
+        """
+        Uploads a file to SharePoint directly from a bytes object.
+
+        Args:
+            binary_content (bytes): The binary content of the file.
+            file_name (str): The name to give the file in SharePoint.
+            folder_name (str): The folder in the document library where the file will be uploaded.
+        """
+
+        if self.ctx:
+            try:
+                folder_url = f"/teams/{self.site_name}/{self.document_library}/{folder_name}"
+                target_folder = self.ctx.web.get_folder_by_server_relative_url(folder_url)
+
+                target_folder.upload_file(file_name, binary_content).execute_query()
+                print(f"File '{file_name}' uploaded successfully to '{folder_url}'.")
+            except Exception as e:
+                print(f"Failed to upload file '{file_name}': {e}")

{mbu_dev_shared_components-0.4.2 → mbu_dev_shared_components-0.4.4}/mbu_dev_shared_components/solteqtand/app_handler.py

@@ -805,7 +805,7 @@ class SolteqTandApp:
             # 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"'From' input: {from_date.strftime(format='%d. %B %Y')} " +
                 f"Current value: {dt_picker_from.Name}"))
 
         dt_picker_to = self.wait_for_control(
@@ -831,7 +831,7 @@ class SolteqTandApp:
         except Exception:
             print("Dates after insert not matching input")
             print((
-                f"'To' input: {to_date.strftime(format="%d. %B %Y")} " +
+                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):

{mbu_dev_shared_components-0.4.2 → mbu_dev_shared_components-0.4.4}/mbu_dev_shared_components.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mbu_dev_shared_components
-Version: 0.4.2
+Version: 0.4.4
 Summary: Shared components to use in RPA projects
 Author-email: MBU <rpa@mbu.aarhus.dk>
 Classifier: Programming Language :: Python :: 3

{mbu_dev_shared_components-0.4.2 → mbu_dev_shared_components-0.4.4}/mbu_dev_shared_components.egg-info/SOURCES.txt

@@ -6,6 +6,10 @@ mbu_dev_shared_components.egg-info/SOURCES.txt
 mbu_dev_shared_components.egg-info/dependency_links.txt
 mbu_dev_shared_components.egg-info/requires.txt
 mbu_dev_shared_components.egg-info/top_level.txt
+mbu_dev_shared_components/database/__init__.py
+mbu_dev_shared_components/database/constants.py
+mbu_dev_shared_components/database/logging.py
+mbu_dev_shared_components/database/utility.py
 mbu_dev_shared_components/getorganized/__init__.py
 mbu_dev_shared_components/getorganized/auth.py
 mbu_dev_shared_components/getorganized/cases.py

{mbu_dev_shared_components-0.4.2 → mbu_dev_shared_components-0.4.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mbu_dev_shared_components"
-version = "0.4.2"  # Specify the version manually here
+version = "0.4.4"  # Specify the version manually here
 authors = [
   { name="MBU", email="rpa@mbu.aarhus.dk" },
 ]