from bs4 import BeautifulSoup
from selenium.webdriver import Chrome
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
import psycopg2
import wget
import time

connection = psycopg2.connect(host='localhost', dbname='dbdata', user='postgres', password='Q1w2e3r4')
cursor = connection.cursor()



s = Service('C:\data\hrome\chromedriver.exe')
browser = webdriver.Chrome(service=s)
browser.get('https://blog.okko.tv/selections/vremya-letit-bystro-10-filmov-kotorye-smotryatsya-na-odnom-dykhanii')
time.sleep (20)
html_text = browser.page_source
soup = BeautifulSoup(html_text, 'lxml')

creat_table="""create table notes
    (id serial primary key, name varchar(150),
    price varchar(150),
    scr varchar(150))"""
cursor.execute(creat_table)
connection.commit()


names= soup.find_all('h2', class_="film-card-title")
prices = soup.find_all('span', class_="f-c-rating")
pictures = soup.find_all('figure', class_="wp-block-image size-large")
for i in range(len(names)):
    url = pictures[i].find('img')['src']
    filename = f"C:\\Users\\rusta\\OneDrive\\Рабочий стол\\pictures111\\{i}.jpg"
    wget.download(url, filename)

    insert_qwery = """INSERT INTO public.notes(name, price, scr)
    VALUES( %s, %s, %s);"""
    record_to_insert = (names[i].text, prices[i].text, filename)
    cursor.execute(insert_qwery, record_to_insert)
    connection.commit()

cursor.close()
connection.close()