4 курс 1 семестр / Лабораторные / Python / LR7_10

import psycopg2
import pandas as pd

connection = psycopg2.connect(database="students",
                              user="postgres",
                              password="123",
                              host="127.0.0.1",
                              port="5432")

cursor = connection.cursor()

#код функции
postgresql_func = """
CREATE OR REPLACE FUNCTION average_salary_by_location1(location varchar)
RETURNS float AS $$
BEGIN
RETURN (
  SELECT AVG(salary)
  FROM employees e
  JOIN locations l ON e.location_id = l.location_id
  WHERE l.city = location
);
END;
$$ LANGUAGE plpgsql;
"""
cursor.execute(postgresql_func) # выполнение запроса

connection.commit() #внесение изменений в БД
connection.close() # закрытие соединения
cursor.close() # закрытие cursor