Спринт 3 - Тема 2 - Урок 9 - Задание 1

import pandas as pd
import psycopg2

# Подключение к базе данных
conn = psycopg2.connect(
    host="localhost",
    port=5432,
    dbname="de",
    user="jovyan",
    password="jovyan"
)
cur = conn.cursor()

# Загрузка данных из файла в DataFrame
user_orders_log = pd.read_csv('/lessons/2. Анализ вводных по задаче/7. Использование файлов и подключение к БД/Задание 1/stage/user_order_log.csv')

# SQL-шаблон для вставки данных
insert_uol = """
    INSERT INTO
        stage.user_order_log
    (
        id, uniq_id, date_time, city_id, city_name, customer_id, first_name, last_name, item_id, item_name, quantity, payment_amount
    )
    VALUES
        {uol_val};
"""

# Загрузка данных по частям
step = int(user_orders_log.shape[0] / 100)
i = 0
while i <= user_orders_log.shape[0]:
    print(i, end='\r')
    uol_val = str([tuple(x) for x in user_orders_log.loc[i:i + step].to_numpy()])[1:-1]

    print(uol_val)

    cur.execute(insert_uol.replace('{uol_val}', uol_val))
    conn.commit()
    i += step + 1

# Закрытие соединения с базой данных
cur.close()
conn.close()