-
Notifications
You must be signed in to change notification settings - Fork 0
/
blackbin.py
113 lines (106 loc) · 5.05 KB
/
blackbin.py
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
import os
from dotenv import load_dotenv
import time
from datetime import datetime, timedelta
from O365 import Account
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.select import Select
from selenium.webdriver.support.ui import WebDriverWait
class BlackBin:
def __init__(self):
load_dotenv() # Load environment variables from .env file
client_id = os.getenv("CLIENT_ID")
client_secret = os.getenv("CLIENT_SECRET")
self.credentials = (client_id, client_secret)
self.options = webdriver.ChromeOptions()
self.options.add_argument("--disable-dev-shm-usage")
self.options.add_argument("--disable-extensions")
self.options.add_argument("--disable-gpu")
# self.options.add_argument("--no-sandbox")
self.options.add_argument("--headless")
self.options.add_experimental_option("excludeSwitches", ['enable-automation'])
user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 " \
"(KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36"
self.options.add_argument('--user-agent={}'.format(user_agent))
self.url = "https://online.belfastcity.gov.uk/find-bin-collection-day/Default.aspx"
self.year = int()
self.month = int()
self.day = int()
def start_chrome(self):
self.driver = webdriver.Remote("http://selenium-server:4444/wd/hub", options=self.options)
def get_bin(self, house_address):
house_address = house_address.upper()
postcode = house_address.split(', ')[2]
self.driver.get(self.url)
time.sleep(1)
# user_agent = self.driver.execute_script("return navigator.userAgent;")
# print(user_agent)
WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable((By.CSS_SELECTOR, "label[for='searchBy_radio_1']"))).click()
WebDriverWait(self.driver, 10).until(EC.presence_of_element_located((By.ID, "Postcode_textbox"))).send_keys(postcode)
self.driver.find_element(By.ID, "AddressLookup_button").click()
select = Select(self.driver.find_element(By.ID, "lstAddresses"))
try:
select.select_by_visible_text(house_address)
self.driver.find_element(By.ID, "SelectAddress_button").click()
try:
table = self.driver.find_element(By.ID, "ItemsGrid").find_elements(By.TAG_NAME, "tr")[1].text
table = table.split(' ')
del table[:3]
month = table[3]
months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun',
'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
if month in months:
self.month = months.index(month) + 1
self.day = int(table[4])
self.year = int(table[5])
print(table)
except NoSuchElementException:
print("The Information is Is Missing From Belfast City Council Website")
info = self.driver.find_element(By.ID, "BinDetailsPnl").text
print(info)
self.get_exit()
quit()
except NoSuchElementException:
print("The Address Is Incorrect!")
print(house_address)
self.get_exit()
quit()
def get_exit(self):
self.driver.quit()
def update_calendar(self, calendar_name=None):
account = Account(self.credentials)
schedule = account.schedule()
if not calendar_name:
calendar = schedule.get_default_calendar()
else:
calendar = schedule.get_calendar(calendar_name=calendar_name)
collection_start = datetime(self.year, self.month, self.day, 0, 0)
collection_end = collection_start + timedelta(days=1)
q = calendar.new_query('start').greater_equal(collection_start)
q.chain('and').on_attribute('end').less_equal(collection_end)
events = calendar.get_events(query=q, include_recurring=False)
for e in events:
if e.subject == 'Bin collection':
print("The Event " + "#" + e.subject + "#" + " Is Already In The Calendar")
quit()
collection = calendar.new_event() # creates a new unsaved event
collection.subject = 'Bin collection'
collection.location = 'Belfast'
collection.start = collection_start
collection.is_all_day = True
collection.remind_before_minutes = 360
print(collection)
collection.save()
if __name__ == '__main__':
# "House_Number Street Name (like: 3 Anna Street), Belfast, POST_CODE"
load_dotenv() # Ensure environment variables are loaded
house = os.getenv("HOUSE_ADDRESS")
calendar_name = os.getenv("CALENDAR_NAME", "Events") # Load event name from .env or use default
bins = BlackBin()
bins.start_chrome()
bins.get_bin(house)
bins.get_exit()
bins.update_calendar(calendar_name)