View file File name : influxdb_database.py Content :""" Management of Influxdb databases ================================ (compatible with InfluxDB version 0.9+) """ def __virtual__(): """ Only load if the influxdb module is available """ if "influxdb.db_exists" in __salt__: return "influxdb_database" return (False, "influxdb module could not be loaded") def present(name, **client_args): """ Ensure that given database is present. name Name of the database to create. """ ret = { "name": name, "changes": {}, "result": True, "comment": "Database {} is already present".format(name), } if not __salt__["influxdb.db_exists"](name, **client_args): if __opts__["test"]: ret["result"] = None ret["comment"] = "Database {} is absent and will be created".format(name) return ret if __salt__["influxdb.create_db"](name, **client_args): ret["comment"] = "Database {} has been created".format(name) ret["changes"][name] = "Present" return ret else: ret["comment"] = "Failed to create database {}".format(name) ret["result"] = False return ret return ret def absent(name, **client_args): """ Ensure that given database is absent. name Name of the database to remove. """ ret = { "name": name, "changes": {}, "result": True, "comment": "Database {} is not present".format(name), } if __salt__["influxdb.db_exists"](name, **client_args): if __opts__["test"]: ret["result"] = None ret["comment"] = "Database {} is present and needs to be removed".format( name ) return ret if __salt__["influxdb.drop_db"](name, **client_args): ret["comment"] = "Database {} has been removed".format(name) ret["changes"][name] = "Absent" return ret else: ret["comment"] = "Failed to remove database {}".format(name) ret["result"] = False return ret return ret