DriveBot/main.py at master · PushingCoder572/DriveBot · 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
from selenium import webdriver
from selenium.webdriver import ChromeOptions
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from sys import exit
from time import sleep
from datetime import datetime
import smtplib, ssl
from os import environ

chrome_options = ChromeOptions()
chrome_options.add_argument('--headless')
chrome_options.add_argument('--no-sandbox')
chrome_options.add_argument('--disable-dev-shm-usage')
driver = webdriver.Chrome(options=chrome_options)


locations = ["Sollentuna".encode(), "Farsta".encode(), "Järfälla".encode(), "Uppsala".encode(), "Södertälje".encode()]


class two_or_more_elements(object):
    def __init__(self, tag):
        self.tag = tag

    def __call__(self, driver):
        temp = driver.find_elements_by_tag_name(self.tag)
        if len(temp) > 1:
            return temp
        else:
            return False


def main(plats):
    driver.get("https://fp.trafikverket.se/boka/#/search/amAAMtMoTeIpM/5/0/0/0")
    id_list = ["examination-type-select", "id-control-searchText-1-1", "vehicle-select"]
    for index, id in enumerate(id_list):
        try:
            element = WebDriverWait(driver, 10).until(
                EC.presence_of_element_located((By.ID, id))
            )
        except:
            driver.quit()
            exit()

        if index == 1:
            element.clear()
            element.send_keys(plats)
        else:
            select = Select(element)
            select.select_by_index(2)

        sleep(0.5)

    driver.find_element_by_id("id-control-searchText-1-1").send_keys(Keys.RETURN)

    wait = WebDriverWait(driver, 10)
    lista_av_tider = wait.until(two_or_more_elements("strong"))

    return lista_av_tider[1].text, plats


sender_email = "DevGodTom1337@gmail.com"
receiver_email = "tom.rehnstrom@student.tabyenskilda.se"
message = """\
Subject: NY TID i {}

HEJ DETTA E ETT MEDDELANDE ATT JAG HITTAT EN NY TID {} i {}."""


def send_email(info):
    port = 465  # For SSL
    password = str(environ.get("EMAIL_PASS")).strip()

    # Create a secure SSL context
    context = ssl.create_default_context()

    with smtplib.SMTP_SSL("smtp.gmail.com", port, context=context) as server:
        server.login("DevGodTom1337@gmail.com", password)
        server.sendmail(sender_email, receiver_email, message.format(info[1], info[0], info[1]))


if __name__ == '__main__':
    iterations = 0
    while iterations < 5:
        with open("tid", "r+") as file:

            pepehandslist = []
            for location in locations:
                tid, plats = main(location.decode())
                date_tid = datetime.strptime(tid.strip(), "%Y-%m-%d %H:%M")
                pepehandslist.append([date_tid, plats])

            pepehandslist.sort()

            file.seek(0)

            for i in pepehandslist:
                print(i[0].strftime("%Y-%m-%d %H:%M"))

            if pepehandslist[0][0] < datetime.strptime(file.readline().strip(), "%Y-%m-%d %H:%M"):
                send_email(pepehandslist[0])
                file.truncate(0)
                file.write(pepehandslist[0][0].strftime("%Y-%m-%d %H:%M"))
                print("WIN", pepehandslist[0][0].strftime("%Y-%m-%d %H:%M"))
            else:
                print("FAIL", pepehandslist[0][0].strftime("%Y-%m-%d %H:%M"))

        iterations += 1
        sleep(10)

    driver.quit()