pyMapCollector/pythonUnversalMapCollector.py at main · nikmedoed/pyMapCollector · 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
107
108
109
110
111
112
113
114
115
116
117
from selenium import webdriver
from time import sleep
from PIL import Image
from io import BytesIO
from tqdm import tqdm
from selenium.webdriver.chrome.options import Options
import cutForPrint
from webdriver_manager.chrome import ChromeDriverManager


class MapCollector:
    def __init__(self, initLink, width, height, boundSize=500, pointIsCenter=True, bodyXpath="/html/body", controls=[]):
        chrome_options = Options()
        chrome_options.add_argument("--headless")
        self.browser = webdriver.Chrome(ChromeDriverManager().install(), options=chrome_options)
        self.browser.set_window_size(2000, 2000)
        self.browser.get(initLink)
        self.width = width
        self.height = height
        self.boundSize = boundSize
        self.boundSizeHalf = boundSize // 2
        self.bodyXpath = bodyXpath
        self.body = self.getbody()
        for i in controls:
            try:
                self.browser.find_element_by_class_name(i).click()
            except:
                pass
        if pointIsCenter:
            self.goToStartPoint()
        else:
            self.slideToOffset(self.boundSizeHalf, self.boundSizeHalf)

    def __del__(self):
        self.browser.close()

    def goToStartPoint(self):
        size = self.browser.get_window_size()
        toMoveX, toMoveY = [int(size[x] * 0.9 // 2) for x in ['width', 'height']]
        distanceToStartX = self.width // 2 - self.boundSizeHalf
        distanceToStartY = self.height // 2 - self.boundSizeHalf

        with tqdm(desc="MoveToStart", total=distanceToStartX + distanceToStartY) as pbar:
            while 1:
                toMoveX = min(distanceToStartX, toMoveX)
                toMoveY = min(distanceToStartY, toMoveY)
                self.slideToOffset(toMoveX, toMoveY)
                distanceToStartX -= toMoveX
                distanceToStartY -= toMoveY
                pbar.update(toMoveX + toMoveY)
                if distanceToStartX == 0 and distanceToStartY == 0:
                    break
        print("The starting position is occupied")

    def slideToOffset(self, x, y):
        webdriver.ActionChains(self.browser).click_and_hold(self.getbody()).move_by_offset(x, y).release().perform()
        sleep(0.5)
        return self

    def getbody(self):
        return self.browser.find_element_by_xpath(self.bodyXpath)

    def smallScreenShoot(self, img):
        return img.crop([x // 2 + s for s in [-self.boundSizeHalf, self.boundSizeHalf] for x in img.size])

    def finalShoot(self):
        width = self.width
        height = self.height
        boundSize = self.boundSize
        smallScreenShoot = self.smallScreenShoot
        browser = self.browser

        finalScreenShoot = Image.new('RGB', (width, height))
        sign = 1
        widminusbs = width - boundSize
        reverse = width - width % boundSize
        with tqdm(desc="CollectingMap", total=((height // boundSize + (1 if height % boundSize > 0 else 0)) * (
                width // boundSize + (1 if width % boundSize > 0 else 0)))) as pbar:
            for y in range(0, height, boundSize):
                for x in range(0, width, boundSize):
                    png = browser.get_screenshot_as_png()
                    im = Image.open(BytesIO(png))
                    finalScreenShoot.paste(smallScreenShoot(im), (x if sign > 0 else reverse - x, y))
                    if widminusbs - x > 0:
                        self.slideToOffset(-boundSize * sign, 0)
                    pbar.update(1)
                sign *= -1
                self.slideToOffset(0, -boundSize)
        return finalScreenShoot


if __name__ == "__main__":
    INIT_LINK = "https://yandex.ru/maps/213/moscow/?ll=37.622504%2C55.753215&z=15"  # 14 захватит ЦКАД при размерах ниже
    INIT_LINK = "https://2gis.ru/moscow?m=37.622446%2C55.75377%2F15"

    SCREENSHOOT_WIDTH = 14472
    SCREENSHOOT_HEIGHT = 17060

    SCREENSHOOT_self_boundSize = 500  # Different services have different safe zones

    PARAMETERS = {
        "yandex": {'controls': ["sidebar-toggle-button__icon"], 'path': "/html/body"},
        "2gis": {'controls': ["_k1uvy"], 'path': '//*[@id="root"]/div/div'},
    }

    params = next((x[1] for x in PARAMETERS.items() if x[0] in INIT_LINK), None)

    collector = MapCollector(INIT_LINK, SCREENSHOOT_WIDTH, SCREENSHOOT_HEIGHT, SCREENSHOOT_self_boundSize,
                             bodyXpath=params["path"],
                             pointIsCenter=True,
                             controls=params["controls"]
                             )

    shoot = collector.finalShoot()
    shoot.save("resultmap.png")
    cutForPrint.cut(shoot, 6, 5)
    print("Gotcha")