46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
# Twisted/Klein imports
|
|
from twisted.logger import Logger
|
|
|
|
# Other library imports
|
|
import requests
|
|
|
|
# Project imports
|
|
from settings import settings
|
|
|
|
|
|
class Notify(object):
|
|
"""
|
|
Class to handle more robust notifications.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.log = Logger("notify")
|
|
|
|
def sendmsg(self, msg, title=None, priority=None, tags=None):
|
|
headers = {"Title": "Bot"}
|
|
if title:
|
|
headers["Title"] = title
|
|
if priority:
|
|
headers["Priority"] = priority
|
|
if tags:
|
|
headers["Tags"] = tags
|
|
requests.post(
|
|
f"{settings.Notify.Host}/{settings.Notify.Topic}",
|
|
data=msg,
|
|
headers=headers,
|
|
)
|
|
|
|
def notify_new_trade(self, amount, currency):
|
|
amount_usd = self.agora.to_usd(amount, currency)
|
|
self.sendmsg(f"Total: {amount_usd}", title="New trade", tags="trades", priority="2")
|
|
|
|
def notify_complete_trade(self, amount, currency):
|
|
amount_usd = self.agora.to_usd(amount, currency)
|
|
self.sendmsg(f"Total: {amount_usd}", title="Trade complete", tags="trades,profit", priority="3")
|
|
|
|
def notify_withdrawal(self, amount_usd):
|
|
self.sendmsg(f"Total: {amount_usd}", title="Withdrawal", tags="profit", priority="4")
|
|
|
|
def notify_need_topup(self, amount_usd_xmr, amount_usd_btc):
|
|
self.sendmsg(f"XMR: {amount_usd_xmr} | BTC: {amount_usd_btc}", title="Topup needed", tags="admin", priority="5")
|