mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-08-24 17:02:24 +02:00
260 lines
7.8 KiB
Python
260 lines
7.8 KiB
Python
![]() |
#!/usr/bin/python
|
||
|
# -*- coding: utf-8 -*-
|
||
|
# Copyright: Ansible Project
|
||
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||
|
|
||
|
from __future__ import absolute_import, division, print_function
|
||
|
__metaclass__ = type
|
||
|
|
||
|
|
||
|
ANSIBLE_METADATA = {'metadata_version': '1.1',
|
||
|
'status': ['preview'],
|
||
|
'supported_by': 'community'}
|
||
|
|
||
|
|
||
|
DOCUMENTATION = '''
|
||
|
---
|
||
|
module: postgresql_schema
|
||
|
short_description: Add or remove PostgreSQL schema from a remote host
|
||
|
description:
|
||
|
- Add or remove PostgreSQL schema from a remote host.
|
||
|
version_added: "2.3"
|
||
|
options:
|
||
|
name:
|
||
|
description:
|
||
|
- Name of the schema to add or remove.
|
||
|
required: true
|
||
|
database:
|
||
|
description:
|
||
|
- Name of the database to connect to.
|
||
|
default: postgres
|
||
|
login_user:
|
||
|
description:
|
||
|
- The username used to authenticate with.
|
||
|
login_password:
|
||
|
description:
|
||
|
- The password used to authenticate with.
|
||
|
login_host:
|
||
|
description:
|
||
|
- Host running the database.
|
||
|
default: localhost
|
||
|
login_unix_socket:
|
||
|
description:
|
||
|
- Path to a Unix domain socket for local connections.
|
||
|
owner:
|
||
|
description:
|
||
|
- Name of the role to set as owner of the schema.
|
||
|
port:
|
||
|
description:
|
||
|
- Database port to connect to.
|
||
|
default: 5432
|
||
|
state:
|
||
|
description:
|
||
|
- The schema state.
|
||
|
default: present
|
||
|
choices: [ "present", "absent" ]
|
||
|
notes:
|
||
|
- This module uses I(psycopg2), a Python PostgreSQL database adapter. You must ensure that psycopg2 is installed on
|
||
|
the host before using this module. If the remote host is the PostgreSQL server (which is the default case), then PostgreSQL must also be installed
|
||
|
on the remote host. For Ubuntu-based systems, install the C(postgresql), C(libpq-dev), and C(python-psycopg2) packages on the remote host before
|
||
|
using this module.
|
||
|
requirements: [ psycopg2 ]
|
||
|
author: "Flavien Chantelot <contact@flavien.io>"
|
||
|
'''
|
||
|
|
||
|
EXAMPLES = '''
|
||
|
# Create a new schema with name "acme"
|
||
|
- postgresql_schema:
|
||
|
name: acme
|
||
|
|
||
|
# Create a new schema "acme" with a user "bob" who will own it
|
||
|
- postgresql_schema:
|
||
|
name: acme
|
||
|
owner: bob
|
||
|
|
||
|
'''
|
||
|
|
||
|
RETURN = '''
|
||
|
schema:
|
||
|
description: Name of the schema
|
||
|
returned: success, changed
|
||
|
type: string
|
||
|
sample: "acme"
|
||
|
'''
|
||
|
|
||
|
import traceback
|
||
|
|
||
|
try:
|
||
|
import psycopg2
|
||
|
import psycopg2.extras
|
||
|
except ImportError:
|
||
|
postgresqldb_found = False
|
||
|
else:
|
||
|
postgresqldb_found = True
|
||
|
|
||
|
from ansible.module_utils.basic import AnsibleModule
|
||
|
from ansible.module_utils.database import SQLParseError, pg_quote_identifier
|
||
|
from ansible.module_utils._text import to_native
|
||
|
|
||
|
|
||
|
class NotSupportedError(Exception):
|
||
|
pass
|
||
|
|
||
|
|
||
|
# ===========================================
|
||
|
# PostgreSQL module specific support methods.
|
||
|
#
|
||
|
|
||
|
def set_owner(cursor, schema, owner):
|
||
|
query = "ALTER SCHEMA %s OWNER TO %s" % (
|
||
|
pg_quote_identifier(schema, 'schema'),
|
||
|
pg_quote_identifier(owner, 'role'))
|
||
|
cursor.execute(query)
|
||
|
return True
|
||
|
|
||
|
|
||
|
def get_schema_info(cursor, schema):
|
||
|
query = """
|
||
|
SELECT schema_owner AS owner
|
||
|
FROM information_schema.schemata
|
||
|
WHERE schema_name = %(schema)s
|
||
|
"""
|
||
|
cursor.execute(query, {'schema': schema})
|
||
|
return cursor.fetchone()
|
||
|
|
||
|
|
||
|
def schema_exists(cursor, schema):
|
||
|
query = "SELECT schema_name FROM information_schema.schemata WHERE schema_name = %(schema)s"
|
||
|
cursor.execute(query, {'schema': schema})
|
||
|
return cursor.rowcount == 1
|
||
|
|
||
|
|
||
|
def schema_delete(cursor, schema):
|
||
|
if schema_exists(cursor, schema):
|
||
|
query = "DROP SCHEMA %s" % pg_quote_identifier(schema, 'schema')
|
||
|
cursor.execute(query)
|
||
|
return True
|
||
|
else:
|
||
|
return False
|
||
|
|
||
|
|
||
|
def schema_create(cursor, schema, owner):
|
||
|
if not schema_exists(cursor, schema):
|
||
|
query_fragments = ['CREATE SCHEMA %s' % pg_quote_identifier(schema, 'schema')]
|
||
|
if owner:
|
||
|
query_fragments.append('AUTHORIZATION %s' % pg_quote_identifier(owner, 'role'))
|
||
|
query = ' '.join(query_fragments)
|
||
|
cursor.execute(query)
|
||
|
return True
|
||
|
else:
|
||
|
schema_info = get_schema_info(cursor, schema)
|
||
|
if owner and owner != schema_info['owner']:
|
||
|
return set_owner(cursor, schema, owner)
|
||
|
else:
|
||
|
return False
|
||
|
|
||
|
|
||
|
def schema_matches(cursor, schema, owner):
|
||
|
if not schema_exists(cursor, schema):
|
||
|
return False
|
||
|
else:
|
||
|
schema_info = get_schema_info(cursor, schema)
|
||
|
if owner and owner != schema_info['owner']:
|
||
|
return False
|
||
|
else:
|
||
|
return True
|
||
|
|
||
|
# ===========================================
|
||
|
# Module execution.
|
||
|
#
|
||
|
|
||
|
|
||
|
def main():
|
||
|
module = AnsibleModule(
|
||
|
argument_spec=dict(
|
||
|
login_user=dict(default="postgres"),
|
||
|
login_password=dict(default="", no_log=True),
|
||
|
login_host=dict(default=""),
|
||
|
login_unix_socket=dict(default=""),
|
||
|
port=dict(default="5432"),
|
||
|
schema=dict(required=True, aliases=['name']),
|
||
|
owner=dict(default=""),
|
||
|
database=dict(default="postgres"),
|
||
|
state=dict(default="present", choices=["absent", "present"]),
|
||
|
),
|
||
|
supports_check_mode=True
|
||
|
)
|
||
|
|
||
|
if not postgresqldb_found:
|
||
|
module.fail_json(msg="the python psycopg2 module is required")
|
||
|
|
||
|
schema = module.params["schema"]
|
||
|
owner = module.params["owner"]
|
||
|
state = module.params["state"]
|
||
|
database = module.params["database"]
|
||
|
changed = False
|
||
|
|
||
|
# To use defaults values, keyword arguments must be absent, so
|
||
|
# check which values are empty and don't include in the **kw
|
||
|
# dictionary
|
||
|
params_map = {
|
||
|
"login_host": "host",
|
||
|
"login_user": "user",
|
||
|
"login_password": "password",
|
||
|
"port": "port"
|
||
|
}
|
||
|
kw = dict((params_map[k], v) for (k, v) in module.params.items()
|
||
|
if k in params_map and v != '')
|
||
|
|
||
|
# If a login_unix_socket is specified, incorporate it here.
|
||
|
is_localhost = "host" not in kw or kw["host"] == "" or kw["host"] == "localhost"
|
||
|
if is_localhost and module.params["login_unix_socket"] != "":
|
||
|
kw["host"] = module.params["login_unix_socket"]
|
||
|
|
||
|
try:
|
||
|
db_connection = psycopg2.connect(database=database, **kw)
|
||
|
# Enable autocommit so we can create databases
|
||
|
if psycopg2.__version__ >= '2.4.2':
|
||
|
db_connection.autocommit = True
|
||
|
else:
|
||
|
db_connection.set_isolation_level(psycopg2
|
||
|
.extensions
|
||
|
.ISOLATION_LEVEL_AUTOCOMMIT)
|
||
|
cursor = db_connection.cursor(
|
||
|
cursor_factory=psycopg2.extras.DictCursor)
|
||
|
except Exception as e:
|
||
|
module.fail_json(msg="unable to connect to database: %s" % to_native(e), exception=traceback.format_exc())
|
||
|
|
||
|
try:
|
||
|
if module.check_mode:
|
||
|
if state == "absent":
|
||
|
changed = not schema_exists(cursor, schema)
|
||
|
elif state == "present":
|
||
|
changed = not schema_matches(cursor, schema, owner)
|
||
|
module.exit_json(changed=changed, schema=schema)
|
||
|
|
||
|
if state == "absent":
|
||
|
try:
|
||
|
changed = schema_delete(cursor, schema)
|
||
|
except SQLParseError as e:
|
||
|
module.fail_json(msg=to_native(e), exception=traceback.format_exc())
|
||
|
|
||
|
elif state == "present":
|
||
|
try:
|
||
|
changed = schema_create(cursor, schema, owner)
|
||
|
except SQLParseError as e:
|
||
|
module.fail_json(msg=to_native(e), exception=traceback.format_exc())
|
||
|
except NotSupportedError as e:
|
||
|
module.fail_json(msg=to_native(e), exception=traceback.format_exc())
|
||
|
except SystemExit:
|
||
|
# Avoid catching this on Python 2.4
|
||
|
raise
|
||
|
except Exception as e:
|
||
|
module.fail_json(msg="Database query failed: %s" % to_native(e), exception=traceback.format_exc())
|
||
|
|
||
|
module.exit_json(changed=changed, schema=schema)
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
main()
|