Mini Shell

#! /opt/imh-python/bin/python3
""" Reset database user passwords for CMS. """
# Author: Daniel K

from rads.color import yellow, red, green
from cms_tools.cms import CMSStatus
from cms_tools.helpers import change_db_pass
from cms_tools.common import cms_tools_setup


def main():
    """Main function for script"""
    cms_search = cms_tools_setup()
    failed_cms = []
    failed_resets = []
    successful_cms = []
    for the_cms in cms_search.find_cms():
        if the_cms.status >= CMSStatus.db_is_set:
            if change_db_pass(the_cms.cpuser, the_cms.db_user, the_cms.db_pass):
                successful_cms.append(the_cms)
            else:
                failed_resets.append(the_cms)
        else:
            failed_cms.append(the_cms)
    print(yellow(" === cms_pw ==="))
    if failed_cms:
        print("\nFailed to find credentials for the follwoing CMS:")
        for the_cms in failed_cms:
            print(red(f"{the_cms.directory_root} which is {the_cms.type}"))
    if failed_resets:
        print("\nFailed to reset passwords for the following CMS:")
        for the_cms in failed_resets:
            print(red(f"{the_cms.directory_root} which is {the_cms.type}"))
    if successful_cms:
        print("\nSuccessful passwords:")
        for the_cms in successful_cms:
            print(green(f"{the_cms.directory_root} which is {the_cms.type}"))


if __name__ == "__main__":
    main()