Example #1
0
def dbsession(request, worker_id):
    """
    Properly setup, yield and teardown an Oracle backed SQLAlchemy session

    Make sure this is in sync with func_scoped_dbsession
    """
    settings = DEFAULT_TEST_SETTINGS
    if request.config.getoption("--use-sqlite3"):
        settings["adaero.use_local_sqlite3"] = True
        settings["adaero.worker_id"] = worker_id

    config = testing.setUp(settings=settings)
    config.include("adaero.models")
    settings = config.get_settings()

    from adaero.models import get_engine, get_session_factory, get_tm_session

    engine = get_engine(settings)
    session_factory = get_session_factory(engine)
    session = get_tm_session(session_factory, transaction.manager)

    from adaero.models.all import Base, SEQUENCES

    Base.metadata.drop_all(engine)

    for seq in SEQUENCES:
        seq.create(engine)
    Base.metadata.create_all(engine)

    yield session

    testing.tearDown()
    transaction.abort()
    Base.metadata.drop_all(engine)
Example #2
0
def email_job(settings):
    log.info("Starting email job...")
    start_time_s = time.time()
    engine = get_engine(settings)
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)
    ldapsource = build_ldapauth_from_settings(settings)
    check_and_send_email(dbsession, ldapsource, settings)
    total_time_s = time.time() - start_time_s
    log.info("Finished email job, took {0:.2f} seconds".format(total_time_s))
Example #3
0
def add_question(ctx, question, caption):
    engine = ctx.obj[ENGINE_KEY]
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)
    with transaction.manager:
        q = FeedbackQuestion(question_template=question, caption=caption)
        dbsession.add(q)
    with transaction.manager:
        result = (dbsession.query(FeedbackQuestion).filter(
            FeedbackQuestion.question_template == question).one())
        print("Added question with id=%s" % result.id)
Example #4
0
def nominate_everyone(ctx):
    engine = ctx.obj[ENGINE_KEY]
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)

    with transaction.manager:
        period = Period.get_current_period(dbsession)
        for user, nominee in (dbsession.query(User, Nominee).outerjoin(
                Nominee, Nominee.username == User.username).all()):
            if not nominee:
                dbsession.add(Nominee(user=user, period=period))
Example #5
0
def generate_periods(ctx):
    engine = ctx.obj[ENGINE_KEY]
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)
    with transaction.manager:
        for i in range(1, 3):
            dates_dict = generate_period_dates(
                Period.INACTIVE_SUBPERIOD,
                lambda days: (datetime.utcnow() - timedelta(days=i * 30) +
                              timedelta(days=days)),
            )
            period = Period(name=u"Period %s" % i, template_id=1, **dates_dict)
            dbsession.add(period)
Example #6
0
def reset_email_flags(ctx):
    engine = ctx.obj[ENGINE_KEY]
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)
    with transaction.manager:
        period = Period.get_current_period(dbsession)
        print("resetting email flags for period {.name}")
        period.enrol_email_last_sent = None
        period.enrol_reminder_email_last_sent = None
        period.entry_email_last_sent = None
        period.entry_reminder_email_last_sent = None
        period.review_email_last_sent = None
        period.feedback_available_mail_last_sent = None
        dbsession.merge(period)
Example #7
0
def add_template(ctx, question):
    engine = ctx.obj[ENGINE_KEY]
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)
    rows = []
    with transaction.manager:
        for i, q in enumerate(question):
            position = i + 1
            tr = FeedbackTemplateRow(position=position, q_id=q)
            rows.append(tr)
        template = FeedbackTemplate(rows=rows)
        dbsession.add(template)
        transaction.commit()
        print("Added template with id=%s" % template.id)
Example #8
0
def adjust(ctx):
    subperiod = SUBPERIOD_CHOICES[ctx.obj[SUBPERIOD_KEY]]
    engine = ctx.obj[ENGINE_KEY]
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)

    dates_dict = generate_period_dates(
        subperiod, lambda days: datetime.utcnow() + timedelta(days=days))

    with transaction.manager:
        period = Period.get_current_period(dbsession)
        print("setting dates for period {.name} to {}".format(
            period, pformat(dates_dict)))
        for k, v in dates_dict.items():
            setattr(period, k, v)
        dbsession.merge(period)
Example #9
0
def rebuild_database(engine, settings):
    do_drop_database = _yes_or_no(
        'Are you sure you want to drop the database "%s"?' %
        settings[constants.DB_NAME_KEY])
    if not do_drop_database:
        print("Exiting without changes")
        sys.exit(0)

    Base.metadata.drop_all(engine)

    for seq in SEQUENCES:
        seq.create(engine)

    Base.metadata.create_all(engine)

    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)

    return dbsession
Example #10
0
def add_test_periods(ctx):
    engine = ctx.obj[ENGINE_KEY]
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)
    QUESTION_IDS_AND_TEMPLATES = [
        (1, u"What should {display_name} CONTINUE doing?", None),
        (10, u"What should {display_name} STOP doing or do less of?", None),
        (3, u"What should {display_name} START doing or do more of?", None),
        (
            7,
            u"What is the general perception of {display_name}?",
            u"There are often general perceptions of a person that are not "
            u"always based on your direct experiences. This is can still be "
            u"very helpful input to provide.",
        ),
    ]
    template_id = add_template(dbsession, QUESTION_IDS_AND_TEMPLATES, 1)
    add_test_period_with_template(
        dbsession,
        Period.ENROLLMENT_SUBPERIOD,
        template_id,
        1,
        u"Q1 2018",
        add_nominees=False,
    )
    add_test_period_with_template(
        dbsession,
        Period.ENROLLMENT_SUBPERIOD,
        template_id,
        2,
        u"Q4 2017",
        offset_from_utc_now_days=-30,
        add_nominees=False,
    )
    add_test_period_with_template(
        dbsession,
        Period.ENROLLMENT_SUBPERIOD,
        template_id,
        3,
        u"Q3 2017",
        offset_from_utc_now_days=-60,
        add_nominees=False,
    )
    add_test_period_with_template(
        dbsession,
        Period.ENROLLMENT_SUBPERIOD,
        template_id,
        4,
        u"Q2 2017",
        offset_from_utc_now_days=-90,
        add_nominees=False,
    )
    add_test_period_with_template(
        dbsession,
        Period.ENROLLMENT_SUBPERIOD,
        template_id,
        5,
        u"Q1 2017",
        offset_from_utc_now_days=-120,
        add_nominees=False,
    )
    add_test_period_with_template(
        dbsession,
        Period.ENROLLMENT_SUBPERIOD,
        template_id,
        6,
        u"Q4 2016",
        offset_from_utc_now_days=-150,
        add_nominees=False,
    )
Example #11
0
def get_transaction_scoped_dbsession(engine):
    session_factory = get_session_factory(engine)
    dbsession = get_tm_session(session_factory, transaction.manager)
    return dbsession