def fuzzify(engine, config):
    """Do the actual fuzzification based on the loaded attributes of
       the models."""
    modelObjects = {}
    Session = sessionmaker(bind=engine)
    session = Session()
    metadata = MetaData(bind=engine, reflect=True)
    cascade_fkeys(metadata)

    for table_name, columns in config.items():
        tables = []
        if table_name in metadata.tables.keys():
            tables.append(utils.get_table(engine, table_name))
        if 'shadow_' + table_name in metadata.tables.keys():
            tables.append(utils.get_table(engine, 'shadow_' + table_name))
        for table in tables:
            print "Doing table: " + str(table)
            modelObjects[table] = type('Model_' + str(table), (object, ), {})
            mapper(modelObjects[table], table)
            q = session.query(modelObjects[table])
            for row in q.all():
                for column, column_type in columns.items():
                    if hasattr(row, column):
                        before = getattr(row, column)
                        after = randomness(before, column_type)
                        setattr(row, column, after)
    cascade_fkeys(metadata, restore=True)
    session.commit()
    session.flush()
def fuzzify(engine, config):
    """Do the actual fuzzification based on the loaded attributes of
       the models."""
    Session = sessionmaker(bind=engine)
    session = Session()
    metadata = MetaData(bind=engine, reflect=True)
    cascade_fkeys(metadata)

    for model_name, columns in config.items():
        table_name = getattr(models, model_name).__tablename__
        tables = [getattr(models, model_name)]
        if 'shadow_' + table_name in metadata.tables.keys():
            tables.append(utils.get_table(engine, 'shadow_' + table_name))
        for table in tables:
            q = session.query(table)
            for row in q.all():
                for column, column_type in columns:
                    setattr(row, column,
                            randomness(getattr(row, column), column_type))

    session.commit()
    cascade_fkeys(metadata, restore=True)
Esempio n. 3
0
    def _transmogrify(self, string, coltype, anontype):
        """ Anonymise the provided string, based upon it's type """
        # Note(mrda): TODO: handle mapping

        # Handle quoted strings
        need_single_quotes = False
        if string[0] == "'" and string[-1] == "'":
            need_single_quotes = True
            string = string[1:-1]

        randomised = randomise.randomness(string, anontype)

        if CONF.debug:
            print (
                '    ....transmogrifying from value "%s" to value "%s"'
                " with type %s, anon type %s" % (string, randomised, coltype, anontype)
            )

        if need_single_quotes:
            randomised = "'" + randomised + "'"

        return randomised