def GetDbEngineType()

in perfkitbenchmarker/sql_engine_utils.py [0:0]

• 23 lines of code
• 15 McCabe index (conditional complexity)

def GetDbEngineType(db_engine: str) -> str:
  """Converts the engine type from db_engine.

  The same engine type can have multiple names because of differences
  in cloud provider or versions.

  Args:
    db_engine: db_engine defined in the spec

  Returns:
    Engine type in string.
  """

  # AWS uses sqlserver-se and sqlserver-ex as db_egine for sql server
  if db_engine in AWS_SQLSERVER_ENGINES:
    return SQLSERVER
  elif (
      db_engine == AWS_AURORA_POSTGRES_ENGINE
      or db_engine == FLEXIBLE_SERVER_POSTGRES
  ):
    return POSTGRES
  elif (
      db_engine == AWS_AURORA_MYSQL_ENGINE or db_engine == FLEXIBLE_SERVER_MYSQL
  ):
    return MYSQL
  elif db_engine == ALLOYDB or db_engine == OMNI or db_engine == TIMESCALEDB:
    return POSTGRES
  elif db_engine == SPANNER_POSTGRES:
    return SPANNER_POSTGRES
  elif db_engine == SPANNER_GOOGLESQL:
    return SPANNER_GOOGLESQL
  elif db_engine == MARIADB:
    return MYSQL

  if db_engine not in ENGINE_TYPES:
    raise TypeError('Unsupported engine type', db_engine)
  return db_engine