9520ba3f8b
This reverts commit fedd7d1995.
Reason for revert: Activate rollback is broken with this change. Reverting to cover the rollback scenario as well.
Change-Id: I10163cf3e2c4a532d351cb952c7dbd4484661816
Signed-off-by: Luis Eduardo Bonatti <LuizEduardo.Bonatti@windriver.com>
91 lines
2.3 KiB
Python
91 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# Copyright (c) 2025 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# This script will remove load, host_upgrade and software_upgrade
|
|
# database table
|
|
#
|
|
|
|
import logging
|
|
import sys
|
|
|
|
from packaging import version
|
|
import psycopg2
|
|
|
|
from software.utilities.utils import configure_logging
|
|
|
|
DEFAULT_POSTGRES_PORT = 5432
|
|
|
|
LOG = logging.getLogger('main_logger')
|
|
|
|
|
|
def main():
|
|
action = None
|
|
from_release = None
|
|
to_release = None
|
|
postgres_port = DEFAULT_POSTGRES_PORT
|
|
arg = 1
|
|
while arg < len(sys.argv):
|
|
if arg == 1:
|
|
from_release = sys.argv[arg]
|
|
elif arg == 2:
|
|
to_release = sys.argv[arg]
|
|
elif arg == 3:
|
|
action = sys.argv[arg]
|
|
elif arg == 4:
|
|
postgres_port = sys.argv[arg]
|
|
pass
|
|
else:
|
|
print("Invalid option %s." % sys.argv[arg])
|
|
return 1
|
|
arg += 1
|
|
configure_logging()
|
|
LOG.info(
|
|
"%s invoked from_release = %s to_release = %s action = %s"
|
|
% (sys.argv[0], from_release, to_release, action)
|
|
)
|
|
res = 0
|
|
to_release_version = version.Version(to_release)
|
|
minimum_version = version.Version("25.09")
|
|
if action == 'migrate' and to_release_version == minimum_version:
|
|
try:
|
|
conn = psycopg2.connect("dbname=sysinv user=postgres port=%s"
|
|
% postgres_port)
|
|
delete_software_upgrade_database(conn)
|
|
delete_host_upgrade_database(conn)
|
|
delete_load_database(conn)
|
|
conn.close()
|
|
except Exception as e:
|
|
LOG.exception("Error: {}".format(e))
|
|
res = 1
|
|
return res
|
|
|
|
|
|
def delete_load_database(conn):
|
|
delete_cmd = "drop table if exists loads;"
|
|
db_update(conn, delete_cmd)
|
|
LOG.info("Loads table removed with success")
|
|
|
|
|
|
def delete_host_upgrade_database(conn):
|
|
delete_cmd = "drop table if exists host_upgrade;"
|
|
db_update(conn, delete_cmd)
|
|
LOG.info("Host_upgrade table removed with success")
|
|
|
|
|
|
def delete_software_upgrade_database(conn):
|
|
delete_cmd = "drop table if exists software_upgrade;"
|
|
db_update(conn, delete_cmd)
|
|
LOG.info("Software_upgrade table removed with success")
|
|
|
|
|
|
def db_update(conn, query):
|
|
with conn.cursor() as cur:
|
|
cur.execute(query)
|
|
conn.commit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|