Psycopg2 полностью динамический c обновление заявления - PullRequest
Новая
       7

Psycopg2 полностью динамический c обновление заявления

2 голосов
/ 11 марта 2020

В настоящее время я сталкиваюсь с проблемой создания полностью обновленного оператора update c. Мне нужно имя таблицы, столбцы таблицы, значения и возможность изменять всякий раз, когда итерация для l oop. Мне удалось сделать это для оператора вставки, но я не уверен, как можно добиться того же для оператора обновления.

Моя проблема в том, что я не могу понять, как передать два параметра (% s ) к вызову execute_values.

Обратите внимание, что из соображений конфиденциальности, фактическая структура базы данных не отображается, а фактические данные получаются динамически. Для целей тестирования я только что реализовал данные stati c. Таблицы базы данных также имеют гораздо больше полей. 

import psycopg2
from psycopg2 import sql
from psycopg2.extras import execute_values


table_names = ["employee_work_details", "employee_personal_details"]
table_columns = {"employee_work_details": ["employee_id", "employee_name", "employee_department"], "employee_personal_details": ["employee_p_id", "employee_name", "employee_nok", "employee_home_address"]}
table_values = {"employee_work_details": [("012", "Raymond C", "IT_Department")], "employee_personal_details": [("921", "Raymond C", "Tracy C", "123 Wonderland Road")]}
table_where_key = {"employee_work_details": "employee_id", "employee_personal_details": "employee_p_id"}


def establish_postgre_connection():

    postgresql_connect = psycopg2.connect(host=XX.XX.XX.XX, database="company_xxx", user="Raymond", password="xxxx", port="5432")
    postgresql_cursor = postgresql_connect.cursor()

    # Dynamically get some data here...

    postgresql_cursor = database_update(postgresql_cursor)

    postgresql_connect.commit()
    postgresql_cursor.close()


def database_update(postgresql_cursor):

    for table_name in table_names:
        query_table_name = sql.Identifier(table_name)
        query_table_columns = sql.SQL(",").join(map(sql.Identifier, table_columns[table_name]))
        query_table_where_key = sql.Identifier(table_where_key[table_name])
        # Sample Query "UPDATE employee_work_details SET (employee_name,employee_department) = %s WHERE employee_id = %s"
        update_statement = sql.SQL("UPDATE {} SET {} = %s WHERE {} = %s").format(query_table_name, query_table_columns, query_table_where_key)
        execute_values(postgresql_cursor, update_statement, table_values[table_name])  # How do I pass in two %s

    return postgresql_cursor

1 Ответ

2 голосов
/ 18 марта 2020

Вывод ниже основан на примере запроса, который вы упомянули. 

table_names = ["employee_work_details", "employee_personal_details"]
table_columns = {"employee_work_details": ["employee_id", "employee_name", "employee_department"],
                 "employee_personal_details": ["employee_p_id", "employee_name", "employee_nok",
                                               "employee_home_address"]}
table_values = {"employee_work_details": [("012", "Raymond C", "IT_Department")],
                "employee_personal_details": [("921", "Raymond C", "Tracy C", "123 Wonderland Road")]}
table_where_key = {"employee_work_details": "employee_id", "employee_personal_details": "employee_p_id"}


def database_update():
    for table_name in table_names:
        print(f"UPDATE {table_name} SET ({', '.join(table_columns[table_name])}) = %s WHERE {table_where_key[table_name]} = %s")



database_update()

Вывод: 


UPDATE employee_work_details SET (employee_id, employee_name, employee_department) = %s WHERE employee_id = %s
UPDATE employee_personal_details SET (employee_p_id, employee_name, employee_nok, employee_home_address) = %s WHERE employee_p_id = %s

Sql с заданным условием. 

def database_update():
    for table_name in table_names:
        set_sql = ""
        for columns, values in zip(table_columns[table_name], table_values[table_name]):
            set_sql = f"{set_sql}, {columns} = '{values}'" if set_sql else f"{columns} = '{values}'"
        sql = f"UPDATE {table_name} SET {set_sql} WHERE {table_where_key[table_name]} = %s"

        print(sql) # Sql with set condition

Вывод: 

UPDATE employee_work_details SET employee_id = '012', employee_name = 'Raymond C', employee_department = 'IT_Department' WHERE employee_id = %s
UPDATE employee_personal_details SET employee_p_id = '921', employee_name = 'Raymond C', employee_nok = 'Tracy C', employee_home_address = '123 Wonderland Road' WHERE employee_p_id = %s
...