Database Existence Check & Management

This function checks if a database exists, if it does, it drops it, if not, it creates it.

Technology Stack : SQLAlchemy-Utils

Code Type : The type of code

Code Difficulty : Intermediate

                
                    
from sqlalchemy_utils import database_exists, create_database, drop_database

def create_or_drop_database_if_exists(engine, database_name):
    """
    Create or drop a database if it exists.
    """
    if database_exists(engine.url, name=database_name):
        drop_database(engine.url, name=database_name)
        print(f"Database '{database_name}' dropped.")
    else:
        create_database(engine.url, name=database_name)
        print(f"Database '{database_name}' created.")