Vh5ddlmZmZmZeZdZdZdZddl m Z ddl m Z ddl mZmZddlmZmZmZmZmZmZGd d eZd Zed k(rey y ))absolute_importdivisionprint_functiona --- module: postgresql_copy short_description: Copy data between a file/program and a PostgreSQL table description: - Copy data between a file/program and a PostgreSQL table. options: copy_to: description: - Copy the contents of a table to a file. - Can also copy the results of a SELECT query. - Mutually exclusive with I(copy_from) and I(dst). type: path aliases: [ to ] copy_from: description: - Copy data from a file to a table (appending the data to whatever is in the table already). - Mutually exclusive with I(copy_to) and I(src). type: path aliases: [ from ] src: description: - Copy data from I(copy_from) to I(src=tablename). - Used with I(copy_to) only. type: str aliases: [ source ] dst: description: - Copy data to I(dst=tablename) from I(copy_from=/path/to/data.file). - Used with I(copy_from) only. type: str aliases: [ destination ] columns: description: - List of column names for the src/dst table to COPY FROM/TO. type: list elements: str aliases: [ column ] program: description: - Mark I(src)/I(dst) as a program. Data will be copied to/from a program. - See block Examples and PROGRAM arg description U(https://www.postgresql.org/docs/current/sql-copy.html). type: bool default: false options: description: - Options of COPY command. - See the full list of available options U(https://www.postgresql.org/docs/current/sql-copy.html). type: dict login_db: description: - Name of database to connect to. - The V(db) alias is deprecated and will be removed in version 5.0.0. type: str aliases: [ db ] 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 trust_input: description: - If C(false), check whether values of parameters are potentially dangerous. - It makes sense to use C(false) only when SQL injections are possible. type: bool default: true version_added: '0.2.0' notes: - Supports PostgreSQL version 9.4+. - COPY command is only allowed to database superusers. attributes: check_mode: support: partial details: - If I(check_mode=true), we just check the src/dst table availability and return the COPY query that actually has not been executed. - If i(check_mode=true) and the source has been passed as SQL, the module will execute it and roll the transaction back, but pay attention it can affect database performance (e.g., if SQL collects a lot of data). seealso: - name: COPY command reference description: Complete reference of the COPY command documentation. link: https://www.postgresql.org/docs/current/sql-copy.html author: - Andrew Klychkov (@Andersson007) extends_documentation_fragment: - community.postgresql.postgres a - name: Copy text TAB-separated data from file /tmp/data.txt to acme table community.postgresql.postgresql_copy: copy_from: /tmp/data.txt dst: acme - name: Copy CSV (comma-separated) data from file /tmp/data.csv to columns id, name of table acme community.postgresql.postgresql_copy: copy_from: /tmp/data.csv dst: acme columns: id,name options: format: csv - name: > Copy text vertical-bar-separated data from file /tmp/data.txt to bar table. The NULL values are specified as N community.postgresql.postgresql_copy: copy_from: /tmp/data.csv dst: bar options: delimiter: '|' null: 'N' - name: Copy data from acme table to file /tmp/data.txt in text format, TAB-separated community.postgresql.postgresql_copy: src: acme copy_to: /tmp/data.txt - name: Copy data from SELECT query to/tmp/data.csv in CSV format community.postgresql.postgresql_copy: src: 'SELECT * FROM acme' copy_to: /tmp/data.csv options: format: csv - name: Copy CSV data from my_table to gzip community.postgresql.postgresql_copy: src: my_table copy_to: 'gzip > /tmp/data.csv.gz' program: true options: format: csv - name: > Copy data from columns id, name of table bar to /tmp/data.txt. Output format is text, vertical-bar-separated, NULL as N community.postgresql.postgresql_copy: src: bar columns: - id - name copy_to: /tmp/data.csv options: delimiter: '|' null: 'N' ak queries: description: List of executed queries. returned: success type: str sample: [ "COPY test_table FROM '/tmp/data_file.txt' (FORMAT csv, DELIMITER ',', NULL 'NULL')" ] src: description: Data source. returned: success type: str sample: "mytable" dst: description: Data destination. returned: success type: str sample: "/tmp/data.csv" ) AnsibleModule) iteritems) check_inputpg_quote_identifier) connect_to_dbensure_required_libsexec_sqlget_conn_paramspg_cursor_argspostgres_common_argument_specc.eZdZdZdZdZdZdZdZy) PgCopyDataaImplements behavior of COPY FROM, COPY TO PostgreSQL command. Arguments: module (AnsibleModule) -- object of AnsibleModule class cursor (cursor) -- cursor object of psycopg library Attributes: module (AnsibleModule) -- object of AnsibleModule class cursor (cursor) -- cursor object of psycopg library changed (bool) -- something was changed after execution or not executed_queries (list) -- executed queries dst (str) -- data destination table (when copy_from) src (str) -- data source table (when copy_to) opt_need_quotes (tuple) -- values of these options must be passed to SQL in quotes cf||_||_g|_d|_d|_d|_d|_y)NF) DELIMITERNULLQUOTEESCAPEENCODING)modulecursorexecuted_querieschangeddstsrcopt_need_quotes)selfrrs x/home/dcms/DCMS/lib/python3.12/site-packages/ansible_collections/community/postgresql/plugins/modules/postgresql_copy.py__init__zPgCopyData.__init__s7  "  c|jjd|_|jjd|_dt |jdzg}|jjj dr:|j ddj|jjdz|j d|jjj d r|j d |j d |jz|jjj d r|j |j|jjrX|j|j|_ |jr+|jj d j|yyt|d j|drd|_ yy)z&Implements COPY FROM command behavior. copy_fromrCOPY %stablecolumns(%s),FROMprogramPROGRAM'%s'options T return_boolN)rparamsrrr getappendjoin_PgCopyData__transform_options check_mode_PgCopyData__check_tablerrr r query_fragmentss r!r%zPgCopyData.copy_fromsv;;%%k2;;%%e,$':488W'MMN ;;   ! !) ,  " "6CHHT[[5G5G 5R,S#S Tv& ;;   ! !) ,  " "9 -v01 ;;   ! !) ,  " "4#;#;#= > ;; ! !--dhh7DL||%%,,SXXo-FGchh7TJ# Kr#c@|jjd|_|jjd|_d|jj vrd|jzg}ndt |jdzg}|jjj dr:|jdd j|jjdz|jd |jjj d r|jd |jd |jz|jjj dr|j|j|jjrX|j|j|_ |jr+|jjdj|yyt|dj|drd|_ yy)z$Implements COPY TO command behavior.rcopy_toSELECT z COPY (%s)r&r'r(r)r*TOr,r-r.r/r0Tr1N)rr3rrupperr r4r5r6r7r8r9rrr r:s r!r=zPgCopyData.copy_tos;;%%e,;;%%i0 ( (*TXX56O )+>txx+QQRO ;;   ! !) ,  " "6CHHT[[5G5G 5R,S#S Tt$ ;;   ! !) ,  " "9 -v01 ;;   ! !) ,  " "4#;#;#= > ;; ! !--dhh7DL||%%,,SXXo-FGchh7TJ# Kr#ctt|jjdD]A\}}|j|jvs#d|z|jjd|<Ct|jjdDcgc] \}}|d|}}}ddj |zScc}}w)z.Transform options dict into a suitable string.r/r.r0r)z, )rrr3r@rr6)r keyvalopts r!__transform_optionszPgCopyData.__transform_options)s#DKK$6$6y$AB BJS#yy{d2225;c\ ""9-c2 B7@ @R@RS\@]6^_ c#s#__ #&&`s B4c|d|jvrt||dyt|dt|dzdy)aCheck table or SQL in transaction mode for check_mode. Return True if it is OK. Arguments: table (str) - Table name that needs to be checked. It can be SQL SELECT statement that was passed instead of the table name. r>F)add_to_executedTzSELECT 1 FROM %sr')r@r r )r r's r! __check_tablezPgCopyData.__check_table2sE  % T5% 8),?w,OO!& (r#N) __name__ __module__ __qualname____doc__r"r%r=r7r9r#r!rrs!"  $rslA@ ^ @8 t $5.@@PZz zFr#