Utility functions to help integrate Sqlalchemy with django

utils_for_django_sqlalchemy.py

"""Common utility functions sqlalchemy in django"""


def dj_pg_db_url_to_sqlachemy(dj_pg_db_url):
    """
    transforms a dj_database_url postgres url to
    an sqlalchemy onereplacing 'postgres://...' with 'posgresql://...'
    """
    try:
        return 'postgresql%s' % (dj_pg_db_url[dj_pg_db_url.find('://'):])
    except Exception as exp:
        print(str(exp))


def dj_test_db_url(dj_db_url):
    """
    Converts the url the normal database
    to the url for the test database
    """
    try:
        last_slash_position = dj_db_url.rfind('/') + 1
        test_db_name = 'test_%s' % (dj_db_url[last_slash_position:],)
        return '%s%s' % (dj_db_url[:last_slash_position], test_db_name,)
    except Exception as exp:
        print(str(exp))