shows current plugin versions in layout.typesettings and performs the update to new version

update_layout/update_layout.py

@@ -0,0 +1,181 @@
+import psycopg2
+import re
+import os
+import argparse
+
+# ANSI escape sequences for colors
+BLACK = '\033[30m'
+RED = '\033[31m'
+GREEN = '\033[32m'
+YELLOW = '\033[33m'  # orange on some systems
+BLUE = '\033[34m'
+MAGENTA = '\033[35m'
+CYAN = '\033[36m'
+LIGHT_GRAY = '\033[37m'
+DARK_GRAY = '\033[90m'
+BRIGHT_RED = '\033[91m'
+BRIGHT_GREEN = '\033[92m'
+BRIGHT_YELLOW = '\033[93m'
+BRIGHT_BLUE = '\033[94m'
+BRIGHT_MAGENTA = '\033[95m'
+BRIGHT_CYAN = '\033[96m'
+WHITE = '\033[97m'
+RESET = '\033[0m'  # called to return to standard terminal text color
+
+# Function to color a string based on a provided pattern and color
+def color_string(string, pattern, color=RED):
+    return string.replace(pattern, f'{color}{pattern}{RESET}')
+
+DEFAULT_CONTEXT_LENGHT = 20
+
+# Default configuration file path
+DEFAULT_CONFIG_FILE = "/home/life/Portal-Bundle/portal-setup-wizard.properties"
+
+# Function to read database connection parameters from the configuration file
+def read_db_config(config_file_path):
+    config = {}
+    with open(config_file_path, 'r') as file:
+        for line in file:
+            if line.strip() and '=' in line and not line.startswith("#"):
+                key, value = line.strip().split('=', 1)
+                config[key.strip()] = value.strip()
+    return config
+
+# Function to connect to the database
+def connect_to_db(db_host, db_port, db_name, db_user, db_password):
+    conn = psycopg2.connect(
+        host=db_host,
+        port=db_port,
+        dbname=db_name,
+        user=db_user,
+        password=db_password
+    )
+    return conn
+
+# Function to update typesettings
+def update_typesettings(cursor, current_pluginid, next_pluginid, log_file):
+    try:
+        search_query = """
+            SELECT layoutid, typesettings FROM layout
+            WHERE typesettings LIKE %s;
+        """
+        cursor.execute(search_query, ('%' + current_pluginid + '%',))
+        rows = cursor.fetchall()
+
+        update_query = """
+            UPDATE layout
+            SET typesettings = replace(typesettings, %s, %s)
+            WHERE layoutid = %s;
+        """
+        
+        for row in rows:
+            layoutid, typesettings = row
+            updated_typesettings = typesettings.replace(current_pluginid, next_pluginid)
+            cursor.execute(update_query, (current_pluginid, next_pluginid, layoutid))
+            log_file.write(f"Executed: {cursor.mogrify(update_query, (updated_typesettings, layoutid))}\n")
+            log_file.write(f"Restore command: {update_query % (typesettings, layoutid)}\n")
+        print(f"Replaced {current_pluginid} with {next_pluginid} in typesettings.")
+
+    except Exception as e:
+        print(f"Error during update: {e}")
+        log_file.write(f"Error during update: {e}\n")
+
+# Function to search for entries containing CURRENT_PLUGINID with context
+def search_current_pluginid(cursor, current_pluginid, enable_color, context_size=10):
+    try:
+        search_query = """
+            SELECT layoutid, typesettings FROM layout
+            WHERE typesettings LIKE %s;
+        """
+        cursor.execute(search_query, ('%' + current_pluginid + '%',))
+        rows = cursor.fetchall()
+        print(f"Entries containing {current_pluginid}:")
+        for layoutid, typesettings in rows:
+            # Find all occurrences of the pattern and print with context
+            matches = [(m.start(), m.end()) for m in re.finditer(current_pluginid, typesettings)]
+            for start, end in matches:
+                context_start = max(0, start - context_size)
+                context_end = min(len(typesettings), end + context_size)
+                context_str = typesettings[context_start:context_end]
+                if enable_color:
+                    context_str = color_string(context_str, current_pluginid, BRIGHT_RED)
+                print(f"Layout ID: {layoutid}, Context: ...{context_str}...")
+    except Exception as e:
+        print(f"Error during search: {e}")
+
+# Parsing command-line arguments
+parser = argparse.ArgumentParser(
+    description='Update and search layout typesettings.',
+    epilog='''
+Examples of usage:
+
+1. Update typesettings:
+   python3 update_layout.py --current-pluginid CURRENT_PLUGINID --next-pluginid NEXT_PLUGINID --update
+
+2. Search for entries with current plugin ID (default action):
+   python3 update_layout.py --current-pluginid CURRENT_PLUGINID
+
+3. Use a configuration file for database parameters:
+   python3 update_layout.py --config-file /path/to/portal-setup-wizard.properties --current-pluginid CURRENT_PLUGINID
+    ''',
+    formatter_class=argparse.RawTextHelpFormatter
+)
+parser.add_argument('--config-file', default=DEFAULT_CONFIG_FILE, help="Path to the configuration file with database parameters (default: {})".format(DEFAULT_CONFIG_FILE))
+parser.add_argument('--db-host', help="Database host")
+parser.add_argument('--db-port', type=int, help="Database port")
+parser.add_argument('--db-name', help="Database name")
+parser.add_argument('--db-user', help="Database user")
+parser.add_argument('--db-password', help="Database password")
+parser.add_argument('--current-pluginid', type=str, required=True, help='Current plugin ID to search for')
+parser.add_argument('--next-pluginid', type=str, help='Next plugin ID to replace with')
+parser.add_argument('--update', action='store_true', help='Update the typesettings by replacing CURRENT_PLUGINID with NEXT_PLUGINID')
+parser.add_argument('--color', type=bool, default=True, help='Enable colored output (default: true)')
+parser.add_argument('--context-size', type=int, default=DEFAULT_CONTEXT_LENGHT, help='Number of characters of context to show around the found pattern (default: {})'.format(DEFAULT_CONTEXT_LENGHT))
+
+args = parser.parse_args()
+
+if not any(vars(args).values()):
+    parser.print_help()
+    exit(1)
+
+# Initialize database connection parameters with None
+db_host = db_port = db_name = db_user = db_password = None
+
+# Determine the database connection parameters
+if args.config_file:
+    db_config = read_db_config(args.config_file)
+    db_host = db_config['jdbc.default.url'].split('/')[2].split(':')[0]
+    db_port = int(db_config['jdbc.default.url'].split('/')[2].split(':')[1])
+    db_name = db_config['jdbc.default.url'].split('/')[-1]
+    db_user = db_config['jdbc.default.username']
+    db_password = db_config['jdbc.default.password']
+
+if args.db_host:
+    db_host = args.db_host
+
+if args.db_port:
+    db_port = args.db_port
+
+if args.db_name:
+    db_name = args.db_name
+
+if args.db_user:
+    db_user = args.db_user
+
+if args.db_password:
+    db_password = args.db_password
+
+# Connect to the database
+conn = connect_to_db(db_host, db_port, db_name, db_user, db_password)
+cursor = conn.cursor()
+
+log_filename = "update_log.txt"
+with open(log_filename, 'a') as log_file:
+    if args.update:
+        update_typesettings(cursor, args.current_pluginid, args.next_pluginid, log_file)
+    else:
+        search_current_pluginid(cursor, args.current_pluginid, args.color, args.context_size)
+
+    conn.commit()
+    cursor.close()
+    conn.close()