Esempio n. 1
0
def _storeConfigEntryDB(key, value, typ=None, desc=None):
    """
    lowest level for storing database entries in the config table
    """

    confEntries = Session.query(Config).filter(Config.Key == unicode(key))
    theConf = None

    # update
    if confEntries.count() == 1:
        theConf = confEntries[0]
        theConf.Value = unicode(value)
        if (typ is not None):
            theConf.Type = unicode(typ)
        if (desc is not None):
            theConf.Description = unicode(desc)

    # insert
    elif confEntries.count() == 0:
        theConf = Config(
                        Key=unicode(key),
                        Value=unicode(value),
                        Type=unicode(typ),
                        Description=unicode(desc)
                        )
    if theConf is not None:
        Session.add(theConf)

    return 101
Esempio n. 2
0
def _storeConfigEntryDB(key, value, typ=None, desc=None):
    """
    lowest level for storing database entries in the config table
    """

    confEntries = Config.query.filter_by(Key=str(key))
    theConf = None

    # update
    if confEntries.count() == 1:
        theConf = confEntries[0]
        theConf.Value = str(value)
        theConf.Type = typ
        theConf.Description = desc

    # insert
    elif confEntries.count() == 0:
        theConf = Config(Key=str(key),
                         Value=str(value),
                         Type=str(typ),
                         Description=str(desc))
    if theConf is not None:
        db.session.add(theConf)

    return 101
Esempio n. 3
0
def test_setup_db_erase_all(app, engine, capsys, erase):
    app.echo = Echo(verbosity=1)

    # GIVEN a database with records
    app.cli_cmd = "init-database"
    setup_db(app)
    init_db_tables(app, drop_data=True, add_defaults=True)

    KEY = "linotp.foobar"
    item = Config(Key=KEY, Value="123", Type="int", Description="test item")
    db.session.add(item)
    db.session.commit()
    assert db.session.query(Config).filter_by(Key=KEY).count() == 1
    db.session.remove()

    # WHEN I invoke `setup_db`
    setup_db(app)
    init_db_tables(app, drop_data=erase, add_defaults=False)

    if erase:
        # Additional record should have disappeared
        assert db.session.query(Config).filter_by(Key=KEY).count() == 0
    else:
        # Additional record should still be there
        assert db.session.query(Config).filter_by(Key=KEY).count() == 1

        item = db.session.query(Config).filter_by(Key=KEY).first()
        db.session.delete(item)
        db.session.commit()
Esempio n. 4
0
    def set_version(self, version):
        """
        set the version new db model number

        - on update: update the entry
        - on new: create new db entry

        :param version: set the new db model version
        """

        if version == self.current_version:
            return

        config_entry = self._query_version()

        if config_entry:
            config_entry.Value = version
        else:
            config_entry = Config(Key=self.db_model_key, Value=version)

        db.session.add(config_entry)
Esempio n. 5
0
def _storeConfigDB(key, val, typ=None, desc=None):
    value = val
    log.debug('storeConfigDB: key %r : value %r' % (key, value))

    if (not key.startswith("linotp.")):
        key = "linotp." + key

    confEntries = Session.query(Config).filter(Config.Key == unicode(key))
    theConf = None

    if typ is not None and typ == 'password':
        value = encryptPassword(val)
        en = decryptPassword(value)
        if (en != val):
            raise Exception(
                "StoreConfig: Error during encoding password type!")

    # # update
    if confEntries.count() == 1:
        theConf = confEntries[0]
        theConf.Value = unicode(value)
        if (typ is not None):
            theConf.Type = unicode(typ)
        if (desc is not None):
            theConf.Description = unicode(desc)

    # # insert
    elif confEntries.count() == 0:
        theConf = Config(Key=unicode(key),
                         Value=unicode(value),
                         Type=unicode(typ),
                         Description=unicode(desc))
    if theConf is not None:
        Session.add(theConf)

    return 101