예제 #1
0
    def forest(self, forest_name, host=None, connection=None):
        """
        Get the named forest.
        """
        if host is None:
            if connection is None:
                db = Forest(forest_name,
                            connection=self.connection,
                            save_connection=self.save_connection)
            else:
                db = Forest(forest_name,
                            connection=connection,
                            save_connection=False)
        else:
            if connection is None:
                db = Forest(forest_name,
                            host.host_name(),
                            connection=self.connection,
                            save_connection=self.save_connection)
            else:
                db = Forest(forest_name,
                            host.host_name(),
                            connection=connection,
                            save_connection=False)

        if connection is None:
            return db.read(self.connection)
        else:
            return db.read(connection)
예제 #2
0
    def create(self, args, config, connection):
        forest = Forest(args['name'],
                        args['forest_host'],
                        connection=connection)
        if forest.exists():
            self.logger.error("Forest already exists: {0}".format(
                args['name']))
            sys.exit(1)

        if args['json'] is not None:
            forest = self._read(args['name'], args['json'])
            if forest.host() is None:
                forest.set_host(args['forest_host'])

        self._properties(forest, args)
        dbname = forest.database()

        if dbname is not None:
            database = Database(dbname)
            database.read(connection)
        else:
            database = None

        self.logger.info("Create forest {0}...".format(args['name']))
        forest.create(connection=connection)

        if database is not None:
            database.add_forest_name(forest.forest_name())
            database.update(connection)
예제 #3
0
    def get(self, args, config, connection):
        forest = Forest(args['name'], connection=connection)
        if not forest.exists():
            print("Error: Forest does not exist: {0}".format(args['name']))
            sys.exit(1)

        forest.read()
        self.jprint(forest)
예제 #4
0
    def delete(self, args, config, connection):
        forest = Forest(args['name'],
                        args['forest_host'],
                        connection=connection)
        if not forest.exists():
            return

        level = args['level']
        replicas = args['replicas']

        print("Delete forest {0}...".format(args['name']))
        forest.delete(level, replicas, connection)
예제 #5
0
    def modify(self, args, config, connection):
        name = args['name']
        forest = Forest(name, connection=connection)
        if not forest.exists():
            print("Error: Forest does not exist: {0}".format(name))
            sys.exit(1)

        if args['json'] is not None:
            forest = self._read(None, args['json'], connection=connection)
            if forest.host() is None and 'forest_host' in args:
                forest.set_host(args['forest_host'])
            forest.name = name

        self._properties(forest, args)
        print("Modify forest {0}...".format(name))
        forest.update(connection=connection)
예제 #6
0
    def create(self, args, config, connection):
        name = args['name']
        host = args['forest_host']

        if args['json'] is not None:
            forest = self._read(name, args['json'], connection=connection)
            name = forest.forest_name()
            host = forest.host()
        else:
            forest = Forest(name, host, connection=connection)

        if forest.exists():
            self.logger.error("Forest already exists: {0}".format(name))
            sys.exit(1)

        self._properties(forest, args)
        dbname = forest.database()

        # Strip out properties that we know the server will reject
        cluster = LocalCluster(connection)
        cluster.read()
        if cluster.security_version() is None:
            for key in ['database-replication', 'failover-enable']:
                if key in forest._config:
                    del (forest._config[key])
                    self.logger.debug(
                        "Ignoring {0}, not supported by server".format(key))

        if dbname is not None:
            database = Database(dbname)
            database.read(connection)
        else:
            database = None

        self.logger.info("Create forest {0}...".format(name))
        forest.create(connection=connection)

        if database is not None:
            database.add_forest_name(forest.forest_name())
            database.update(connection)
        user.set_password(base64.urlsafe_b64encode(os.urandom(32)).decode('utf-8'))
        print("{0} user: {1}".format(verb, name))
        user.create(conn)
    else:
        verb = "Updating"
        user = User.unmarshal(config)
        print("{0} user: {1}".format(verb, name))
        user.update(conn)

# Create forests
for config in data['forests']:
    name = config['forest-name']
    f = Forest.lookup(conn, name)
    if f is None:
        print("Need to create forest: {0}".format(name))
        f = Forest(name)
        f.create(conn)

# Update forests
for config in data['forests']:
    name = config['forest-name']
    f = Forest.unmarshal(config)
    print("Updating forest: {0}".format(name))
    f.update(conn)

# Create databases
for config in data['databases']:
    name = config['database-name']
    db = Database.lookup(conn, name)
    if db is None:
        print("Need to create database: {0}".format(name))