def main(argv=sys.argv):
if len(argv) < 2:
usage(argv)
config_uri = argv[1]
#options = parse_vars(argv[2:])
setup_logging(config_uri)
#settings = get_appsettings(config_uri, options=options)
settings = get_appsettings(config_uri)
engine = engine_from_config(settings, 'sqlalchemy.')
DBSession.configure(bind=engine)
Base.metadata.create_all(engine)
with transaction.manager:
# Create the groups for two types of users (for now)
auth_group = Group(u'auth', u'Regular user account.')
admin_group = Group(u'admin', u'Administrative user account.')
# Add them to the db session
DBSession.add(auth_group)
DBSession.add(admin_group)
transaction.commit()
# Create the admin user so we can always access the system as admin.
su = User(first_name=u'Ixia', last_name=u'User', username=u'admin',
email=u'*****@*****.**', mobile=random.randint(2140000000, 2149999999))
su._set_password('admin')
su.groups.append(admin_group)
DBSession.add(su)
transaction.commit()
# Configure global system parameters.
config = GlobalConfigs(bps=u'192.168.0.132', metasploit=u'192.168.0.150', splunk=u'192.168.0.133',
ips=u'', ngfw=u'192.168.0.134', dlp=u'192.168.0.140',
windows=u'192.168.0.132', kali=u'192.168.0.170', atip=u'192.168.0.171', version=u'01.00.00')
DBSession.add(config)
transaction.commit()
def test_by_username(self):
Base.metadata.create_all(self.engine)
with transaction.manager:
user = User.by_username('admin')
self.assertEqual(user.name, 'admin')
self.assertEqual(user.password, 'admin')
