python_lab1/main.py at main · Mishele123/python_lab1 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
import time
import os
import requests
from fake_useragent import UserAgent


driver = webdriver.Firefox()
driver.implicitly_wait(10)
wait = WebDriverWait(driver, 500)
ua = UserAgent()


def scrapeImages(key: str) -> None:
    max_count = 1100
    numbers_of_scrolls = max_count / 300 + 1
    url = f"https://yandex.ru/images/search?text={key}"
    driver.get(url=url)

    # скороллинг страницы
    # for x in range(int(numbers_of_scrolls)):
    #     for xx in range(10):
    #         driver.execute_script("window.scrollTo(0, 1000000)")
    #         time.sleep(2)
    #     time.sleep(10)
    #     try:
    #         print("Поиск кнопки")
    #         driver.find_element(By.XPATH, "/html/body/div[4]/div[2]/div/div[3]/a").click()
    #         print("Кнопка нажата")
    #     except:
    #         break

    time.sleep(10)

    images_small = driver.find_elements(By.CSS_SELECTOR, "img.serp-item__thumb")
    images_big = driver.find_elements(By.CSS_SELECTOR, "a.serp-item__link")

    print(len(images_big))
    print(len(images_small))

    print("-------------------------------------------")

    for i in range(0, max_count + 1):
        # кликаем на большую картинку
        try:
            images_big[i].click()
            time.sleep(2)
            img_element = driver.find_element(By.CSS_SELECTOR, ".MMImage-Origin")
            img_url_big = img_element.get_attribute("src")

            img_url_small = images_small[i].get_attribute("src")

            # создание имен картинок
            file_name = f"{str(i).rjust(4, '0')}.jpg"

            # Скачивание большой картинки
            img_path_big = os.path.join(f"dataset/big/{key}", file_name)
            responce_big = requests.get(img_url_big, headers={
                "User-Agent" : ua.random,
            }, verify=False)
            with open(img_path_big, "wb") as f:
                f.write(responce_big.content)

            # Закрыть окно с большим изображением
            driver.find_element(By.CSS_SELECTOR, 'div.MMViewerModal-Close').click()

            # Скачивание маленькой картинки
            img_path_small = os.path.join(f"dataset/small/{key}", file_name)
            response_small = requests.get(img_url_small, headers={
                "User-Agent" : ua.random,
            }, verify=False)
            with open(img_path_small, "wb") as f:
                f.write(response_small.content)

            print(f"Скачано изображение {i + 1}/{max_count}")
        except:
            continue

    time.sleep(10)

    driver.close()
    driver.quit()

def createFolders() -> None:
    try:
        os.makedirs("dataset", exist_ok=True)
        os.makedirs("dataset/big", exist_ok=True)
        os.makedirs("dataset/small", exist_ok=True)
        os.makedirs("dataset/big/tiger", exist_ok=True)
        os.makedirs("dataset/big/leopard", exist_ok=True)

        os.makedirs("dataset/small/tiger", exist_ok=True)
        os.makedirs("dataset/small/leopard", exist_ok=True)
    except:
        print("Error")


def main():
    createFolders()
    scrapeImages("tiger")


if __name__ == "__main__":
    main()