Database Creation and Dropping Utility

  • Share this:

Code introduction


This function checks if a database exists at the given URL and if not, it creates it. If the database exists and if_exists parameter is 'replace', it drops the existing database and creates a new one.


Technology Stack : Python, SQLAlchemy-Utils

Code Type : Python Function

Code Difficulty : Intermediate


                
                    
def generate_random_sqlalchemy_utils_function():
    import random
    from sqlalchemy_utils import database_exists, create_database, drop_database

    def create_or_drop_database(url, if_exists='replace'):
        """
        This function checks if a database exists at the given URL, and if not, it creates it.
        If the database exists and if_exists is 'replace', it drops the existing database and creates a new one.
        """
        if not database_exists(url):
            create_database(url)
        elif if_exists == 'replace':
            drop_database(url)
            create_database(url)

    return create_or_drop_database