holado 0.22.0.dev0__tar.gz → 0.22.0.dev1__tar.gz

{holado-0.22.0.dev0 → holado-0.22.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: holado
-Version: 0.22.0.dev0
+Version: 0.22.0.dev1
 Summary: HolAdo framework
 Project-URL: Homepage, https://gitlab.com/holado_framework/python
 Project-URL: Issues, https://gitlab.com/holado_framework/python/-/issues

holado-0.22.0.dev1/packaging/VERSION

@@ -0,0 +1 @@
+0.22.0.dev1

{holado-0.22.0.dev0 → holado-0.22.0.dev1}/src/holado_report/campaign/campaign_manager.py

@@ -365,8 +365,8 @@ class CampaignManager(object):
         return camp['id']
     
     def update_or_add_campaign_scenario(self, campaign_id, uid, name, *, report_path=None, status=None, status_at_str=None, details=None):
-        filter_data = {'campaign_id': campaign_id, 'uid':uid, 'name': name}
-        data = {}
+        filter_data = {'campaign_id': campaign_id, 'uid':uid}
+        data = {'name': name}
         if report_path is not None:
             data['report_path'] = report_path
         if status is not None:

holado-0.22.0.dev1/src/holado_test/tools/test_server/server/migrate/migrate_to_v0.22.py

@@ -0,0 +1,137 @@
+# /// script
+# requires-python = ">=3.10"
+# dependencies = [
+#   "holado",
+# ]
+# ///
+
+import datetime
+import re
+
+from prettytable import PrettyTable, TableStyle, HRuleStyle
+import os
+import logging
+
+
+logger = None
+
+
+def __get_work_dir_path():
+    here = os.path.abspath(os.path.dirname(__file__))
+    return os.path.abspath(here)
+
+def initialize():
+    from holado_helper.script.initialize_script import initialize, parse_logging_args  # @UnresolvedImport
+    log_args = parse_logging_args()
+    res = initialize(work_dir_path=__get_work_dir_path(),
+                     log_level=log_args.log_level, log_in_file=log_args.log_in_file, with_session_path=log_args.log_in_file)
+
+    global logger
+    logger = logging.getLogger(__name__)
+
+    return res
+
+
+def migrate_db():
+    from holado.common.context.session_context import SessionContext
+    from holado_report.campaign.campaign_manager import CampaignManager
+
+    campaign_manager = CampaignManager(db_name="server/campaigns")
+    db_client = SessionContext.instance().resource_manager.get_db_client(campaign_manager._CampaignManager__campaign_scenarios_table_manager._TableDataManager__db_name,
+                                                                         is_persistent=campaign_manager._CampaignManager__campaign_scenarios_table_manager._TableDataManager__is_persistent)
+
+    db_client.execute("DROP INDEX idx_campaign_scenarios_name_status_at")
+    db_client.execute("DROP INDEX idx_campaign_scenarios_campaign_id_name")
+
+    db_client.execute("ALTER TABLE campaign_scenarios RENAME TO campaign_scenarios_bak")
+    db_client.execute( campaign_manager._CampaignManager__get_campaign_scenarios_table_sql_create() )
+    db_client.execute("INSERT INTO campaign_scenarios SELECT id, campaign_id, name as uid, name, report_path, status, status_at, details FROM campaign_scenarios_bak")
+    db_client.execute("DROP TABLE campaign_scenarios_bak")
+
+    db_client.execute("CREATE INDEX idx_campaign_scenarios_uid_status_at on campaign_scenarios(uid, status_at)")
+    db_client.execute("CREATE UNIQUE INDEX idx_campaign_scenarios_campaign_id_uid on campaign_scenarios(campaign_id, uid)")
+
+def update_history(args):
+    from holado.common.context.session_context import SessionContext
+    from holado_report.campaign.campaign_manager import CampaignManager
+
+    campaign_manager = CampaignManager(db_name="server/campaigns")
+    db_client = SessionContext.instance().resource_manager.get_db_client(campaign_manager._CampaignManager__campaign_scenarios_table_manager._TableDataManager__db_name,
+                                                                         is_persistent=campaign_manager._CampaignManager__campaign_scenarios_table_manager._TableDataManager__is_persistent)
+
+    # Get scenario data
+    scenario_data_list = db_client.execute("SELECT DISTINCT uid, name, json_extract(details, '$.scenario') as scenario FROM campaign_scenarios WHERE uid <> name", result_as_dict_list=True)
+    data_list = []
+    for scenario_data in scenario_data_list:
+        uid, name, scenario_title = scenario_data['uid'], scenario_data['name'], scenario_data['scenario']
+        file_path, line_number = name.split(':')
+
+        if args.path_pattern is not None:
+            if not re.match(args.path_pattern, file_path):
+                continue
+
+        data_list.append((uid, file_path, int(line_number), scenario_title))
+    logger.print(f"Found {len(data_list)} scenarios as sources:\n" + '\n'.join(str(d) for d in data_list))
+
+    for data in data_list:
+        uid, file_path, line_number, scenario_title = data
+
+        if args.path_pattern is not None:
+            if not re.match(args.path_pattern, file_path):
+                continue
+
+        for delta in range(args.line_delta_start, args.line_delta_end):
+            d_name = f"{file_path}:{line_number + delta}"
+
+            if args.dry_run:
+                if args.ignore_scenario_title:
+                    result = db_client.execute(f"SELECT count(*) FROM campaign_scenarios WHERE uid = name and name = '{d_name}'")
+                    count = int(result[0][0].content)
+                    if count:
+                        logger.print(f"{count} UID to replace (delta: {delta}): '{d_name}' -> '{uid}'")
+                else:
+                    result = db_client.execute(f"SELECT count(*) FROM campaign_scenarios WHERE uid = name and name = '{d_name}' and json_extract(details, '$.scenario') = '{scenario_title}'")
+                    count = int(result[0][0].content)
+                    if count:
+                        logger.print(f"{count} UID to replace (delta: {delta}): '{d_name}' -> '{uid}'  (for scenario: {scenario_title})")
+            else:
+                if args.ignore_scenario_title:
+                    result = db_client.execute(f"UPDATE campaign_scenarios SET uid = '{uid}' WHERE uid = name and name = '{d_name}'")
+                    count = int(result)
+                    if count:
+                        logger.print(f"{count} UID replaced (delta: {delta}): '{d_name}' -> '{uid}'")
+                else:
+                    result = db_client.execute(f"UPDATE campaign_scenarios SET uid = '{uid}' WHERE uid = name and name = '{d_name}' and json_extract(details, '$.scenario') = '{scenario_title}'")
+                    count = int(result)
+                    if count:
+                        logger.print(f"{count} UID replaced (delta: {delta}): '{d_name}' -> '{uid}'  (for scenario: {scenario_title})")
+
+
+def main():
+    from holado_helper.script.initialize_script import get_logging_argument_parser
+
+    parser = get_logging_argument_parser(add_help=True)
+    parser.add_argument('action', choices=['migrate_db', 'update_history'], default=None, help='Action name')
+    parser.add_argument('-b', '--line-delta-start', dest='line_delta_start', type=int, default=-10,
+                        help="Start delta on scenario line number (for action 'update_history')")
+    parser.add_argument('-e', '--line-delta-end', dest='line_delta_end', type=int, default=0,
+                        help="End delta on scenario line number (for action 'update_history')")
+    parser.add_argument('-i', '--ignore-scenario-title', dest='ignore_scenario_title', default=False, action="store_true",
+                        help='If set, apply update even if scenario title is different')
+    parser.add_argument('-d', '--dry-run', dest='dry_run', default=False, action="store_true",
+                        help='If set, dry run')
+    parser.add_argument('-p', '--path-pattern', dest='path_pattern', type=str, default=None,
+                        help='If set, apply update only on relative file paths matching this pattern')
+    args = parser.parse_args()
+
+    if args.action == 'migrate_db':
+        migrate_db()
+    elif args.action == 'update_history':
+        update_history(args)
+    else:
+        raise ValueError(f"Invalid action '{args.action}'")
+
+if __name__ == "__main__":
+    initialize()
+    main()
+

holado-0.22.0.dev0/packaging/VERSION

@@ -1 +0,0 @@
-0.22.0.dev