Function setup_database

tests/test_db.py:39–66

(dbname, driver=None, pooling=False)

def setup_database(dbname, driver=None, pooling=False):
    if dbname == "sqlite":
        db = web.database(dbn=dbname, db="webpy.db", pooling=pooling, driver=driver)
    elif dbname == "postgres":
        db = web.database(
            dbn=dbname,
            host=os.getenv("WEBPY_DB_HOST", "localhost"),
            port=int(os.getenv("WEBPY_DB_PG_PORT", 5432)),
            db=os.getenv("WEBPY_DB_NAME", "webpy"),
            user=os.getenv("WEBPY_DB_USER", os.getenv("USER")),
            pw=os.getenv("WEBPY_DB_PASSWORD", ""),
            pooling=pooling,
            driver=driver,
        )
    else:
        db = web.database(
            dbn=dbname,
            host=os.getenv("WEBPY_DB_HOST", "localhost"),
            port=int(os.getenv("WEBPY_DB_MYSQL_PORT", 3306)),
            db=os.getenv("WEBPY_DB_NAME", "webpy"),
            user=os.getenv("WEBPY_DB_USER", os.getenv("USER")),
            pw=os.getenv("WEBPY_DB_PASSWORD", ""),
            pooling=pooling,
            driver=driver,
        )

    db.printing = True
    return db


class DBTest(unittest.TestCase):