VhyIddlmZmZmZeZdZdZdZddl m Z ddl m Z ddl mZddlmZmZmZmZmZmZmZmZGd d eZd Zed k(rey y ))absolute_importdivisionprint_functionag --- module: postgresql_tablespace short_description: Add or remove PostgreSQL tablespaces from remote hosts description: - Adds or removes PostgreSQL tablespaces from remote hosts. options: tablespace: description: - Name of the tablespace to add or remove. required: true type: str aliases: - name location: description: - Path to the tablespace directory in the file system. - Ensure that the location exists and has right privileges. type: path aliases: - path state: description: - Tablespace state. - I(state=present) implies the tablespace must be created if it doesn't exist. - I(state=absent) implies the tablespace must be removed if present. I(state=absent) is mutually exclusive with I(location), I(owner), i(set). - See the Notes section for information about check mode restrictions. type: str default: present choices: [ absent, present ] owner: description: - Name of the role to set as an owner of the tablespace. - If this option is not specified, the tablespace owner is a role that creates the tablespace. type: str set: description: - Dict of tablespace options to set. Supported from PostgreSQL 9.0. - For more information see U(https://www.postgresql.org/docs/current/sql-createtablespace.html). - When reset is passed as an option's value, if the option was set previously, it will be removed. type: dict rename_to: description: - DEPRECATED (see the L(discussion,https://github.com/ansible-collections/community.postgresql/issues/820)). This option will be removed in version 5.0.0. To rename a tablespace, use the M(community.postgresql.postgresql_query) module. - New name of the tablespace. - The new name cannot begin with pg_, as such names are reserved for system tablespaces. type: str session_role: description: - Switch to session_role after connecting. The specified session_role must be a role that the current login_user is a member of. - Permissions checking for SQL commands is carried out as though the session_role were the one that had logged in originally. type: str login_db: description: - Name of database to connect to and run queries against. - The V(db) alias is deprecated and will be removed in version 5.0.0. type: str aliases: - db trust_input: description: - If C(false), check whether values of parameters I(tablespace), I(location), I(owner), I(rename_to), I(session_role), I(settings_list) are potentially dangerous. - It makes sense to use C(false) only when SQL injections via the parameters are possible. type: bool default: true version_added: '0.2.0' comment: description: - Sets a comment on the tablespace. - To reset the comment, pass an empty string. type: str version_added: '3.3.0' attributes: check_mode: support: partial details: - I(state=absent) and I(state=present) (the second one if the tablespace doesn't exist) do not support check mode because the corresponding PostgreSQL DROP and CREATE TABLESPACE commands can not be run inside the transaction block. seealso: - name: PostgreSQL tablespaces description: General information about PostgreSQL tablespaces. link: https://www.postgresql.org/docs/current/manage-ag-tablespaces.html - name: CREATE TABLESPACE reference description: Complete reference of the CREATE TABLESPACE command documentation. link: https://www.postgresql.org/docs/current/sql-createtablespace.html - name: ALTER TABLESPACE reference description: Complete reference of the ALTER TABLESPACE command documentation. link: https://www.postgresql.org/docs/current/sql-altertablespace.html - name: DROP TABLESPACE reference description: Complete reference of the DROP TABLESPACE command documentation. link: https://www.postgresql.org/docs/current/sql-droptablespace.html author: - Flavien Chantelot (@Dorn-) - Antoine Levy-Lambert (@antoinell) - Andrew Klychkov (@Andersson007) - Daniele Giudice (@RealGreenDragon) extends_documentation_fragment: - community.postgresql.postgres a - name: Create a new tablespace called acme and set bob as an its owner community.postgresql.postgresql_tablespace: name: acme owner: bob location: /data/foo comment: "Bob's tablespace" - name: Create a new tablespace called bar with tablespace options community.postgresql.postgresql_tablespace: name: bar set: random_page_cost: 1 seq_page_cost: 1 - name: Reset random_page_cost option community.postgresql.postgresql_tablespace: name: bar set: random_page_cost: reset - name: Drop tablespace called bloat community.postgresql.postgresql_tablespace: name: bloat state: absent a queries: description: List of queries that was tried to be executed. returned: success type: str sample: [ "CREATE TABLESPACE bar LOCATION '/incredible/ssd'" ] tablespace: description: Tablespace name. returned: success type: str sample: 'ssd' owner: description: Tablespace owner. returned: success type: str sample: 'Bob' comment: description: Tablespace comment. returned: success type: str sample: 'Test tablespace' options: description: Tablespace options. returned: success type: dict sample: { 'random_page_cost': 1, 'seq_page_cost': 1 } location: description: Path to the tablespace in the file system. returned: success type: str sample: '/incredible/fast/ssd' newname: description: New tablespace name. returned: if existent type: str sample: new_ssd state: description: Tablespace state at the end of execution. returned: success type: str sample: 'present' ) AnsibleModule) iteritems) check_input) connect_to_dbensure_required_libsexec_sqlget_conn_paramspg_cursor_argspostgres_common_argument_specset_autocommit set_commentcLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) PgTablespacea$Class for working with PostgreSQL tablespaces. Args: module (AnsibleModule) -- object of AnsibleModule class cursor (cursor) -- cursor object of psycopg library name (str) -- name of the tablespace Attrs: module (AnsibleModule) -- object of AnsibleModule class cursor (cursor) -- cursor object of psycopg library name (str) -- name of the tablespace exists (bool) -- flag the tablespace exists in the DB or not owner (str) -- tablespace owner location (str) -- path to the tablespace directory in the file system executed_queries (list) -- list of executed queries new_name (str) -- new name for the tablespace opt_not_supported (bool) -- flag indicates a tablespace option is supported or not c||_||_||_d|_d|_i|_d|_g|_d|_d|_ d|_ |jy)NF) modulecursornameexistsownersettingslocationexecuted_queriesnew_nameopt_not_supportedcommentget_info)selfrrrs ~/home/dcms/DCMS/lib/python3.12/site-packages/ansible_collections/community/postgresql/plugins/modules/postgresql_tablespace.py__init__zPgTablespace.__init__sZ       " !&  ct|dd}t|dd}|rd}nd}|s d|_d|z}nd |z}t||d zd |jid }|sd|_y|d dr|d|_|d d|_|d dr3|d dD](}|j d}|d|j |d <*|d dr |d d|_|d d|d dnd|_yy)zGet tablespace information.zjSELECT 1 FROM information_schema.columns WHERE table_name = 'pg_tablespace' AND column_name = 'spcoptions'F)add_to_executedzkSELECT 1 FROM information_schema.columns WHERE table_name = 'pg_tablespace' AND column_name = 'spclocation' spclocationzpg_tablespace_location(t.oid)TzSELECT shobj_description(t.oid, 'pg_tablespace') AS comment, r.rolname, (SELECT Null) spcoptions, %s loc_string FROM pg_catalog.pg_tablespace AS t JOIN pg_catalog.pg_roles AS r ON t.spcowner = r.oid zSELECT shobj_description(t.oid, 'pg_tablespace') AS comment, r.rolname, t.spcoptions, %s loc_string FROM pg_catalog.pg_tablespace AS t JOIN pg_catalog.pg_roles AS r ON t.spcowner = r.oid zWHERE t.spcname = %(name)sr) query_paramsr&rrolname spcoptions= loc_stringrNr) r rrrrsplitrrr)r!optrqueryresis r"r zPgTablespace.get_infosOt>OTV D#DUZ\ $H6H%)D ".19 9E .19 9E tU%AA%+TYY$7PDK q6) DKQ *DJ1vl#Q -/A A*+A$DMM!A$'/1vl# #A| 4 03Ay0A0M3q6),SUDL r$cDd|jd|d}t||dS)zCreate tablespace. Return True if success, otherwise, return False. args: location (str) -- tablespace directory path in the FS zCREATE TABLESPACE "z " LOCATION ''T return_boolrr )r!rr0s r"createzPgTablespace.create+s">BYYQe66r$c8t|d|jzdS)zSDrop tablespace. Return True if success, otherwise, return False. zDROP TABLESPACE "%s"Tr5)r r)r!s r"dropzPgTablespace.drop6s 4tyy@dSSr$cd||jk(ryd|jd|d}t||dS)zSet tablespace owner. Return True if success, otherwise, return False. args: new_owner (str) -- name of a new owner for the tablespace" FALTER TABLESPACE "z " OWNER TO ""Tr5)rrr )r! new_ownerr0s r" set_ownerzPgTablespace.set_owner=s7  " #:>INe66r$c~||jk(ryt|j|d|j||jS)zSet tablespace comment. Return True if success, otherwise, return False. args: comment (str) -- comment to set for the tablespace" F tablespace)rrrrr)r!r check_modes r"rzPgTablespace.set_commentKs; dll "4;;tyy%t'<'<> >r$cRd|jd|d}||_t||dS)zRename tablespace. Return True if success, otherwise, return False. args: newname (str) -- new name for the tablespace" r<z " RENAME TO "r=Tr5)rrr )r!newnamer0s r"renamezPgTablespace.renameYs(;?))WM e66r$c0|jryd}|D]}||dk(r0||jvs|j|}d|j|<;||jvst|||j|k7sh|j |d||d}|S)zSet tablespace settings (options). If some setting has been changed, set changed = True. After all settings list is handling, return changed. args: new_settings (list) -- list of new settings FresetNz = 'r4)rr_PgTablespace__reset_settingstr_PgTablespace__set_setting)r! new_settingschangedr2s r" set_settingszPgTablespace.set_settingses  ! ! QAA') %"2215G'+DMM!$4==(c,q/.BdmmTUFV.V,,A|A-OP Qr$cDd|jd|d}t||dS)zReset tablespace setting. Return True if success, otherwise, return False. args: setting (str) -- string in format "setting_name = 'setting_value'" r<z " RESET ()Tr5r7r!settingr0s r"__reset_settingzPgTablespace.__reset_settings!7;iiIe66r$cDd|jd|d}t||dS)zSet tablespace setting. Return True if success, otherwise, return False. args: setting (str) -- string in format "setting_name = 'setting_value'" r<z" SET (rOTr5r7rPs r" __set_settingzPgTablespace.__set_settings!59IIwGe66r$N)__name__ __module__ __qualname____doc__r#r r8r:r?rrErMrHrJr$r"rrs<& 3Vj 7T 7 > 76 7 7r$rct}|jtdddgtddddgtddg td td tdd gd d ddgtdd dtd tddtdd t|d}|jd}|jd}|jd}|jd}|jd}|jd}|jd}|jd} |jd} |dk(r|s|s|s|r|j d | s9|sd} n"t |D cgc] \} } | d!| } } } t||||||| | t|t||jd"#}t|||jrd"nd$\}}|jd,it}d"}d"}t|||}|jr>|r<||j k7r-|j d%|j"d&|j d' |jsP|dk(rK|r|j d(|z |s|j d) d}t%|d|j'|}n^|jr$|dk(rd}t%|d|j)}n.|jr"|r |j"|k7r|j+|}|dk(r|j-|dk(rl|jr`|r|j/|xs|}|r|j1|xs|}| |j3| |jxs|}|j-|s-|jr|j5n|j7|j9|j9t|d|j"|j:|j<|j>|j |j@*}|dk(r!d|d<|jBr|jB|d+<n |dk(rd|d<|jDd,i|ycc} } w)-NrITr)typerequiredaliasespresentabsent)r[defaultchoicespath)r[r])r[dictdbz5.0.0zcommunity.postgresql)rversioncollection_name)r[r]deprecated_aliases)r[removed_in_versionremoved_from_collectionbool)r[r`) rAstaterrsetlogin_db rename_to session_role trust_inputr) argument_specsupports_check_moderArkrrrnrlrorprzFstate=absent is mutually exclusive location, owner, rename_to, and set)msgz = F)warn_db_default) autocommitz Tablespace 'z"' exists with different location 'r4z/Tablespace %s does not exist, nothing to renamezV'location' parameter must be passed with state=present if the tablespace doesn't exist)rLrkrArqueriesoptionsrrrDrY)#rupdatercrparams fail_jsonrrr r r rBrr rrrrrr8r:rEr r?rMrrollbackcommitcloserrrrr exit_json)rqrrArkrrrnrrorpr settings_listkv conn_params db_connectiondummyrrurLtblspacekws r"mainrsa13MUTF8Dy8Y:OP6F84 f 54&"#9 F Eg/EGu%fd3%.#(# F |,J MM' "E}}Z(H MM' "E k*I}}U#H==0L-- .KmmI&G h%99 :  MrsqA@ m ^ 6) V5.   B76B7TSl zFr$