ustock.py

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'))