import os

if os.environ.get('DJANGO_DB_ENGINE') == 'mysql':
    import pymysql
    pymysql.install_as_MySQLdb()