💰

Author: jeyraof

entrypoint.sh

@@ -3,4 +3,4 @@ set -e
 
 export PATH="$HOME/.local/bin:$PATH"
 
-poetry run python bot.py
+poetry run python "$@"

bot.py renamed to stock.py

@@ -0,0 +1,52 @@
+import disnake
+from disnake.ext import tasks
+from requests_html import HTMLSession
+
+import os
+import json
+
+class Scheduler(disnake.Client):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.last_price = None
+        self.interval = os.getenv('INTERVAL') or 15
+        self.short_code = os.getenv('SHORT_CODE')
+
+    async def on_ready(self):
+        print(f"Logged in as {self.user} (ID: {self.user.id})")
+        print(f"SHORT_CODE: {self.short_code}")
+        print(f"INTERVAL: {self.interval}")
+        print("------")
+
+    async def on_connect(self):
+        self.fetch_price.start()
+    
+    @tasks.loop(seconds=int(os.getenv('INTERVAL') or 15))
+    async def fetch_price(self):
+        await self.wait_until_ready()
+
+        session = HTMLSession()
+        r = session.get(f"https://www.ustockplus.com/stock/_-{self.short_code}")
+
+        try:
+            next_data = r.html.find('#__NEXT_DATA__', first=True).text
+            data = json.loads(next_data)['props']['pageProps']['stockDetail']
+            display_name = data['name']
+
+            trade_price = data['prevClosingPrice']
+
+            price_detail = f"({round(data['changeRate'], 2)}%)"
+
+            if self.user.display_name != display_name:
+                await self.user.edit(username=display_name)
+
+            if self.last_price != trade_price:
+                await self.change_presence(activity=disnake.Game(name=f"💰 {trade_price} {price_detail}"))
+                self.last_price = trade_price
+
+        except Exception as e:
+            print(e)
+            pass
+
+scheduler = Scheduler()
+scheduler.run(os.environ.get('DISCORD_BOT_TOKEN'))