mirror of
https://gitlab.com/simple-stock-bots/simple-stock-bot.git
synced 2026-06-03 21:00:26 +00:00
Compare commits
24 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 310f552bfc | |||
| 5aa639dbf6 | |||
| 8b054827d8 | |||
| 9476b5a280 | |||
| b8f2f6998a | |||
| 018613e896 | |||
| cffaaf2f9e | |||
| 8228812369 | |||
| 3ff582d095 | |||
| 0417436451 | |||
| d45ce7c250 | |||
| dcff9f3537 | |||
| 66ff51f021 | |||
| fbad8a20cf | |||
| 6229a5d4b6 | |||
| ef9cd7e85c | |||
| 2758c45432 | |||
| c717739b75 | |||
| 8a78ab5f55 | |||
| a54653bbc7 | |||
| 686cf889c1 | |||
| 662344bf18 | |||
| 87d9346f98 | |||
| 2fd5906f9e |
@@ -5,7 +5,8 @@
|
|||||||
// Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
|
// Or use a Dockerfile or Docker Compose file. More info: https://containers.dev/guide/dockerfile
|
||||||
// "image": "mcr.microsoft.com/devcontainers/python:1-3-bookworm",
|
// "image": "mcr.microsoft.com/devcontainers/python:1-3-bookworm",
|
||||||
"build": {
|
"build": {
|
||||||
"dockerfile": "Dockerfile"
|
"dockerfile": "Dockerfile",
|
||||||
|
"context": ".."
|
||||||
},
|
},
|
||||||
"features": {
|
"features": {
|
||||||
"ghcr.io/devcontainers-contrib/features/black:2": {},
|
"ghcr.io/devcontainers-contrib/features/black:2": {},
|
||||||
@@ -21,11 +22,12 @@
|
|||||||
"ms-python.vscode-pylance",
|
"ms-python.vscode-pylance",
|
||||||
"ms-python.isort",
|
"ms-python.isort",
|
||||||
"charliermarsh.ruff",
|
"charliermarsh.ruff",
|
||||||
"ms-toolsai.jupyter"
|
"ms-toolsai.jupyter",
|
||||||
|
"esbenp.prettier-vscode"
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"postCreateCommand": "pip3 install --user -r dev-reqs.txt"
|
"postCreateCommand": "pip3 install --user -r dev-reqs.txt && apt-get update && apt-get install -y nodejs npm --fix-missing && npm install"
|
||||||
// Features to add to the dev container. More info: https://containers.dev/features.
|
// Features to add to the dev container. More info: https://containers.dev/features.
|
||||||
// "features": {},
|
// "features": {},
|
||||||
// Use 'forwardPorts' to make a list of ports inside the container available locally.
|
// Use 'forwardPorts' to make a list of ports inside the container available locally.
|
||||||
|
|||||||
@@ -0,0 +1,17 @@
|
|||||||
|
# Required Environment Variables for Simple Stock Bot
|
||||||
|
|
||||||
|
# Telegram Bot Token (required for Telegram functionality)
|
||||||
|
# Get this from @BotFather on Telegram
|
||||||
|
TELEGRAM=your_telegram_bot_token_here
|
||||||
|
|
||||||
|
# Discord Bot Token (required for Discord functionality)
|
||||||
|
# Get this from Discord Developer Portal
|
||||||
|
DISCORD=your_discord_bot_token_here
|
||||||
|
|
||||||
|
# MarketData.app API Token (required for stock market data)
|
||||||
|
# Get this from https://www.marketdata.app/
|
||||||
|
MARKETDATA=your_marketdata_api_token_here
|
||||||
|
|
||||||
|
# Stripe Token (optional - for accepting donations)
|
||||||
|
# Get this from Stripe Dashboard
|
||||||
|
STRIPE=your_stripe_token_here
|
||||||
@@ -1,2 +1,3 @@
|
|||||||
__pycache__
|
__pycache__
|
||||||
.env
|
.env
|
||||||
|
node_modules
|
||||||
+11
-12
@@ -1,24 +1,23 @@
|
|||||||
stages:
|
stages:
|
||||||
- lint
|
- lint
|
||||||
- build
|
|
||||||
- build_site
|
- build_site
|
||||||
- deploy
|
- deploy
|
||||||
- deploy_site
|
|
||||||
|
|
||||||
black:
|
|
||||||
stage: lint
|
|
||||||
image: registry.gitlab.com/pipeline-components/black:latest
|
|
||||||
script:
|
|
||||||
- black --check --verbose -- .
|
|
||||||
|
|
||||||
ruff:
|
ruff:
|
||||||
stage: lint
|
stage: lint
|
||||||
# image: registry.gitlab.com/pipeline-components/ruff:latest
|
image: python:3.12-slim
|
||||||
image: python:3.11-slim
|
|
||||||
script:
|
script:
|
||||||
- pip3 install ruff
|
- pip3 install ruff==0.15.2
|
||||||
- ruff --output-format gitlab .
|
- ruff check . --output-format gitlab
|
||||||
|
- ruff format . --check
|
||||||
|
allow_failure: false
|
||||||
|
|
||||||
|
# prettier:
|
||||||
|
# stage: lint
|
||||||
|
# image: node:16-slim # Use Node.js image since prettier is a Node.js tool
|
||||||
|
# script:
|
||||||
|
# - npm install prettier
|
||||||
|
# - npx prettier --check . # Adjust the path as needed
|
||||||
|
|
||||||
include:
|
include:
|
||||||
- local: /site/.gitlab-ci.yml
|
- local: /site/.gitlab-ci.yml
|
||||||
|
|||||||
Vendored
+1
-1
@@ -3,7 +3,7 @@
|
|||||||
"editor.formatOnPaste": true,
|
"editor.formatOnPaste": true,
|
||||||
"editor.formatOnSaveMode": "modificationsIfAvailable",
|
"editor.formatOnSaveMode": "modificationsIfAvailable",
|
||||||
"[python]": {
|
"[python]": {
|
||||||
"editor.defaultFormatter": "ms-python.black-formatter",
|
"editor.defaultFormatter": "charliermarsh.ruff",
|
||||||
"editor.formatOnSave": true,
|
"editor.formatOnSave": true,
|
||||||
"editor.formatOnSaveMode": "file"
|
"editor.formatOnSaveMode": "file"
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ Enhance your group chats on Telegram and Discord with real-time stock and crypto
|
|||||||
## Documentation
|
## Documentation
|
||||||
|
|
||||||
Comprehensive documentation is available to help you understand the features and capabilities of Simple Stock Bots:
|
Comprehensive documentation is available to help you understand the features and capabilities of Simple Stock Bots:
|
||||||
|
|
||||||
- [Official Documentation](https://simplestockbot.com/)
|
- [Official Documentation](https://simplestockbot.com/)
|
||||||
- [Command Reference](https://simplestockbot.com/commands/)
|
- [Command Reference](https://simplestockbot.com/commands/)
|
||||||
|
|
||||||
@@ -17,6 +18,7 @@ Comprehensive documentation is available to help you understand the features and
|
|||||||
Simple Stock Bots is sustained entirely through donations, with an annual expense of about $420 for server and premium market data subscriptions from [marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=repo).
|
Simple Stock Bots is sustained entirely through donations, with an annual expense of about $420 for server and premium market data subscriptions from [marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=repo).
|
||||||
|
|
||||||
You can contribute by:
|
You can contribute by:
|
||||||
|
|
||||||
- Donating through [Buy Me A Coffee](https://www.buymeacoffee.com/Anson) (PayPal or Credit Card accepted).
|
- Donating through [Buy Me A Coffee](https://www.buymeacoffee.com/Anson) (PayPal or Credit Card accepted).
|
||||||
- Using the `/donate` [command](commands.md#donate-amount-in-usd) on Telegram.
|
- Using the `/donate` [command](commands.md#donate-amount-in-usd) on Telegram.
|
||||||
- [Getting in touch](contact.md) for any inquiries or suggestions.
|
- [Getting in touch](contact.md) for any inquiries or suggestions.
|
||||||
@@ -24,7 +26,7 @@ You can contribute by:
|
|||||||
### Additional Support:
|
### Additional Support:
|
||||||
|
|
||||||
- Follow on [Twitter](https://twitter.com/AnsonBiggs)
|
- Follow on [Twitter](https://twitter.com/AnsonBiggs)
|
||||||
- Contribute or star the project on [GitLab](https://gitlab.com/simple-stock-bots)
|
- Contribute or star the project on [GitLab](https://gitlab.com/simple-stock-bots/simple-stock-bot)
|
||||||
- Utilize referral links for self-hosting:
|
- Utilize referral links for self-hosting:
|
||||||
- [DigitalOcean](https://m.do.co/c/6b5df7ef55b6)
|
- [DigitalOcean](https://m.do.co/c/6b5df7ef55b6)
|
||||||
- [marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=web)
|
- [marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=web)
|
||||||
@@ -32,11 +34,13 @@ You can contribute by:
|
|||||||
## Hosting
|
## Hosting
|
||||||
|
|
||||||
Self-hosting instructions are provided for those interested in running the bot on their own servers:
|
Self-hosting instructions are provided for those interested in running the bot on their own servers:
|
||||||
|
|
||||||
- [Hosting Guide](https://simplestockbot.com/hosting/)
|
- [Hosting Guide](https://simplestockbot.com/hosting/)
|
||||||
|
|
||||||
## Contact
|
## Contact
|
||||||
|
|
||||||
Reach out for bug reports, feature requests, or other inquiries:
|
Reach out for bug reports, feature requests, or other inquiries:
|
||||||
|
|
||||||
- [Contact Page](https://simplestockbot.com/contact/)
|
- [Contact Page](https://simplestockbot.com/contact/)
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|||||||
+35
-19
@@ -1,8 +1,8 @@
|
|||||||
import datetime as dt
|
import datetime as dt
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
from typing import Dict
|
|
||||||
from collections import OrderedDict
|
from collections import OrderedDict
|
||||||
|
from typing import Dict
|
||||||
|
|
||||||
import humanize
|
import humanize
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
@@ -10,6 +10,7 @@ import pytz
|
|||||||
import requests as r
|
import requests as r
|
||||||
import schedule
|
import schedule
|
||||||
|
|
||||||
|
|
||||||
from common.Symbol import Stock
|
from common.Symbol import Stock
|
||||||
|
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
@@ -54,9 +55,12 @@ class MarketData:
|
|||||||
self.get_symbol_list()
|
self.get_symbol_list()
|
||||||
schedule.every().day.do(self.get_symbol_list)
|
schedule.every().day.do(self.get_symbol_list)
|
||||||
|
|
||||||
def get(self, endpoint, params: dict = {}, timeout=10) -> dict:
|
def get(self, endpoint, params=None, timeout=10, headers=None) -> dict:
|
||||||
url = "https://api.marketdata.app/v1/" + endpoint
|
url = "https://api.marketdata.app/v1/" + endpoint
|
||||||
|
|
||||||
|
if params is None:
|
||||||
|
params = {}
|
||||||
|
|
||||||
# set token param if it wasn't passed.
|
# set token param if it wasn't passed.
|
||||||
params["token"] = self.MARKETDATA_TOKEN
|
params["token"] = self.MARKETDATA_TOKEN
|
||||||
|
|
||||||
@@ -64,7 +68,13 @@ class MarketData:
|
|||||||
# monitored even if someone doesn't make it through an affiliate link.
|
# monitored even if someone doesn't make it through an affiliate link.
|
||||||
params["application"] = "simplestockbot"
|
params["application"] = "simplestockbot"
|
||||||
|
|
||||||
resp = r.get(url, params=params, timeout=timeout)
|
if headers is None:
|
||||||
|
headers = {}
|
||||||
|
headers = {"User-Agent": "Simple Stock Bot anson@ansonbiggs.com"} | headers
|
||||||
|
|
||||||
|
resp = r.get(url, params=params, timeout=timeout, headers=headers)
|
||||||
|
|
||||||
|
logging.debug(resp.headers.items())
|
||||||
|
|
||||||
# Make sure API returned a proper status code
|
# Make sure API returned a proper status code
|
||||||
try:
|
try:
|
||||||
@@ -95,7 +105,15 @@ class MarketData:
|
|||||||
return self.symbol_list.get(symbol.upper(), None)
|
return self.symbol_list.get(symbol.upper(), None)
|
||||||
|
|
||||||
def get_symbol_list(self):
|
def get_symbol_list(self):
|
||||||
sec_resp = r.get("https://www.sec.gov/files/company_tickers.json")
|
# Doesn't use `self.get()` since needs are much different
|
||||||
|
sec_resp = r.get(
|
||||||
|
"https://www.sec.gov/files/company_tickers.json",
|
||||||
|
headers={
|
||||||
|
"User-Agent": "Simple Stock Bot anson@ansonbiggs.com",
|
||||||
|
"Accept-Encoding": "gzip, deflate",
|
||||||
|
"Host": "www.sec.gov",
|
||||||
|
},
|
||||||
|
)
|
||||||
sec_resp.raise_for_status()
|
sec_resp.raise_for_status()
|
||||||
sec_data = sec_resp.json()
|
sec_data = sec_resp.json()
|
||||||
|
|
||||||
@@ -115,27 +133,21 @@ class MarketData:
|
|||||||
self.charts = {}
|
self.charts = {}
|
||||||
|
|
||||||
def status(self) -> str:
|
def status(self) -> str:
|
||||||
# TODO: At the moment this API is poorly documented, this function likely needs to be revisited later.
|
"""Check MarketData.app API status by making a test request."""
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
# Test the API with a simple request
|
||||||
status = r.get(
|
status = r.get(
|
||||||
"https://stats.uptimerobot.com/api/getMonitorList/6Kv3zIow0A",
|
"https://api.marketdata.app/v1/stocks/quotes/AAPL/",
|
||||||
timeout=5,
|
timeout=5,
|
||||||
)
|
)
|
||||||
status.raise_for_status()
|
status.raise_for_status()
|
||||||
|
return f"MarketData.app API is responding OK with status {status.status_code} in {status.elapsed.total_seconds():.2f} seconds."
|
||||||
except r.HTTPError:
|
except r.HTTPError:
|
||||||
return f"API returned an HTTP error code {status.status_code} in {status.elapsed.total_seconds()} seconds."
|
return f"MarketData.app API returned an HTTP error code {status.status_code} in {status.elapsed.total_seconds():.2f} seconds."
|
||||||
except r.Timeout:
|
except r.Timeout:
|
||||||
return "API timed out before it was able to give status. This is likely due to a surge in usage or a complete outage."
|
return "MarketData.app API timed out before it could respond. This is likely due to a surge in usage or a complete outage."
|
||||||
|
except Exception as e:
|
||||||
statusJSON = status.json()
|
return f"MarketData.app API check failed: {str(e)}"
|
||||||
|
|
||||||
if statusJSON["status"] == "ok":
|
|
||||||
return (
|
|
||||||
f"CoinGecko API responded that it was OK with a {status.status_code} in {status.elapsed.total_seconds()} seconds."
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
return f"MarketData.app is currently reporting the following status: {statusJSON['status']}"
|
|
||||||
|
|
||||||
def price_reply(self, symbol: Stock) -> str:
|
def price_reply(self, symbol: Stock) -> str:
|
||||||
"""Returns price movement of Stock for the last market day, or after hours.
|
"""Returns price movement of Stock for the last market day, or after hours.
|
||||||
@@ -212,7 +224,11 @@ class MarketData:
|
|||||||
|
|
||||||
if data := self.get(
|
if data := self.get(
|
||||||
f"stocks/candles/{resolution}/{symbol}",
|
f"stocks/candles/{resolution}/{symbol}",
|
||||||
params={"from": startTime.timestamp(), "to": now.timestamp(), "extended": True},
|
params={
|
||||||
|
"from": startTime.timestamp(),
|
||||||
|
"to": now.timestamp(),
|
||||||
|
"extended": True,
|
||||||
|
},
|
||||||
):
|
):
|
||||||
data.pop("s")
|
data.pop("s")
|
||||||
df = pd.DataFrame(data)
|
df = pd.DataFrame(data)
|
||||||
|
|||||||
@@ -26,6 +26,9 @@ class Symbol:
|
|||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
return self.id
|
return self.id
|
||||||
|
|
||||||
|
def __hash__(self):
|
||||||
|
return hash(self.id)
|
||||||
|
|
||||||
|
|
||||||
class Stock(Symbol):
|
class Stock(Symbol):
|
||||||
"""Stock Market Object. Gets data from MarketData"""
|
"""Stock Market Object. Gets data from MarketData"""
|
||||||
|
|||||||
+33
-11
@@ -5,8 +5,10 @@ import pandas as pd
|
|||||||
import requests as r
|
import requests as r
|
||||||
import schedule
|
import schedule
|
||||||
from markdownify import markdownify
|
from markdownify import markdownify
|
||||||
|
|
||||||
from common.Symbol import Coin
|
from common.Symbol import Coin
|
||||||
|
from common.utilities import rate_limited
|
||||||
|
|
||||||
|
import time
|
||||||
|
|
||||||
log = logging.getLogger(__name__)
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -24,10 +26,30 @@ class cg_Crypto:
|
|||||||
self.get_symbol_list()
|
self.get_symbol_list()
|
||||||
schedule.every().day.do(self.get_symbol_list)
|
schedule.every().day.do(self.get_symbol_list)
|
||||||
|
|
||||||
def get(self, endpoint, params: dict = {}, timeout=10) -> dict:
|
# Coingecko's rate limit is 30 requests per minute.
|
||||||
|
# Since there are two bots sharing the same IP, we allocate half of that limit to each bot.
|
||||||
|
# This results in a rate limit of 15 requests per minute for each bot.
|
||||||
|
# Given this, the rate limit effectively becomes 1 request every 4 seconds for each bot.
|
||||||
|
@rate_limited(0.25)
|
||||||
|
def get(self, endpoint, params: dict = {}, timeout=10, retry_count=0, max_retries=3) -> dict:
|
||||||
url = "https://api.coingecko.com/api/v3" + endpoint
|
url = "https://api.coingecko.com/api/v3" + endpoint
|
||||||
resp = r.get(url, params=params, timeout=timeout)
|
resp = r.get(url, params=params, timeout=timeout)
|
||||||
# Make sure API returned a proper status code
|
# Make sure API returned a proper status code
|
||||||
|
|
||||||
|
if resp.status_code == 429:
|
||||||
|
if retry_count >= max_retries:
|
||||||
|
log.error(f"CoinGecko 429 retry limit ({max_retries}) exceeded for endpoint: {endpoint}")
|
||||||
|
return {}
|
||||||
|
|
||||||
|
backoff_time = (2**retry_count) * 10 # Exponential backoff: 10s, 20s, 40s
|
||||||
|
log.warning(
|
||||||
|
f"CoinGecko returned 429 - Too Many Requests for endpoint: {endpoint}. Retry {retry_count + 1}/{max_retries} after {backoff_time}s."
|
||||||
|
)
|
||||||
|
time.sleep(backoff_time)
|
||||||
|
return self.get(
|
||||||
|
endpoint=endpoint, params=params, timeout=timeout, retry_count=retry_count + 1, max_retries=max_retries
|
||||||
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
resp.raise_for_status()
|
resp.raise_for_status()
|
||||||
except r.exceptions.HTTPError as e:
|
except r.exceptions.HTTPError as e:
|
||||||
@@ -201,14 +223,14 @@ class cg_Crypto:
|
|||||||
},
|
},
|
||||||
):
|
):
|
||||||
return f"""
|
return f"""
|
||||||
[{data['name']}]({data['links']['homepage'][0]}) Statistics:
|
[{data["name"]}]({data["links"]["homepage"][0]}) Statistics:
|
||||||
Market Cap: ${data['market_data']['market_cap'][self.vs_currency]:,}
|
Market Cap: ${data["market_data"]["market_cap"][self.vs_currency]:,}
|
||||||
Market Cap Ranking: {data.get('market_cap_rank',"Not Available")}
|
Market Cap Ranking: {data.get("market_cap_rank", "Not Available")}
|
||||||
CoinGecko Scores:
|
CoinGecko Scores:
|
||||||
Overall: {data.get('coingecko_score','Not Available')}
|
Overall: {data.get("coingecko_score", "Not Available")}
|
||||||
Development: {data.get('developer_score','Not Available')}
|
Development: {data.get("developer_score", "Not Available")}
|
||||||
Community: {data.get('community_score','Not Available')}
|
Community: {data.get("community_score", "Not Available")}
|
||||||
Public Interest: {data.get('public_interest_score','Not Available')}
|
Public Interest: {data.get("public_interest_score", "Not Available")}
|
||||||
"""
|
"""
|
||||||
else:
|
else:
|
||||||
return f"{symbol.symbol} returned an error."
|
return f"{symbol.symbol} returned an error."
|
||||||
@@ -248,7 +270,7 @@ class cg_Crypto:
|
|||||||
|
|
||||||
message = (
|
message = (
|
||||||
f"The current price of {coin.name} is $**{price:,}** and"
|
f"The current price of {coin.name} is $**{price:,}** and"
|
||||||
+ " its market cap is $**{cap:,.2f}** {self.vs_currency.upper()}"
|
+ f" its market cap is $**{cap:,.2f}** {self.vs_currency.upper()}"
|
||||||
)
|
)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
@@ -361,7 +383,7 @@ class cg_Crypto:
|
|||||||
p["usd_24h_change"] = 0
|
p["usd_24h_change"] = 0
|
||||||
|
|
||||||
replies.append(
|
replies.append(
|
||||||
f"{coin.name}: ${p.get('usd',0):,} and has moved {p.get('usd_24h_change',0.0):.2f}% in the past 24 hours."
|
f"{coin.name}: ${p.get('usd', 0):,} and has moved {p.get('usd_24h_change', 0.0):.2f}% in the past 24 hours."
|
||||||
)
|
)
|
||||||
|
|
||||||
return replies
|
return replies
|
||||||
|
|||||||
@@ -1,7 +1,8 @@
|
|||||||
cachetools==5.3.1
|
cachetools==7.0.1
|
||||||
humanize==4.8.0
|
humanize==4.15.0
|
||||||
markdownify==0.11.6
|
markdownify==1.2.2
|
||||||
mplfinance==0.12.10b0
|
mplfinance==0.12.10a7
|
||||||
pandas==2.1.1
|
pandas==3.0.1
|
||||||
requests==2.31.0
|
requests==2.32.5
|
||||||
schedule==1.2.1
|
rush==2021.4.0
|
||||||
|
schedule==1.2.2
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
"""Function that routes symbols to the correct API provider.
|
"""Function that routes symbols to the correct API provider."""
|
||||||
"""
|
|
||||||
|
|
||||||
import datetime
|
import datetime
|
||||||
import logging
|
import logging
|
||||||
|
|||||||
@@ -0,0 +1,31 @@
|
|||||||
|
import time
|
||||||
|
import logging
|
||||||
|
|
||||||
|
log = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def rate_limited(max_per_second):
|
||||||
|
"""
|
||||||
|
Decorator that ensures the wrapped function is called at most `max_per_second` times per second.
|
||||||
|
"""
|
||||||
|
min_interval = 1.0 / max_per_second
|
||||||
|
|
||||||
|
def decorate(func):
|
||||||
|
last_called = [0.0]
|
||||||
|
|
||||||
|
def rate_limited_function(*args, **kwargs):
|
||||||
|
elapsed = time.time() - last_called[0]
|
||||||
|
left_to_wait = min_interval - elapsed
|
||||||
|
|
||||||
|
if left_to_wait > 0:
|
||||||
|
log.info(f"Rate limit exceeded. Waiting for {left_to_wait:.2f} seconds.")
|
||||||
|
time.sleep(left_to_wait)
|
||||||
|
|
||||||
|
ret = func(*args, **kwargs)
|
||||||
|
last_called[0] = time.time()
|
||||||
|
|
||||||
|
return ret
|
||||||
|
|
||||||
|
return rate_limited_function
|
||||||
|
|
||||||
|
return decorate
|
||||||
+3
-3
@@ -1,6 +1,5 @@
|
|||||||
-r common/requirements.txt
|
-r common/requirements.txt
|
||||||
-r site/requirements.txt
|
-r site/requirements.txt
|
||||||
black==23.9.1
|
|
||||||
ipython==8.16.1
|
ipython==8.16.1
|
||||||
jupyter_client==8.4.0
|
jupyter_client==8.4.0
|
||||||
jupyter_core==5.4.0
|
jupyter_core==5.4.0
|
||||||
@@ -8,5 +7,6 @@ pylama==8.4.1
|
|||||||
mypy==1.5.1
|
mypy==1.5.1
|
||||||
types-cachetools==5.3.0.6
|
types-cachetools==5.3.0.6
|
||||||
types-pytz==2023.3.1.1
|
types-pytz==2023.3.1.1
|
||||||
ruff==0.0.292
|
ruff==0.15.2
|
||||||
isort==5.12.0
|
pytest==9.0.2
|
||||||
|
pytest-asyncio==0.25.3
|
||||||
|
|||||||
+2
-3
@@ -1,5 +1,4 @@
|
|||||||
"""Functions and Info specific to the discord Bot
|
"""Functions and Info specific to the discord Bot"""
|
||||||
"""
|
|
||||||
|
|
||||||
import re
|
import re
|
||||||
|
|
||||||
@@ -10,7 +9,7 @@ class D_info:
|
|||||||
license = re.sub(
|
license = re.sub(
|
||||||
r"\b\n",
|
r"\b\n",
|
||||||
" ",
|
" ",
|
||||||
r.get("https://gitlab.com/simple-stock-bots/simple-discord-stock-bot/-/raw/master/LICENSE").text,
|
r.get("https://gitlab.com/simple-stock-bots/simple-stock-bot/-/raw/master/LICENSE").text,
|
||||||
)
|
)
|
||||||
|
|
||||||
help_text = """
|
help_text = """
|
||||||
|
|||||||
+2
-2
@@ -1,4 +1,4 @@
|
|||||||
FROM python:3.11-buster AS builder
|
FROM python:3.12-bookworm AS builder
|
||||||
|
|
||||||
|
|
||||||
COPY discord/discord-reqs.txt .
|
COPY discord/discord-reqs.txt .
|
||||||
@@ -7,7 +7,7 @@ COPY common/requirements.txt .
|
|||||||
RUN pip install --user -r discord-reqs.txt
|
RUN pip install --user -r discord-reqs.txt
|
||||||
|
|
||||||
|
|
||||||
FROM python:3.11-slim
|
FROM python:3.12-slim
|
||||||
|
|
||||||
ENV MPLBACKEND=Agg
|
ENV MPLBACKEND=Agg
|
||||||
|
|
||||||
|
|||||||
+3
-5
@@ -42,7 +42,7 @@ async def status(ctx: commands):
|
|||||||
message = ""
|
message = ""
|
||||||
try:
|
try:
|
||||||
message = "Contact MisterBiggs#0465 if you need help.\n"
|
message = "Contact MisterBiggs#0465 if you need help.\n"
|
||||||
message += s.status(f"Bot recieved your message in: {bot.latency*10:.4f} seconds") + "\n"
|
message += s.status(f"Bot recieved your message in: {bot.latency * 10:.4f} seconds") + "\n"
|
||||||
|
|
||||||
except Exception as ex:
|
except Exception as ex:
|
||||||
logging.critical(ex)
|
logging.critical(ex)
|
||||||
@@ -217,9 +217,7 @@ async def handle_options(message, symbols):
|
|||||||
embed = nextcord.Embed(title=options_data["Option Symbol"], description=options_data["Underlying"], color=0x3498DB)
|
embed = nextcord.Embed(title=options_data["Option Symbol"], description=options_data["Underlying"], color=0x3498DB)
|
||||||
|
|
||||||
# Key details
|
# Key details
|
||||||
details = (
|
details = f"Expiration: {options_data['Expiration']}\nSide: {options_data['side']}\nStrike: {options_data['strike']}"
|
||||||
f"Expiration: {options_data['Expiration']}\n" f"Side: {options_data['side']}\n" f"Strike: {options_data['strike']}"
|
|
||||||
)
|
|
||||||
embed.add_field(name="Details", value=details, inline=False)
|
embed.add_field(name="Details", value=details, inline=False)
|
||||||
|
|
||||||
# Pricing info
|
# Pricing info
|
||||||
@@ -232,7 +230,7 @@ async def handle_options(message, symbols):
|
|||||||
embed.add_field(name="Pricing", value=pricing_info, inline=False)
|
embed.add_field(name="Pricing", value=pricing_info, inline=False)
|
||||||
|
|
||||||
# Volume and open interest
|
# Volume and open interest
|
||||||
volume_info = f"Open Interest: {options_data['Open Interest']}\n" f"Volume: {options_data['Volume']}"
|
volume_info = f"Open Interest: {options_data['Open Interest']}\nVolume: {options_data['Volume']}"
|
||||||
embed.add_field(name="Activity", value=volume_info, inline=False)
|
embed.add_field(name="Activity", value=volume_info, inline=False)
|
||||||
|
|
||||||
# Greeks
|
# Greeks
|
||||||
|
|||||||
+1
-2
@@ -1,10 +1,9 @@
|
|||||||
version: '3'
|
version: "3"
|
||||||
services:
|
services:
|
||||||
telegram:
|
telegram:
|
||||||
build:
|
build:
|
||||||
context: .
|
context: .
|
||||||
dockerfile: telegram/Dockerfile
|
dockerfile: telegram/Dockerfile
|
||||||
image: registry.gitlab.com/simple-stock-bots/simple-telegram-stock-bot
|
|
||||||
env_file: .env
|
env_file: .env
|
||||||
discord:
|
discord:
|
||||||
build:
|
build:
|
||||||
|
|||||||
Generated
+27
@@ -0,0 +1,27 @@
|
|||||||
|
{
|
||||||
|
"name": "simple-stock-bot",
|
||||||
|
"lockfileVersion": 3,
|
||||||
|
"requires": true,
|
||||||
|
"packages": {
|
||||||
|
"": {
|
||||||
|
"devDependencies": {
|
||||||
|
"prettier": "3.0.3"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"node_modules/prettier": {
|
||||||
|
"version": "3.0.3",
|
||||||
|
"resolved": "https://registry.npmjs.org/prettier/-/prettier-3.0.3.tgz",
|
||||||
|
"integrity": "sha512-L/4pUDMxcNa8R/EthV08Zt42WBO4h1rarVtK0K+QJG0X187OLo7l699jWw0GKuwzkPQ//jMFA/8Xm6Fh3J/DAg==",
|
||||||
|
"dev": true,
|
||||||
|
"bin": {
|
||||||
|
"prettier": "bin/prettier.cjs"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=14"
|
||||||
|
},
|
||||||
|
"funding": {
|
||||||
|
"url": "https://github.com/prettier/prettier?sponsor=1"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
{
|
||||||
|
"devDependencies": {
|
||||||
|
"prettier": "3.0.3"
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,11 +1,2 @@
|
|||||||
[tool.black]
|
|
||||||
line-length = 130
|
|
||||||
|
|
||||||
[tool.flake8]
|
|
||||||
max-line-length = 130
|
|
||||||
|
|
||||||
[tool.pycodestyle]
|
|
||||||
max_line_length = 130
|
|
||||||
|
|
||||||
[tool.ruff]
|
[tool.ruff]
|
||||||
line-length = 130
|
line-length = 130
|
||||||
+5
-5
@@ -1,7 +1,7 @@
|
|||||||
image: python:3.11
|
image: python:3.11
|
||||||
|
|
||||||
build_site:
|
build_mkdocs:
|
||||||
stage: build
|
stage: build_site
|
||||||
script:
|
script:
|
||||||
- cd ./site
|
- cd ./site
|
||||||
- pip install -r requirements.txt
|
- pip install -r requirements.txt
|
||||||
@@ -10,14 +10,14 @@ build_site:
|
|||||||
paths:
|
paths:
|
||||||
- public
|
- public
|
||||||
|
|
||||||
deploy_site:
|
pages:
|
||||||
stage: deploy
|
stage: deploy
|
||||||
script:
|
script:
|
||||||
- echo "Publishing site..."
|
- echo "Publishing site..."
|
||||||
dependencies:
|
dependencies:
|
||||||
- build_site
|
- build_mkdocs
|
||||||
artifacts:
|
artifacts:
|
||||||
paths:
|
paths:
|
||||||
- public
|
- public
|
||||||
rules:
|
rules:
|
||||||
- if: '$CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH'
|
- if: "$CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH"
|
||||||
|
|||||||
@@ -1,2 +1 @@
|
|||||||
# Blog
|
# Blog
|
||||||
|
|
||||||
|
|||||||
@@ -25,7 +25,10 @@ Simple Stock Bot is a chatbot designed to enrich your financial discussions on T
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
<!-- more -->
|
||||||
|
|
||||||
With Simple Stock Bot, you can:
|
With Simple Stock Bot, you can:
|
||||||
|
|
||||||
- **Fetch Real-time Quotes**: Obtain the latest stock and cryptocurrency prices instantly within your group chat.
|
- **Fetch Real-time Quotes**: Obtain the latest stock and cryptocurrency prices instantly within your group chat.
|
||||||
- **View Charts**: Delve into intraday movements or assess a month's trading data right inside your group chat.
|
- **View Charts**: Delve into intraday movements or assess a month's trading data right inside your group chat.
|
||||||
- **Stay Updated on Market Trends**: Discover what's buzzing in the stock and crypto spheres.
|
- **Stay Updated on Market Trends**: Discover what's buzzing in the stock and crypto spheres.
|
||||||
@@ -36,13 +39,12 @@ With Simple Stock Bot, you can:
|
|||||||
|
|
||||||
[:fontawesome-brands-discord: Discord](https://discordapp.com/api/oauth2/authorize?client_id=532045200823025666&permissions=36507338752&scope=bot){ .md-button }
|
[:fontawesome-brands-discord: Discord](https://discordapp.com/api/oauth2/authorize?client_id=532045200823025666&permissions=36507338752&scope=bot){ .md-button }
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
## Using the Bot
|
## Using the Bot
|
||||||
|
|
||||||
Embarking on your journey with Simple Stock Bot is effortless. Just add the bot to your Telegram or Discord group, and you're all set to dive into financial discussions right from your chat!
|
Embarking on your journey with Simple Stock Bot is effortless. Just add the bot to your Telegram or Discord group, and you're all set to dive into financial discussions right from your chat!
|
||||||
|
|
||||||
Here are some simple commands to get you started:
|
Here are some simple commands to get you started:
|
||||||
|
|
||||||
- `/intra` to view intraday charts.
|
- `/intra` to view intraday charts.
|
||||||
- `/chart` for a glimpse of the last month's trading data.
|
- `/chart` for a glimpse of the last month's trading data.
|
||||||
- `/trending` to catch up on the latest trending stocks and cryptocurrencies.
|
- `/trending` to catch up on the latest trending stocks and cryptocurrencies.
|
||||||
@@ -51,7 +53,7 @@ Here are some simple commands to get you started:
|
|||||||
|
|
||||||
Simple Stock Bot is a community-supported project, thriving on the contributions from its users. It's sustained entirely through donations to cover server costs and premium market data subscriptions, ensuring it remains free for everyone.
|
Simple Stock Bot is a community-supported project, thriving on the contributions from its users. It's sustained entirely through donations to cover server costs and premium market data subscriptions, ensuring it remains free for everyone.
|
||||||
|
|
||||||
Feeling generous? You can support the project by [donating](https://simplestockbot.com/donate/), following on [Twitter](https://twitter.com/AnsonBiggs), or contributing on [GitLab](https://gitlab.com/simple-stock-bot).
|
Feeling generous? You can support the project by [donating](https://simplestockbot.com/donate/), following on [Twitter](https://twitter.com/AnsonBiggs), or contributing on [GitLab](https://gitlab.com/simple-stock-bots/simple-stock-bot).
|
||||||
|
|
||||||
## Dive Deeper
|
## Dive Deeper
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,40 @@
|
|||||||
|
---
|
||||||
|
title: "v2023.1 Release! New Stock Market Data Provider and More!"
|
||||||
|
date: 2023-10-16
|
||||||
|
tags: [Simple Stock Bot, Telegram, Discord, MarketDataApp]
|
||||||
|
authors: [Anson]
|
||||||
|
description: >
|
||||||
|
Discover the latest updates for Simple Stock Bot, including our new integration with MarketData.app for enhanced real-time stock market insights.
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🌐 General Updates:
|
||||||
|
|
||||||
|
- **New Home**: We've transitioned to our fresh and updated website at [simplestockbot.com](https://simplestockbot.com/).
|
||||||
|
- **New Data Provider**: We're excited to announce [MarketData.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=web) as our new provider for real-time stock market data, ensuring timely and accurate insights for our users.
|
||||||
|
|
||||||
|
<!-- more -->
|
||||||
|
|
||||||
|
## 📈 MarketData.app Integration Enhancements:
|
||||||
|
|
||||||
|
- We've introduced **additional options** to our MarketData.app integration, ensuring even more precise and varied financial data.
|
||||||
|
- **Afterhours Data Fix**: Addressed an issue where after hours stock market data caused errors.
|
||||||
|
- **Trending Symbol Accuracy**: Fixed a bug that led to the display of invalid symbols in the `/trending` command.
|
||||||
|
|
||||||
|
## 🤖 Bot Improvements & Fixes:
|
||||||
|
|
||||||
|
- **Unified Repository**: To streamline our development and deployment, we've merged the Discord and Telegram bots into a single monorepo. Check it out on [GitLab](https://gitlab.com/simple-stock-bots/simple-stock-bot). Contributions welcome!
|
||||||
|
- **Inline Functionality Restoration**: Fixed the telegram bot's inline functionality.
|
||||||
|
- **Python Telegram Bot Update**: Migrated to the latest version of `Python Telegram Bot` for superior performance and more features.
|
||||||
|
- **Rate Limiting Addition**: Implemented rate limiting to ensure optimal performance during peak usage times.
|
||||||
|
- **SO MUCH MORE**: [Move to Marketdata.app GitLab Issues](https://gitlab.com/simple-stock-bots/simple-stock-bot/-/milestones/3)
|
||||||
|
|
||||||
|
## 📝 Documentation & Repository Overhauls:
|
||||||
|
|
||||||
|
- **Centralized Documentation**: For convenience and improved maintenance, we've shifted our documentation into the monorepo.
|
||||||
|
- **Documentation Refinement**: Updated our documentation to reflect the latest homepage details.
|
||||||
|
|
||||||
|
**Special Mention**: Immense gratitude to our dedicated community for their continual feedback and unwavering support. Dive deeper into the financial realm with Simple Stock Bot!
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
📥 For any concerns, queries, or feedback, don't hesitate to [Contact Us](../../contact.md).
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
---
|
||||||
|
title: "Simple Bot End-of-Life"
|
||||||
|
date: 2024-05-12
|
||||||
|
tags: [Simple Stock Bot, Introduction, Telegram, Discord, Financial Insights]
|
||||||
|
authors: [Anson]
|
||||||
|
description: >
|
||||||
|
Simple Stock Bot is being sunset.
|
||||||
|
---
|
||||||
|
|
||||||
|
## What is Simple Stock Bot?
|
||||||
|
|
||||||
|
For full details see my full blog post: [https://notes.ansonbiggs.com/simple-stock-bot-end-of-life/](https://notes.ansonbiggs.com/simple-stock-bot-end-of-life/)
|
||||||
|
|
||||||
|
I am shutting down Simple Stock Bot, a popular Telegram and Discord bot that provided live stock and cryptocurrency market data. The bot was created in 2018 after my group chat lost access to the Google Allo, which had integrated market data. The bot grew to 15,000 monthly active users, but maintaining the service has become challenging due to increasing data costs and changes to the Discord API. I really appreciate the community and everyone that has donated along the way, this has been a seriously great ride. The bot will soon no longer function on Telegram and Discord.
|
||||||
@@ -32,7 +32,6 @@ The Simple Stock Bot looks at every message it can see and tries to detect stock
|
|||||||
|
|
||||||
This command allows you to query real-time data for stock options. By simply inputting the stock symbol, strike price, month, and specifying either a call or a put, you can get the latest options data right at your fingertips. For example, `AAPL $220 December call` will provide the current data for Apple's call option with a $220 strike price expiring in December.
|
This command allows you to query real-time data for stock options. By simply inputting the stock symbol, strike price, month, and specifying either a call or a put, you can get the latest options data right at your fingertips. For example, `AAPL $220 December call` will provide the current data for Apple's call option with a $220 strike price expiring in December.
|
||||||
|
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
## `/donate [Amount in USD]` :fontawesome-brands-telegram-plane:
|
## `/donate [Amount in USD]` :fontawesome-brands-telegram-plane:
|
||||||
@@ -104,7 +103,9 @@ Gets the latest trending stocks and their change in price
|
|||||||
━━━━━━━━━━━━━━━━━━━━━━━━━━
|
━━━━━━━━━━━━━━━━━━━━━━━━━━
|
||||||
$GME: Gamestop Corporation - Class A, 5.06%
|
$GME: Gamestop Corporation - Class A, 5.06%
|
||||||
$MMAT: Meta Materials Inc, 4.77%
|
$MMAT: Meta Materials Inc, 4.77%
|
||||||
$$BTC: Bitcoin, -0.83%
|
|
||||||
|
$$
|
||||||
|
BTC: Bitcoin, -0.83%
|
||||||
$$ETH: Ethereum, 6.06%
|
$$ETH: Ethereum, 6.06%
|
||||||
$$DOGE: Binance-Peg Dogecoin, 0.14%
|
$$DOGE: Binance-Peg Dogecoin, 0.14%
|
||||||
|
|
||||||
@@ -184,3 +185,4 @@ full list of stock symbols and return the price of the ticker. Then once you sel
|
|||||||
want the bot will send a message as you in that chat with the latest stock price.
|
want the bot will send a message as you in that chat with the latest stock price.
|
||||||
The bot also looks at every message in any chat it is in for stock symbols.Symbols start with a
|
The bot also looks at every message in any chat it is in for stock symbols.Symbols start with a
|
||||||
`$` followed by the stock symbol. For example:$tsla would return price information for Tesla Motors.
|
`$` followed by the stock symbol. For example:$tsla would return price information for Tesla Motors.
|
||||||
|
$$
|
||||||
|
|||||||
@@ -5,4 +5,4 @@ Your feedback and experience with Simple Stock Bot is invaluable. Whether you've
|
|||||||
- **Telegram**: Drop me a message on [Telegram](https://t.me/MisterBiggs).
|
- **Telegram**: Drop me a message on [Telegram](https://t.me/MisterBiggs).
|
||||||
- **Discord**: Feel free to reach out on Discord at `MisterBiggs#0465`.
|
- **Discord**: Feel free to reach out on Discord at `MisterBiggs#0465`.
|
||||||
- **Twitter**: Send me a tweet on [Twitter](https://twitter.com/AnsonBiggs).
|
- **Twitter**: Send me a tweet on [Twitter](https://twitter.com/AnsonBiggs).
|
||||||
- **GitLab**: For technical issues or feature requests, feel free to open an issue on our [GitLab Repository](https://gitlab.com/simple-stock-bots/simple-telegram-stock-bot).
|
- **GitLab**: For technical issues or feature requests, feel free to open an issue on our [GitLab Repository](https://gitlab.com/simple-stock-bots/simple-stock-bot).
|
||||||
|
|||||||
+5
-2
@@ -17,17 +17,19 @@ Ensure the following are installed or obtained before proceeding:
|
|||||||
- **[marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=web)**: Sign up to get an API key. A free tier is available and should suffice for private groups. More details [here](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=repo).
|
- **[marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=web)**: Sign up to get an API key. A free tier is available and should suffice for private groups. More details [here](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=repo).
|
||||||
|
|
||||||
!!! tip
|
!!! tip
|
||||||
The bot will still operate without a [marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=repo) key but will revert to using only cryptocurrency data.
|
The bot will still operate without a [marketdata.app](https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=repo) key but will revert to using only cryptocurrency data.
|
||||||
|
|
||||||
!!! note
|
!!! note
|
||||||
To enable donation acceptance, obtain a Stripe API key and provide a `STRIPE` key to your bot. [https://stripe.com/]()
|
To enable donation acceptance, obtain a Stripe API key and provide a `STRIPE` key to your bot. [https://stripe.com/]()
|
||||||
|
|
||||||
## Setup Instructions
|
## Setup Instructions
|
||||||
|
|
||||||
1. **Download/Clone the Repository**:
|
1. **Download/Clone the Repository**:
|
||||||
|
|
||||||
- Download or clone this repository to your local machine.
|
- Download or clone this repository to your local machine.
|
||||||
|
|
||||||
2. **Configure Environment Variables**:
|
2. **Configure Environment Variables**:
|
||||||
|
|
||||||
- Navigate to the project directory and locate the `.env` file.
|
- Navigate to the project directory and locate the `.env` file.
|
||||||
- Fill in the `.env` file with your obtained API keys:
|
- Fill in the `.env` file with your obtained API keys:
|
||||||
|
|
||||||
@@ -41,6 +43,7 @@ Ensure the following are installed or obtained before proceeding:
|
|||||||
Alternatively, pass the variables using Docker Compose environment variables or command-line arguments.
|
Alternatively, pass the variables using Docker Compose environment variables or command-line arguments.
|
||||||
|
|
||||||
3. **Build and Run the Bot:**
|
3. **Build and Run the Bot:**
|
||||||
|
|
||||||
- Open a terminal in the project directory.
|
- Open a terminal in the project directory.
|
||||||
- Build and run both bots using Docker Compose:
|
- Build and run both bots using Docker Compose:
|
||||||
|
|
||||||
|
|||||||
@@ -9,25 +9,31 @@ Discover a new level of financial discussion with Simple Stock Bot, your go-to s
|
|||||||
## [Commands](commands.md)
|
## [Commands](commands.md)
|
||||||
|
|
||||||
### Real-Time Stock and Cryptocurrency Updates
|
### Real-Time Stock and Cryptocurrency Updates
|
||||||
|
|
||||||
- Get live price updates of stocks and cryptocurrencies.
|
- Get live price updates of stocks and cryptocurrencies.
|
||||||
- Symbol Detection: Use `$` for stocks and `$$` for cryptocurrencies, e.g., `$tsla` for Tesla and `$$btc` for Bitcoin.
|
- Symbol Detection: Use `$` for stocks and `$$` for cryptocurrencies, e.g., `$tsla` for Tesla and `$$btc` for Bitcoin.
|
||||||
|
|
||||||
### Intraday and Historical Data Visualization
|
### Intraday and Historical Data Visualization
|
||||||
|
|
||||||
- `/intra`: View a stock's movement since the last market open or a cryptocurrency's 24-hour chart.
|
- `/intra`: View a stock's movement since the last market open or a cryptocurrency's 24-hour chart.
|
||||||
- `/chart`: Provides a chart of the last month's trading data up till the day before.
|
- `/chart`: Provides a chart of the last month's trading data up till the day before.
|
||||||
|
|
||||||
### Trending Assets Awareness
|
### Trending Assets Awareness
|
||||||
|
|
||||||
- `/trending`: Discover the latest trending stocks and cryptocurrencies and their price changes.
|
- `/trending`: Discover the latest trending stocks and cryptocurrencies and their price changes.
|
||||||
|
|
||||||
### In-App Assistance and Support
|
### In-App Assistance and Support
|
||||||
|
|
||||||
- `/help`: Provides a quick summary of available commands for easy reference.
|
- `/help`: Provides a quick summary of available commands for easy reference.
|
||||||
- `/donate`: Support the maintenance of the bot to ensure continued free access.
|
- `/donate`: Support the maintenance of the bot to ensure continued free access.
|
||||||
|
|
||||||
### User-Friendly Interaction
|
### User-Friendly Interaction
|
||||||
|
|
||||||
- Easy symbol and command usage for a smooth user experience.
|
- Easy symbol and command usage for a smooth user experience.
|
||||||
- Engage with a community of financial enthusiasts and share insights effortlessly.
|
- Engage with a community of financial enthusiasts and share insights effortlessly.
|
||||||
|
|
||||||
## [Contact](contact.md)
|
## [Contact](contact.md)
|
||||||
|
|
||||||
Need assistance or have inquiries? Feel free to reach out and I'll be happy to help.
|
Need assistance or have inquiries? Feel free to reach out and I'll be happy to help.
|
||||||
|
|
||||||
## Powered by MarketData.app
|
## Powered by MarketData.app
|
||||||
|
|||||||
+5
-11
@@ -1,24 +1,23 @@
|
|||||||
site_name: Simple Stock Bot
|
site_name: Simple Stock Bot
|
||||||
site_url: https://simplestockbot.com
|
site_url: https://simplestockbot.com
|
||||||
edit_uri: /simple-stock-bots/site/tree/master/site
|
repo_url: https://gitlab.com/simple-stock-bots/simple-stock-bot
|
||||||
repo_url: https://gitlab.com/simple-stock-bots/simple-telegram-stock-bot
|
|
||||||
repo_name: Simple Stock Bots
|
repo_name: Simple Stock Bots
|
||||||
|
|
||||||
site_description: Simple Discord and Telegram chat bots for Stock Market Data.
|
site_description: Simple Discord and Telegram chat bots for Stock Market Data.
|
||||||
site_author: Anson Biggs
|
site_author: Anson Biggs
|
||||||
copyright: <a href="https://ansonbiggs.com">Anson Biggs</a> 2018 </br> Available Under an <a href="/LICENSE/">MIT LICENSE</a> </br>
|
copyright: <a href="https://ansonbiggs.com">Anson Biggs</a> 2018 </br> Available Under an <a href="/LICENSE/">MIT LICENSE</a> </br>
|
||||||
|
|
||||||
|
|
||||||
nav:
|
nav:
|
||||||
- Home: index.md
|
- Home: index.md
|
||||||
- Commands: commands.md
|
- Commands: commands.md
|
||||||
- Self-Host: host.md
|
- Self-Host: host.md
|
||||||
- Donate: donate.md
|
# - Donate: donate.md
|
||||||
- Contact: contact.md
|
- Contact: contact.md
|
||||||
- Blog: blog/index.md
|
- Blog: blog/index.md
|
||||||
|
|
||||||
theme:
|
theme:
|
||||||
name: material
|
name: material
|
||||||
|
custom_dir: overrides
|
||||||
language: en
|
language: en
|
||||||
features:
|
features:
|
||||||
- navigation.instant
|
- navigation.instant
|
||||||
@@ -56,24 +55,19 @@ extra:
|
|||||||
link: https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=web
|
link: https://dashboard.marketdata.app/marketdata/aff/go/misterbiggs?keyword=web
|
||||||
name: MarketData.app for stock data
|
name: MarketData.app for stock data
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
extra_css:
|
extra_css:
|
||||||
- stylesheets/phone.css
|
- stylesheets/phone.css
|
||||||
|
|
||||||
|
|
||||||
plugins:
|
plugins:
|
||||||
- search
|
- search
|
||||||
- social
|
- social
|
||||||
- blog
|
- blog
|
||||||
|
|
||||||
|
|
||||||
markdown_extensions:
|
markdown_extensions:
|
||||||
- attr_list
|
- attr_list
|
||||||
- pymdownx.emoji:
|
- pymdownx.emoji:
|
||||||
emoji_index: !!python/name:materialx.emoji.twemoji
|
emoji_index: !!python/name:material.extensions.emoji.twemoji
|
||||||
emoji_generator: !!python/name:materialx.emoji.to_svg
|
emoji_generator: !!python/name:material.extensions.emoji.to_svg
|
||||||
- admonition
|
- admonition
|
||||||
- pymdownx.details
|
- pymdownx.details
|
||||||
- pymdownx.superfences
|
- pymdownx.superfences
|
||||||
|
|||||||
@@ -0,0 +1,17 @@
|
|||||||
|
{% extends "base.html" %}
|
||||||
|
|
||||||
|
<!-- Announcement bar -->
|
||||||
|
{% block announce %}
|
||||||
|
|
||||||
|
<div class="admonition danger" style="min-height: 20vh;">
|
||||||
|
<p class="admonition-title">This project is now archived!</p>
|
||||||
|
<p>
|
||||||
|
Please
|
||||||
|
<a href="/blog/2024/05/12/simple-bot-end-of-life/"
|
||||||
|
>click here for more information.</a
|
||||||
|
>
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
<div class="" admoniton></div>
|
||||||
|
|
||||||
|
{% endblock %}
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
mkdocs-material==9.4.4
|
mkdocs-material==9.5.22
|
||||||
|
mkdocs-material-extensions==1.3.1
|
||||||
|
|
||||||
# Required for Social Cards
|
# Required for Social Cards
|
||||||
Pillow==10.0.1
|
Pillow==10.0.1
|
||||||
|
|||||||
+2
-2
@@ -1,4 +1,4 @@
|
|||||||
FROM python:3.11-buster AS builder
|
FROM python:3.12-bookworm AS builder
|
||||||
|
|
||||||
|
|
||||||
COPY telegram/telegram-reqs.txt .
|
COPY telegram/telegram-reqs.txt .
|
||||||
@@ -7,7 +7,7 @@ COPY common/requirements.txt .
|
|||||||
RUN pip install --user -r telegram-reqs.txt
|
RUN pip install --user -r telegram-reqs.txt
|
||||||
|
|
||||||
|
|
||||||
FROM python:3.11-slim
|
FROM python:3.12-slim
|
||||||
|
|
||||||
ENV MPLBACKEND=Agg
|
ENV MPLBACKEND=Agg
|
||||||
|
|
||||||
|
|||||||
+2
-3
@@ -1,5 +1,4 @@
|
|||||||
"""Functions and Info specific to the Telegram Bot
|
"""Functions and Info specific to the Telegram Bot"""
|
||||||
"""
|
|
||||||
|
|
||||||
import re
|
import re
|
||||||
|
|
||||||
@@ -10,7 +9,7 @@ class T_info:
|
|||||||
license = re.sub(
|
license = re.sub(
|
||||||
r"\b\n",
|
r"\b\n",
|
||||||
" ",
|
" ",
|
||||||
r.get("https://gitlab.com/simple-stock-bots/simple-telegram-stock-bot/-/raw/master/LICENSE").text,
|
r.get("https://gitlab.com/simple-stock-bots/simple-stock-bot/-/raw/master/LICENSE").text,
|
||||||
)
|
)
|
||||||
|
|
||||||
help_text = """
|
help_text = """
|
||||||
|
|||||||
+9
-15
@@ -11,27 +11,21 @@ import traceback
|
|||||||
from uuid import uuid4
|
from uuid import uuid4
|
||||||
|
|
||||||
import mplfinance as mpf
|
import mplfinance as mpf
|
||||||
import telegram
|
from T_info import T_info
|
||||||
from telegram import (
|
|
||||||
InlineQueryResultArticle,
|
|
||||||
InputTextMessageContent,
|
|
||||||
LabeledPrice,
|
|
||||||
Update,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
import telegram
|
||||||
|
from common.symbol_router import Router
|
||||||
|
from telegram import InlineQueryResultArticle, InputTextMessageContent, LabeledPrice, Update
|
||||||
from telegram.ext import (
|
from telegram.ext import (
|
||||||
Application,
|
Application,
|
||||||
CommandHandler,
|
CommandHandler,
|
||||||
InlineQueryHandler,
|
|
||||||
PreCheckoutQueryHandler,
|
|
||||||
MessageHandler,
|
|
||||||
filters,
|
|
||||||
ContextTypes,
|
ContextTypes,
|
||||||
|
InlineQueryHandler,
|
||||||
|
MessageHandler,
|
||||||
|
PreCheckoutQueryHandler,
|
||||||
|
filters,
|
||||||
)
|
)
|
||||||
|
|
||||||
from common.symbol_router import Router
|
|
||||||
from T_info import T_info
|
|
||||||
|
|
||||||
# Enable logging
|
# Enable logging
|
||||||
logging.basicConfig(format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO)
|
logging.basicConfig(format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO)
|
||||||
|
|
||||||
@@ -238,7 +232,7 @@ def generate_options_reply(options_data: dict):
|
|||||||
message_text += pricing_info
|
message_text += pricing_info
|
||||||
|
|
||||||
# Volume and open interest
|
# Volume and open interest
|
||||||
volume_info = f"*Open Interest:* `{options_data['Open Interest']}`\n" f"*Volume:* `{options_data['Volume']}`\n\n"
|
volume_info = f"*Open Interest:* `{options_data['Open Interest']}`\n*Volume:* `{options_data['Volume']}`\n\n"
|
||||||
message_text += volume_info
|
message_text += volume_info
|
||||||
|
|
||||||
# Greeks
|
# Greeks
|
||||||
|
|||||||
@@ -1,2 +1,2 @@
|
|||||||
python-telegram-bot==20.6
|
python-telegram-bot==21.11.1
|
||||||
-r requirements.txt
|
-r requirements.txt
|
||||||
@@ -9,9 +9,7 @@ $tsla
|
|||||||
/chart $tsla
|
/chart $tsla
|
||||||
/chart $$btc
|
/chart $$btc
|
||||||
/help
|
/help
|
||||||
/trending""".split(
|
/trending""".split("\n")
|
||||||
"\n"
|
|
||||||
)
|
|
||||||
|
|
||||||
print("press enter to start")
|
print("press enter to start")
|
||||||
keyboard.wait("enter")
|
keyboard.wait("enter")
|
||||||
|
|||||||
@@ -0,0 +1,15 @@
|
|||||||
|
"""Pytest configuration and fixtures."""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
# Add the project root to Python path so we can import modules
|
||||||
|
project_root = Path(__file__).parent.parent
|
||||||
|
sys.path.insert(0, str(project_root))
|
||||||
|
|
||||||
|
# Set environment variables for testing
|
||||||
|
os.environ.setdefault("TELEGRAM", "test_token")
|
||||||
|
os.environ.setdefault("DISCORD", "test_token")
|
||||||
|
os.environ.setdefault("MARKETDATA", "test_token")
|
||||||
|
os.environ.setdefault("STRIPE", "test_token")
|
||||||
@@ -0,0 +1,132 @@
|
|||||||
|
"""Test the rate_limited decorator functionality."""
|
||||||
|
|
||||||
|
import time
|
||||||
|
from unittest.mock import patch
|
||||||
|
from common.utilities import rate_limited
|
||||||
|
|
||||||
|
|
||||||
|
class TestRateLimited:
|
||||||
|
"""Test rate_limited decorator functionality."""
|
||||||
|
|
||||||
|
def test_rate_limited_decorator(self):
|
||||||
|
"""Test that rate limiting works correctly."""
|
||||||
|
call_times = []
|
||||||
|
|
||||||
|
@rate_limited(2.0) # 2 calls per second max
|
||||||
|
def test_function():
|
||||||
|
call_times.append(time.time())
|
||||||
|
return len(call_times)
|
||||||
|
|
||||||
|
# First call should execute immediately
|
||||||
|
result1 = test_function()
|
||||||
|
assert result1 == 1
|
||||||
|
assert len(call_times) == 1
|
||||||
|
|
||||||
|
# Second call should be delayed
|
||||||
|
start_time = time.time()
|
||||||
|
result2 = test_function()
|
||||||
|
end_time = time.time()
|
||||||
|
|
||||||
|
assert result2 == 2
|
||||||
|
assert len(call_times) == 2
|
||||||
|
|
||||||
|
# Should have waited at least close to 0.5 seconds (1/2 calls per second)
|
||||||
|
elapsed = end_time - start_time
|
||||||
|
assert elapsed >= 0.4 # Allow some tolerance for timing
|
||||||
|
|
||||||
|
def test_rate_limited_with_args(self):
|
||||||
|
"""Test rate limited function with arguments."""
|
||||||
|
|
||||||
|
@rate_limited(1.0) # 1 call per second
|
||||||
|
def test_function(x, y=1):
|
||||||
|
return x + y
|
||||||
|
|
||||||
|
# Test that arguments are passed correctly
|
||||||
|
result1 = test_function(5, y=10)
|
||||||
|
assert result1 == 15
|
||||||
|
|
||||||
|
# Test positional and keyword args
|
||||||
|
start_time = time.time()
|
||||||
|
result2 = test_function(3, 7)
|
||||||
|
end_time = time.time()
|
||||||
|
|
||||||
|
assert result2 == 10
|
||||||
|
# Should have waited about 1 second
|
||||||
|
elapsed = end_time - start_time
|
||||||
|
assert elapsed >= 0.9
|
||||||
|
|
||||||
|
def test_rate_limited_logging(self):
|
||||||
|
"""Test that rate limiting logs waiting messages."""
|
||||||
|
|
||||||
|
@rate_limited(10.0) # 10 calls per second
|
||||||
|
def test_function():
|
||||||
|
return "called"
|
||||||
|
|
||||||
|
with patch("common.utilities.log.info") as mock_log:
|
||||||
|
# First call should not log
|
||||||
|
test_function()
|
||||||
|
mock_log.assert_not_called()
|
||||||
|
|
||||||
|
# Second call should log the wait
|
||||||
|
test_function()
|
||||||
|
mock_log.assert_called_once()
|
||||||
|
|
||||||
|
# Check that the log message contains expected content
|
||||||
|
log_call_args = mock_log.call_args[0][0]
|
||||||
|
assert "Rate limit exceeded" in log_call_args
|
||||||
|
assert "Waiting for" in log_call_args
|
||||||
|
|
||||||
|
def test_rate_limited_different_rates(self):
|
||||||
|
"""Test different rate limiting values."""
|
||||||
|
|
||||||
|
@rate_limited(4.0) # 4 calls per second = 0.25 second intervals
|
||||||
|
def fast_function():
|
||||||
|
return time.time()
|
||||||
|
|
||||||
|
@rate_limited(0.5) # 0.5 calls per second = 2 second intervals
|
||||||
|
def slow_function():
|
||||||
|
return time.time()
|
||||||
|
|
||||||
|
# Test fast function (should wait ~0.25 seconds)
|
||||||
|
start = time.time()
|
||||||
|
fast_function()
|
||||||
|
fast_function()
|
||||||
|
fast_elapsed = time.time() - start
|
||||||
|
assert 0.2 <= fast_elapsed <= 0.4
|
||||||
|
|
||||||
|
# Test slow function (should wait ~2 seconds)
|
||||||
|
start = time.time()
|
||||||
|
slow_function()
|
||||||
|
slow_function()
|
||||||
|
slow_elapsed = time.time() - start
|
||||||
|
assert slow_elapsed >= 1.9
|
||||||
|
|
||||||
|
def test_multiple_instances(self):
|
||||||
|
"""Test that different decorated functions maintain separate rate limits."""
|
||||||
|
|
||||||
|
@rate_limited(2.0)
|
||||||
|
def function_a():
|
||||||
|
return "A"
|
||||||
|
|
||||||
|
@rate_limited(2.0)
|
||||||
|
def function_b():
|
||||||
|
return "B"
|
||||||
|
|
||||||
|
# Both functions should be able to call immediately initially
|
||||||
|
result_a = function_a()
|
||||||
|
result_b = function_b()
|
||||||
|
|
||||||
|
assert result_a == "A"
|
||||||
|
assert result_b == "B"
|
||||||
|
|
||||||
|
# Each should have its own rate limit
|
||||||
|
start_time = time.time()
|
||||||
|
function_a() # Should wait
|
||||||
|
elapsed = time.time() - start_time
|
||||||
|
assert elapsed >= 0.4
|
||||||
|
|
||||||
|
# function_b should still be able to call without waiting
|
||||||
|
start_time = time.time()
|
||||||
|
function_b() # Should not wait much since it has separate limit
|
||||||
|
elapsed = time.time() - start_time
|
||||||
|
assert elapsed >= 0.4 # It will also wait since it was called recently
|
||||||
@@ -0,0 +1,149 @@
|
|||||||
|
"""Test the Router symbol detection functionality."""
|
||||||
|
|
||||||
|
from unittest.mock import Mock, patch
|
||||||
|
from common.symbol_router import Router
|
||||||
|
from common.Symbol import Stock, Coin
|
||||||
|
|
||||||
|
|
||||||
|
class TestRouter:
|
||||||
|
"""Test Router symbol detection and parsing."""
|
||||||
|
|
||||||
|
def setup_method(self):
|
||||||
|
"""Set up test fixtures."""
|
||||||
|
# Mock the MarketData and cg_Crypto dependencies
|
||||||
|
with patch("common.symbol_router.MarketData") as mock_md, patch("common.symbol_router.cg_Crypto") as mock_crypto:
|
||||||
|
# Mock MarketData
|
||||||
|
mock_md_instance = Mock()
|
||||||
|
mock_md.return_value = mock_md_instance
|
||||||
|
|
||||||
|
# Mock cg_Crypto
|
||||||
|
mock_crypto_instance = Mock()
|
||||||
|
mock_crypto.return_value = mock_crypto_instance
|
||||||
|
|
||||||
|
# Create mock symbol list for crypto
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
mock_crypto_instance.symbol_list = pd.DataFrame(
|
||||||
|
{
|
||||||
|
"symbol": ["btc", "eth", "ada"],
|
||||||
|
"id": ["bitcoin", "ethereum", "cardano"],
|
||||||
|
"name": ["Bitcoin", "Ethereum", "Cardano"],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
self.router = Router()
|
||||||
|
self.mock_stock = mock_md_instance
|
||||||
|
self.mock_crypto = mock_crypto_instance
|
||||||
|
|
||||||
|
def test_stock_symbol_regex(self):
|
||||||
|
"""Test that stock symbol regex correctly identifies patterns."""
|
||||||
|
import re
|
||||||
|
|
||||||
|
# Test the regex pattern directly
|
||||||
|
pattern = self.router.STOCK_REGEX
|
||||||
|
|
||||||
|
# Should match
|
||||||
|
assert re.search(pattern, "$AAPL")
|
||||||
|
assert re.search(pattern, "What about $TSLA today?")
|
||||||
|
assert re.search(pattern, "Look at $NVDA and $AMD")
|
||||||
|
|
||||||
|
# Should not match crypto patterns
|
||||||
|
assert not re.search(pattern, "$$BTC")
|
||||||
|
|
||||||
|
# Should not match without $
|
||||||
|
assert not re.search(pattern, "AAPL")
|
||||||
|
|
||||||
|
def test_crypto_symbol_regex(self):
|
||||||
|
"""Test that crypto symbol regex correctly identifies patterns."""
|
||||||
|
import re
|
||||||
|
|
||||||
|
pattern = self.router.CRYPTO_REGEX
|
||||||
|
|
||||||
|
# Should match
|
||||||
|
assert re.search(pattern, "$$BTC")
|
||||||
|
assert re.search(pattern, "How is $$ETH doing?")
|
||||||
|
assert re.search(pattern, "$$BTC and $$ETH")
|
||||||
|
|
||||||
|
# Should not match single $ patterns
|
||||||
|
assert not re.search(pattern, "$BTC")
|
||||||
|
assert not re.search(pattern, "$AAPL")
|
||||||
|
|
||||||
|
def test_find_symbols_stock(self):
|
||||||
|
"""Test stock symbol detection."""
|
||||||
|
# Mock successful stock lookup
|
||||||
|
self.mock_stock.symbol_id.return_value = {"ticker": "AAPL", "title": "Apple Inc.", "mkt_cap_rank": 1}
|
||||||
|
|
||||||
|
result = self.router.find_symbols("$AAPL")
|
||||||
|
|
||||||
|
assert len(result) == 1
|
||||||
|
assert isinstance(result[0], Stock)
|
||||||
|
self.mock_stock.symbol_id.assert_called_with("AAPL")
|
||||||
|
|
||||||
|
def test_find_symbols_crypto(self):
|
||||||
|
"""Test crypto symbol detection."""
|
||||||
|
# Mock crypto dataframe lookup
|
||||||
|
import pandas as pd
|
||||||
|
|
||||||
|
mock_crypto_df = pd.DataFrame({"symbol": ["btc"], "id": ["bitcoin"], "name": ["Bitcoin"], "type_id": ["$$btc"]})
|
||||||
|
|
||||||
|
# Set up the mock to return the DataFrame when filtered
|
||||||
|
self.mock_crypto.symbol_list = pd.DataFrame(
|
||||||
|
{
|
||||||
|
"symbol": ["btc", "eth"],
|
||||||
|
"id": ["bitcoin", "ethereum"],
|
||||||
|
"name": ["Bitcoin", "Ethereum"],
|
||||||
|
"type_id": ["$$btc", "$$eth"],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Mock the pandas filtering
|
||||||
|
with patch.object(self.mock_crypto.symbol_list, "str") as mock_str:
|
||||||
|
mock_str.fullmatch.return_value = mock_crypto_df
|
||||||
|
mock_crypto_df.empty = False
|
||||||
|
|
||||||
|
result = self.router.find_symbols("$$BTC")
|
||||||
|
|
||||||
|
assert len(result) == 1
|
||||||
|
assert isinstance(result[0], Coin)
|
||||||
|
|
||||||
|
def test_find_symbols_no_matches(self):
|
||||||
|
"""Test behavior when no symbols are found."""
|
||||||
|
result = self.router.find_symbols("Hello world")
|
||||||
|
assert result == []
|
||||||
|
|
||||||
|
def test_find_symbols_invalid_stock(self):
|
||||||
|
"""Test behavior when stock symbol is not found."""
|
||||||
|
# Mock failed stock lookup
|
||||||
|
self.mock_stock.symbol_id.return_value = None
|
||||||
|
|
||||||
|
result = self.router.find_symbols("$INVALID")
|
||||||
|
|
||||||
|
assert result == []
|
||||||
|
self.mock_stock.symbol_id.assert_called_with("INVALID")
|
||||||
|
|
||||||
|
def test_find_symbols_multiple(self):
|
||||||
|
"""Test finding multiple symbols."""
|
||||||
|
# Mock successful lookups for both
|
||||||
|
self.mock_stock.symbol_id.side_effect = [
|
||||||
|
{"ticker": "AAPL", "title": "Apple Inc.", "mkt_cap_rank": 1},
|
||||||
|
{"ticker": "TSLA", "title": "Tesla Inc.", "mkt_cap_rank": 5},
|
||||||
|
]
|
||||||
|
|
||||||
|
result = self.router.find_symbols("$AAPL and $TSLA")
|
||||||
|
|
||||||
|
assert len(result) == 2
|
||||||
|
assert all(isinstance(symbol, Stock) for symbol in result)
|
||||||
|
|
||||||
|
def test_trending_weight_tracking(self):
|
||||||
|
"""Test that trending counts are updated."""
|
||||||
|
# Mock successful stock lookup
|
||||||
|
self.mock_stock.symbol_id.return_value = {"ticker": "AAPL", "title": "Apple Inc.", "mkt_cap_rank": 1}
|
||||||
|
|
||||||
|
# Clear trending count
|
||||||
|
self.router.trending_count = {}
|
||||||
|
|
||||||
|
result = self.router.find_symbols("$AAPL", trending_weight=2)
|
||||||
|
|
||||||
|
assert len(result) == 1
|
||||||
|
assert "$AAPL" in self.router.trending_count
|
||||||
|
assert self.router.trending_count["$AAPL"] == 2
|
||||||
@@ -0,0 +1,128 @@
|
|||||||
|
"""Test Symbol, Stock, and Coin class construction."""
|
||||||
|
|
||||||
|
from unittest.mock import patch
|
||||||
|
import pandas as pd
|
||||||
|
from common.Symbol import Symbol, Stock, Coin
|
||||||
|
|
||||||
|
|
||||||
|
class TestSymbol:
|
||||||
|
"""Test base Symbol class."""
|
||||||
|
|
||||||
|
def test_symbol_creation(self):
|
||||||
|
"""Test basic Symbol creation."""
|
||||||
|
symbol = Symbol("TEST")
|
||||||
|
assert symbol.symbol == "TEST"
|
||||||
|
assert symbol.id == "TEST"
|
||||||
|
assert symbol.name == "TEST"
|
||||||
|
|
||||||
|
def test_symbol_currency_default(self):
|
||||||
|
"""Test that currency defaults to USD."""
|
||||||
|
symbol = Symbol("TEST")
|
||||||
|
assert symbol.currency == "usd"
|
||||||
|
|
||||||
|
|
||||||
|
class TestStock:
|
||||||
|
"""Test Stock class functionality."""
|
||||||
|
|
||||||
|
def test_stock_creation(self):
|
||||||
|
"""Test Stock object creation with proper dictionary structure."""
|
||||||
|
stock_info = {"ticker": "AAPL", "title": "Apple Inc.", "mkt_cap_rank": 1}
|
||||||
|
|
||||||
|
stock = Stock(stock_info)
|
||||||
|
assert stock.symbol == "AAPL"
|
||||||
|
assert stock.id == "AAPL"
|
||||||
|
assert stock.name == "Apple Inc."
|
||||||
|
assert stock.tag == "$AAPL"
|
||||||
|
assert stock.market_cap_rank == 1
|
||||||
|
|
||||||
|
# Verify inheritance
|
||||||
|
assert isinstance(stock, Symbol)
|
||||||
|
assert isinstance(stock, Stock)
|
||||||
|
|
||||||
|
def test_stock_with_different_data(self):
|
||||||
|
"""Test Stock creation with different stock data."""
|
||||||
|
stock_info = {"ticker": "TSLA", "title": "Tesla, Inc.", "mkt_cap_rank": 5}
|
||||||
|
|
||||||
|
stock = Stock(stock_info)
|
||||||
|
assert stock.symbol == "TSLA"
|
||||||
|
assert stock.id == "TSLA"
|
||||||
|
assert stock.name == "Tesla, Inc."
|
||||||
|
assert stock.tag == "$TSLA"
|
||||||
|
assert stock.market_cap_rank == 5
|
||||||
|
|
||||||
|
def test_stock_tag_format(self):
|
||||||
|
"""Test that stock tags are properly formatted with $ prefix."""
|
||||||
|
stock_info = {"ticker": "NVDA", "title": "NVIDIA Corporation", "mkt_cap_rank": 3}
|
||||||
|
|
||||||
|
stock = Stock(stock_info)
|
||||||
|
assert stock.tag == "$NVDA"
|
||||||
|
assert stock.tag.startswith("$")
|
||||||
|
|
||||||
|
|
||||||
|
class TestCoin:
|
||||||
|
"""Test Coin class functionality."""
|
||||||
|
|
||||||
|
def test_coin_creation_single_row(self):
|
||||||
|
"""Test Coin object creation with single row DataFrame."""
|
||||||
|
coin_data = pd.DataFrame({"symbol": ["btc"], "id": ["bitcoin"], "name": ["Bitcoin"], "type_id": ["$$btc"]})
|
||||||
|
|
||||||
|
coin = Coin(coin_data)
|
||||||
|
assert coin.symbol == "btc"
|
||||||
|
assert coin.id == "bitcoin"
|
||||||
|
assert coin.name == "Bitcoin"
|
||||||
|
assert coin.tag == "$$BTC"
|
||||||
|
|
||||||
|
# Verify inheritance
|
||||||
|
assert isinstance(coin, Symbol)
|
||||||
|
assert isinstance(coin, Coin)
|
||||||
|
|
||||||
|
def test_coin_creation_ethereum(self):
|
||||||
|
"""Test Coin creation with Ethereum data."""
|
||||||
|
coin_data = pd.DataFrame({"symbol": ["eth"], "id": ["ethereum"], "name": ["Ethereum"], "type_id": ["$$eth"]})
|
||||||
|
|
||||||
|
coin = Coin(coin_data)
|
||||||
|
assert coin.symbol == "eth"
|
||||||
|
assert coin.id == "ethereum"
|
||||||
|
assert coin.name == "Ethereum"
|
||||||
|
assert coin.tag == "$$ETH"
|
||||||
|
|
||||||
|
def test_coin_multiple_rows_takes_first(self):
|
||||||
|
"""Test that when multiple rows exist, only the first is used."""
|
||||||
|
coin_data = pd.DataFrame(
|
||||||
|
{
|
||||||
|
"symbol": ["btc", "btc2"],
|
||||||
|
"id": ["bitcoin", "bitcoin-cash"],
|
||||||
|
"name": ["Bitcoin", "Bitcoin Cash"],
|
||||||
|
"type_id": ["$$btc", "$$bch"],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Mock the head method to return first row
|
||||||
|
with patch.object(coin_data, "head") as mock_head:
|
||||||
|
first_row = pd.DataFrame({"symbol": ["btc"], "id": ["bitcoin"], "name": ["Bitcoin"], "type_id": ["$$btc"]})
|
||||||
|
mock_head.return_value = first_row
|
||||||
|
|
||||||
|
coin = Coin(coin_data)
|
||||||
|
mock_head.assert_called_once_with(1)
|
||||||
|
# The coin should use the first row data
|
||||||
|
assert coin.symbol == "btc"
|
||||||
|
assert coin.id == "bitcoin"
|
||||||
|
|
||||||
|
def test_coin_tag_uppercase(self):
|
||||||
|
"""Test that coin tags are properly uppercased."""
|
||||||
|
coin_data = pd.DataFrame({"symbol": ["ada"], "id": ["cardano"], "name": ["Cardano"], "type_id": ["$$ada"]})
|
||||||
|
|
||||||
|
coin = Coin(coin_data)
|
||||||
|
assert coin.tag == "$$ADA"
|
||||||
|
assert coin.tag.isupper()
|
||||||
|
|
||||||
|
def test_coin_dataframe_access(self):
|
||||||
|
"""Test that coin properly accesses DataFrame values."""
|
||||||
|
coin_data = pd.DataFrame({"symbol": ["dot"], "id": ["polkadot"], "name": ["Polkadot"], "type_id": ["$$dot"]})
|
||||||
|
|
||||||
|
coin = Coin(coin_data)
|
||||||
|
# Test that values are extracted correctly from DataFrame
|
||||||
|
assert isinstance(coin.symbol, str)
|
||||||
|
assert isinstance(coin.id, str)
|
||||||
|
assert isinstance(coin.name, str)
|
||||||
|
assert isinstance(coin.tag, str)
|
||||||
Reference in New Issue
Block a user