Fixed first connection error.

import logging

import os

import time

from sqlalchemy import (Column,

                        ForeignKey,

                        Integer,

                            setup_logger)

from sqlalchemy.orm import declarative_base

from sqlalchemy.orm import relationship

from sqlalchemy.exc import OperationalError

logger = setup_logger(name='DB_SETUP')

Base = declarative_base()

    mom_zero_map_2 = relationship("MomentZeroMaps", foreign_keys=[mom_zero_name_2, run_id])

Base.metadata.create_all(engine)

try:

    Base.metadata.create_all(bind=engine)

except OperationalError:

    logger.error('Connection failed. Sleeping and retrying')

    time.sleep(3)

    Base.metadata.create_all(bind=engine)

logger.info('Connection successful! DB initialized as needed.')