Python set_meta_data Examples

Example #1

File: 83e4121b299_schema_v5.py Project: AliBawazeEer/king-phisher

def downgrade():
	op.add_column('messages', sqlalchemy.Column('company_name', sqlalchemy.String))

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 4, session=session)
	session.commit()

Example #2

File: a695de64338_schema_v6.py Project: AliBawazeEer/king-phisher

def downgrade():
	op.drop_table('authenticated_sessions')

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 5, session=session)
	session.commit()

Example #3

def downgrade():
	op.drop_table('storage_data')

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 6, session=session)
	session.commit()

Example #4

File: 83e4121b299_schema_v5.py Project: AliBawazeEer/king-phisher

def upgrade():
	op.drop_column('messages', 'company_name')

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 5, session=session)
	session.commit()

Example #5

def downgrade():
    op.drop_table('authenticated_sessions')

    db_manager.Session.remove()
    db_manager.Session.configure(bind=op.get_bind())
    session = db_manager.Session()
    db_manager.set_meta_data('schema_version', 5, session=session)
    session.commit()

Example #6

File: 83e4121b299_schema_v5.py Project: nebooben/king-phisher

def upgrade():
    op.drop_column("messages", "company_name")

    db_manager.Session.remove()
    db_manager.Session.configure(bind=op.get_bind())
    session = db_manager.Session()
    db_manager.set_meta_data("schema_version", 5, session=session)
    session.commit()

Example #7

File: 7c315088952_schema_v4.py Project: AliBawazeEer/king-phisher

def upgrade():
	op.add_column('campaigns', sqlalchemy.Column('description', sqlalchemy.String))
	op.add_column('messages', sqlalchemy.Column('opener_ip', sqlalchemy.String))
	op.add_column('messages', sqlalchemy.Column('opener_user_agent', sqlalchemy.String))

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 4, session=session)
	session.commit()

Example #8

File: 7c315088952_schema_v4.py Project: AliBawazeEer/king-phisher

def downgrade():
	op.drop_column('campaigns', 'description')
	op.drop_column('messages', 'opener_ip')
	op.drop_column('messages', 'opener_user_agent')

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 3, session=session)
	session.commit()

Example #9

def downgrade():
    op.drop_column('campaigns', 'description')
    op.drop_column('messages', 'opener_ip')
    op.drop_column('messages', 'opener_user_agent')

    db_manager.Session.remove()
    db_manager.Session.configure(bind=op.get_bind())
    session = db_manager.Session()
    db_manager.set_meta_data('schema_version', 3, session=session)
    session.commit()

Example #10

File: database.py Project: oJollyRogero/king-phisher

	def test_set_meta_data(self):
		self._init_db()
		# set a new value
		key = random_string(10)
		value = random_string(20)
		db_manager.set_meta_data(key, value)
		self.assertEqual(db_manager.get_meta_data(key), value)

		# update an existing value
		value = random_string(30)
		db_manager.set_meta_data(key, value)
		self.assertEqual(db_manager.get_meta_data(key), value)

Example #11

File: database.py Project: s2e-museum/king-phisher

    def test_set_meta_data(self):
        self._init_db()
        # set a new value
        key = random_string(10)
        value = random_string(20)
        db_manager.set_meta_data(key, value)
        self.assertEqual(db_manager.get_meta_data(key), value)

        # update an existing value
        value = random_string(30)
        db_manager.set_meta_data(key, value)
        self.assertEqual(db_manager.get_meta_data(key), value)

Example #12

def upgrade():
    op.add_column('campaigns',
                  sqlalchemy.Column('description', sqlalchemy.String))
    op.add_column('messages', sqlalchemy.Column('opener_ip',
                                                sqlalchemy.String))
    op.add_column('messages',
                  sqlalchemy.Column('opener_user_agent', sqlalchemy.String))

    db_manager.Session.remove()
    db_manager.Session.configure(bind=op.get_bind())
    session = db_manager.Session()
    db_manager.set_meta_data('schema_version', 4, session=session)
    session.commit()

Example #13

File: a695de64338_schema_v6.py Project: AliBawazeEer/king-phisher

def upgrade():
	op.create_table(
		'authenticated_sessions',
		sqlalchemy.Column('id', sqlalchemy.String, primary_key=True),
		sqlalchemy.Column('created', sqlalchemy.Integer, nullable=False),
		sqlalchemy.Column('last_seen', sqlalchemy.Integer, nullable=False),
		sqlalchemy.Column('user_id', sqlalchemy.String, sqlalchemy.ForeignKey('users.id'), nullable=False)
	)

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 6, session=session)
	session.commit()

Example #14

File: 24a4a626ff7c_schema_v3.py Project: AliBawazeEer/king-phisher

def upgrade():
	op.create_table(
		'campaign_types',
		sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
		sqlalchemy.Column('name', sqlalchemy.String, nullable=False),
		sqlalchemy.Column('description', sqlalchemy.String)
	)

	op.create_table(
		'company_departments',
		sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
		sqlalchemy.Column('name', sqlalchemy.String, nullable=False),
		sqlalchemy.Column('description', sqlalchemy.String)
	)

	op.create_table(
		'industries',
		sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
		sqlalchemy.Column('name', sqlalchemy.String, nullable=False),
		sqlalchemy.Column('description', sqlalchemy.String)
	)

	op.create_table(
		'companies',
		sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
		sqlalchemy.Column('name', sqlalchemy.String, unique=True, nullable=False),
		sqlalchemy.Column('description', sqlalchemy.String),
		sqlalchemy.Column('industry_id', sqlalchemy.Integer, sqlalchemy.ForeignKey('industries.id')),
		sqlalchemy.Column('url_main', sqlalchemy.String),
		sqlalchemy.Column('url_email', sqlalchemy.String),
		sqlalchemy.Column('url_remote_access', sqlalchemy.String)
	)

	alert_subscriptions_type = sqlalchemy.Enum('email', 'sms', name='alert_subscription_type')
	alert_subscriptions_type.create(op.get_bind(), checkfirst=True)
	op.add_column('alert_subscriptions', sqlalchemy.Column('type', alert_subscriptions_type, default='sms', server_default='sms', nullable=False))
	op.add_column('alert_subscriptions', sqlalchemy.Column('mute_timestamp', sqlalchemy.DateTime))
	op.add_column('campaigns', sqlalchemy.Column('campaign_type_id', sqlalchemy.Integer, sqlalchemy.ForeignKey('campaign_types.id')))
	op.add_column('campaigns', sqlalchemy.Column('company_id', sqlalchemy.Integer, sqlalchemy.ForeignKey('companies.id')))
	op.add_column('campaigns', sqlalchemy.Column('expiration', sqlalchemy.DateTime))
	op.add_column('messages', sqlalchemy.Column('company_department_id', sqlalchemy.Integer, sqlalchemy.ForeignKey('company_departments.id')))
	op.add_column('users', sqlalchemy.Column('email_address', sqlalchemy.String))
	op.add_column('users', sqlalchemy.Column('otp_secret', sqlalchemy.String(16)))

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 3, session=session)
	session.commit()

Example #15

def upgrade():
	op.create_table(
		'storage_data',
		sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
		sqlalchemy.Column('created', sqlalchemy.DateTime),
		sqlalchemy.Column('namespace', sqlalchemy.String),
		sqlalchemy.Column('key', sqlalchemy.String, nullable=False),
		sqlalchemy.Column('value', sqlalchemy.Binary)
	)

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 7, session=session)
	session.commit()

Example #16

def upgrade():
    op.create_table(
        'authenticated_sessions',
        sqlalchemy.Column('id', sqlalchemy.String, primary_key=True),
        sqlalchemy.Column('created', sqlalchemy.Integer, nullable=False),
        sqlalchemy.Column('last_seen', sqlalchemy.Integer, nullable=False),
        sqlalchemy.Column('user_id',
                          sqlalchemy.String,
                          sqlalchemy.ForeignKey('users.id'),
                          nullable=False))

    db_manager.Session.remove()
    db_manager.Session.configure(bind=op.get_bind())
    session = db_manager.Session()
    db_manager.set_meta_data('schema_version', 6, session=session)
    session.commit()

Example #17

File: database.py Project: ChaoticFractalComputingCompany/king-phisher

	def test_set_meta_data(self):
		try:
			db_manager.init_database('sqlite://')
		except Exception as error:
			self.fail("failed to initialize the database (error: {0})".format(error.__class__.__name__))

		# set a new value
		key = random_string(10)
		value = random_string(20)
		db_manager.set_meta_data(key, value)
		self.assertEqual(db_manager.get_meta_data(key), value)

		# update an existing value
		value = random_string(30)
		db_manager.set_meta_data(key, value)
		self.assertEqual(db_manager.get_meta_data(key), value)

Example #18

    def test_set_meta_data(self):
        try:
            db_manager.init_database('sqlite://')
        except Exception as error:
            self.fail("failed to initialize the database (error: {0})".format(
                error.__class__.__name__))

        # set a new value
        key = random_string(10)
        value = random_string(20)
        db_manager.set_meta_data(key, value)
        self.assertEqual(db_manager.get_meta_data(key), value)

        # update an existing value
        value = random_string(30)
        db_manager.set_meta_data(key, value)
        self.assertEqual(db_manager.get_meta_data(key), value)

Example #19

File: 24a4a626ff7c_schema_v3.py Project: AliBawazeEer/king-phisher

def downgrade():
	op.drop_column('alert_subscriptions', 'type')
	op.drop_column('alert_subscriptions', 'mute_timestamp')
	op.drop_column('campaigns', 'campaign_type_id')
	op.drop_column('campaigns', 'company_id')
	op.drop_column('campaigns', 'expiration')
	op.drop_column('messages', 'company_department_id')
	op.drop_column('users', 'email_address')
	op.drop_column('users', 'otp_secret')

	op.drop_table('campaign_types')
	op.drop_table('company_departments')
	op.drop_table('companies')
	op.drop_table('industries')

	db_manager.Session.remove()
	db_manager.Session.configure(bind=op.get_bind())
	session = db_manager.Session()
	db_manager.set_meta_data('schema_version', 2, session=session)
	session.commit()

Example #20

def downgrade():
    op.drop_column('alert_subscriptions', 'type')
    op.drop_column('alert_subscriptions', 'mute_timestamp')
    op.drop_column('campaigns', 'campaign_type_id')
    op.drop_column('campaigns', 'company_id')
    op.drop_column('campaigns', 'expiration')
    op.drop_column('messages', 'company_department_id')
    op.drop_column('users', 'email_address')
    op.drop_column('users', 'otp_secret')

    op.drop_table('campaign_types')
    op.drop_table('company_departments')
    op.drop_table('companies')
    op.drop_table('industries')

    db_manager.Session.remove()
    db_manager.Session.configure(bind=op.get_bind())
    session = db_manager.Session()
    db_manager.set_meta_data('schema_version', 2, session=session)
    session.commit()

Example #21

def upgrade():
    op.create_table(
        'campaign_types',
        sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
        sqlalchemy.Column('name', sqlalchemy.String, nullable=False),
        sqlalchemy.Column('description', sqlalchemy.String))

    op.create_table(
        'company_departments',
        sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
        sqlalchemy.Column('name', sqlalchemy.String, nullable=False),
        sqlalchemy.Column('description', sqlalchemy.String))

    op.create_table(
        'industries',
        sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
        sqlalchemy.Column('name', sqlalchemy.String, nullable=False),
        sqlalchemy.Column('description', sqlalchemy.String))

    op.create_table(
        'companies',
        sqlalchemy.Column('id', sqlalchemy.Integer, primary_key=True),
        sqlalchemy.Column('name',
                          sqlalchemy.String,
                          unique=True,
                          nullable=False),
        sqlalchemy.Column('description', sqlalchemy.String),
        sqlalchemy.Column('industry_id', sqlalchemy.Integer,
                          sqlalchemy.ForeignKey('industries.id')),
        sqlalchemy.Column('url_main', sqlalchemy.String),
        sqlalchemy.Column('url_email', sqlalchemy.String),
        sqlalchemy.Column('url_remote_access', sqlalchemy.String))

    alert_subscriptions_type = sqlalchemy.Enum('email',
                                               'sms',
                                               name='alert_subscription_type')
    alert_subscriptions_type.create(op.get_bind(), checkfirst=True)
    op.add_column(
        'alert_subscriptions',
        sqlalchemy.Column('type',
                          alert_subscriptions_type,
                          default='sms',
                          server_default='sms',
                          nullable=False))
    op.add_column('alert_subscriptions',
                  sqlalchemy.Column('mute_timestamp', sqlalchemy.DateTime))
    op.add_column(
        'campaigns',
        sqlalchemy.Column('campaign_type_id', sqlalchemy.Integer,
                          sqlalchemy.ForeignKey('campaign_types.id')))
    op.add_column(
        'campaigns',
        sqlalchemy.Column('company_id', sqlalchemy.Integer,
                          sqlalchemy.ForeignKey('companies.id')))
    op.add_column('campaigns',
                  sqlalchemy.Column('expiration', sqlalchemy.DateTime))
    op.add_column(
        'messages',
        sqlalchemy.Column('company_department_id', sqlalchemy.Integer,
                          sqlalchemy.ForeignKey('company_departments.id')))
    op.add_column('users', sqlalchemy.Column('email_address',
                                             sqlalchemy.String))
    op.add_column('users',
                  sqlalchemy.Column('otp_secret', sqlalchemy.String(16)))

    db_manager.Session.remove()
    db_manager.Session.configure(bind=op.get_bind())
    session = db_manager.Session()
    db_manager.set_meta_data('schema_version', 3, session=session)
    session.commit()