evol 2

ase_db/get_nodes_type.py

@@ -0,0 +1,93 @@
+import mysql.connector
+import datetime
+import os
+
+def get_timestamp(log_type):
+    """Generates a timestamp string for logging."""
+    now = datetime.datetime.now()
+    return now.strftime("%Y-%m-%d %H:%M:%S")
+
+def get_nodes_type(db_lar, server, username, password, tool, unit, channels, nodetype, ain, din):
+    """
+    Retrieves node type, Ain, Din, and channels from the database for a specific tool and unit.
+
+    Args:
+        db_lar (str): The name of the MySQL database.
+        server (str): The hostname or IP address of the MySQL server.
+        username (str): The MySQL username.
+        password (str): The MySQL password.
+        tool (str): The name of the tool.
+        unit (str): The name of the unit.
+        channels (list): An empty list to store the 'channels' values.
+        nodetype (list): An empty list to store the 'type' values.
+        ain (list): An empty list to store the 'ain' values.
+        din (list): An empty list to store the 'din' values.
+    """
+    try:
+        dbh = mysql.connector.connect(
+            host=server,
+            database=db_lar,
+            user=username,
+            password=password
+        )
+        cursor = dbh.cursor(dictionary=True)
+
+        query = f"""
+            SELECT t.name AS name, n.seq AS seq, n.num AS num, n.channels AS channels, y.type AS type, n.ain AS ain, n.din AS din
+            FROM nodes AS n
+            INNER JOIN tools AS t ON t.id = n.tool_id
+            INNER JOIN units AS u ON u.id = t.unit_id
+            INNER JOIN nodetypes AS y ON n.nodetype_id = y.id
+            WHERE y.type NOT IN ('Anchor Link', 'None') AND t.name = '{tool}' AND u.name = '{unit}'
+            ORDER BY n.num;
+        """
+
+        cursor.execute(query)
+        results = cursor.fetchall()
+
+        print(f"{get_timestamp('log')} - pid {os.getpid()} >> {unit} - {tool}: {cursor.rowcount} rows selected to get node type/Ain/Din/channels.")
+
+        if not results:
+            print(f"{get_timestamp('log')} - pid {os.getpid()} >> Node/Channels/Ain/Din not defined.")
+            print(f"{get_timestamp('log')} - pid {os.getpid()} >> Execution ended.")
+            exit()
+        else:
+            for row in results:
+                channels.append(row['channels'])
+                nodetype.append(row['type'])
+                ain.append(row['ain'])
+                din.append(row['din'])
+
+        cursor.close()
+        dbh.close()
+
+    except mysql.connector.Error as err:
+        error_message = f"{get_timestamp('log')} - pid {os.getpid()} >> Could not connect to database: {err}"
+        print(error_message)
+        raise  # Re-raise the exception if you want the calling code to handle it
+
+if __name__ == '__main__':
+    # Example usage (replace with your actual database credentials and values)
+    db_name = "your_database_name"
+    db_server = "your_server_address"
+    db_user = "your_username"
+    db_password = "your_password"
+    current_tool = "your_tool_name"
+    current_unit = "your_unit_name"
+
+    node_channels = []
+    node_types = []
+    node_ains = []
+    node_dins = []
+
+    try:
+        get_nodes_type(db_name, db_server, db_user, db_password, current_tool, current_unit, node_channels, node_types, node_ains, node_dins)
+
+        print("\nRetrieved Data:")
+        print(f"Channels: {node_channels}")
+        print(f"Node Types: {node_types}")
+        print(f"Ains: {node_ains}")
+        print(f"Dins: {node_dins}")
+
+    except Exception as e:
+        print(f"An error occurred: {e}")