From 84d1e0d8787bc9b59d50f67e96845f62bdbbee23 Mon Sep 17 00:00:00 2001
From: Myned <onemyned@gmail.com>
Date: Wed, 14 Oct 2020 19:48:23 -0400
Subject: [PATCH] Add additional website to notify
---
src/cogs/weeb.py | 47 +++++++++++++++++++++++++++++++++--------------
1 file changed, 33 insertions(+), 14 deletions(-)
diff --git a/src/cogs/weeb.py b/src/cogs/weeb.py
index f5910d7..50b46f2 100644
--- a/src/cogs/weeb.py
+++ b/src/cogs/weeb.py
@@ -22,34 +22,53 @@ class Weeb(cmds.Cog):
self.bot.loop.create_task(self.start())
print('STARTED : weebing')
- async def refresh(self, browser, urls):
- message = ''
+ async def refresh_switchmod(self, browser):
+ message = ''
+ urls = {
+ 'Novelties': 'https://switchmod.net/collections/ended-gbs/products/gmk-metaverse-2?variant=31671816880208',
+ 'Royal': 'https://switchmod.net/collections/ended-gbs/products/gmk-metaverse-2?variant=31671816945744'
+ }
- for item, url in urls.items():
- browser.get(url)
- status = browser.find_elements_by_css_selector('#addToCartText-product-template')[0].text
+ for item, url in urls.items():
+ browser.get(url)
+ status = browser.find_elements_by_css_selector('#addToCartText-product-template')[0].text
- if status != 'SOLD OUT':
- message += f'{item} is in stock!\n{url}\n'
+ if status != 'SOLD OUT':
+ message += f'{item} is in stock at Switchmod!\n<{url}>\n'
- return message
+ await asyncio.sleep(5)
+
+ return message
+
+ async def refresh_deskhero(self, browser):
+ message = ''
+ url = 'https://www.deskhero.ca/products/gmk-metaverse-2'
+
+ browser.get(url)
+ royal_soldout = browser.find_elements_by_css_selector('#data-product-option-1-1')[0].get_attribute('data-soldout')
+ novelties_soldout = browser.find_elements_by_css_selector('#data-product-option-1-3')[0].get_attribute('data-soldout')
+
+ if royal_soldout != 'true':
+ message += f'Royal is in stock at Deskhero!\n<{url}>\n'
+ if novelties_soldout != 'true':
+ message += f'Novelties is in stock at Deskhero!\n<{url}>\n'
+
+ return message
async def start(self):
try:
opts = Options()
opts.headless = True
browser = Chrome(executable_path='/usr/bin/chromedriver', options=opts)
- urls = {
- 'Novelties': 'https://switchmod.net/collections/ended-gbs/products/gmk-metaverse-2?variant=31671816880208',
- 'Royal': 'https://switchmod.net/collections/ended-gbs/products/gmk-metaverse-2?variant=31671816945744'
- }
while self.weebing:
- message = await self.refresh(browser, urls)
+ message = await self.refresh_switchmod(browser)
+ await asyncio.sleep(5)
+ message += await self.refresh_deskhero(browser)
if message:
await self.bot.get_user(self.id).send(message)
- await self.bot.get_user(u.config['owner_id']).send('Message sent')
+ await self.bot.get_user(u.config['owner_id']).send('Something is in stock. Restart to keep checking')
browser.quit()
self.weebing = False