38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
from utils.database.connection import connetti_db
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def get_nodes_type(cfg, tool, unit):
|
|
|
|
with connetti_db(cfg) as conn:
|
|
cur = conn.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 {cfg.dbname}.{cfg.dbnodes} 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;
|
|
"""
|
|
logger.info(f"{unit} - {tool}: Executing query: {query}")
|
|
cur.execute(query)
|
|
results = cur.fetchall()
|
|
logger.info(f"{unit} - {tool}: {cur.rowcount} rows selected to get node type/Ain/Din/channels.")
|
|
cur.close()
|
|
conn.close()
|
|
|
|
if not results:
|
|
logger.info(f"{unit} - {tool}: Node/Channels/Ain/Din not defined.")
|
|
return None, None, None, None
|
|
else:
|
|
channels, types, ains, dins = [], [], [], []
|
|
for row in results:
|
|
channels.append(row['channels'])
|
|
types.append(row['type'])
|
|
ains.append(row['ain'])
|
|
dins.append(row['din'])
|
|
return channels, types, ains, dins
|
|
|