287 lines
11 KiB
Python
287 lines
11 KiB
Python
# Twisted/Klein imports
|
|
from twisted.logger import Logger
|
|
from twisted.internet.task import LoopingCall
|
|
|
|
# Other library imports
|
|
from json import loads
|
|
from forex_python.converter import CurrencyRates
|
|
from agoradesk_py.agoradesk import AgoraDesk
|
|
|
|
# Project imports
|
|
from settings import settings
|
|
|
|
|
|
class Agora(object):
|
|
"""
|
|
AgoraDesk API handler.
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""
|
|
Initialise the AgoraDesk and CurrencyRates APIs.
|
|
Initialise the last_dash storage for detecting new trades.
|
|
"""
|
|
self.log = Logger("agora")
|
|
self.agora = AgoraDesk(settings.Agora.Token)
|
|
self.cr = CurrencyRates()
|
|
|
|
# Cache for detecting new trades
|
|
self.last_dash = set()
|
|
|
|
# Cache for detecting new messages
|
|
self.last_messages = {}
|
|
|
|
def set_irc(self, irc):
|
|
self.irc = irc
|
|
|
|
def set_tx(self, tx):
|
|
self.tx = tx
|
|
|
|
def setup_loop(self):
|
|
"""
|
|
Set up the LoopingCall to get all active trades and messages.
|
|
"""
|
|
self.lc_dash = LoopingCall(self.dashboard)
|
|
self.lc_dash.start(int(settings.Agora.RefreshSec))
|
|
|
|
def dashboard(self, send_irc=True, formatted=False):
|
|
"""
|
|
Returns a dict for the dashboard.
|
|
Calls hooks to parse dashboard info and get all contact messages.
|
|
:return: dict of dashboard info
|
|
:rtype: dict
|
|
"""
|
|
dash = self.agora.dashboard_seller()
|
|
dash_tmp = {}
|
|
if "data" not in dash["response"].keys():
|
|
self.log.error("Data not in dashboard response: {content}", content=dash)
|
|
return False
|
|
if dash["response"]["data"]["contact_count"] > 0:
|
|
for contact in dash["response"]["data"]["contact_list"]:
|
|
contact_id = contact["data"]["contact_id"]
|
|
dash_tmp[contact_id] = contact
|
|
fmt = self.dashboard_hook(dash_tmp)
|
|
self.get_all_messages_hook(dash_tmp)
|
|
if formatted:
|
|
return fmt
|
|
return dash_tmp
|
|
|
|
def dashboard_hook(self, dash, send_irc=True):
|
|
"""
|
|
Get information about our open trades.
|
|
Post new trades to IRC and cache trades for the future.
|
|
:return: human readable list of strings about our trades or False
|
|
:rtype: list or bool
|
|
"""
|
|
dash_tmp = []
|
|
current_trades = []
|
|
for contact_id, contact in dash.items():
|
|
reference = self.tx.tx_to_ref(contact_id)
|
|
if reference:
|
|
current_trades.append(reference)
|
|
buyer = contact["data"]["buyer"]["username"]
|
|
amount = contact["data"]["amount"]
|
|
amount_xmr = contact["data"]["amount_xmr"]
|
|
currency = contact["data"]["currency"]
|
|
if not contact["data"]["is_selling"]:
|
|
continue
|
|
if reference not in self.last_dash:
|
|
reference = self.tx.new_trade(contact_id, buyer, currency, amount, amount_xmr)
|
|
if reference:
|
|
if reference not in current_trades:
|
|
current_trades.append(reference)
|
|
if send_irc:
|
|
self.irc.client.msg(self.irc.client.channel, f"AUTO {reference}: {buyer} {amount}{currency} {amount_xmr}XMR")
|
|
|
|
dash_tmp.append(f"{reference}: {buyer} {amount}{currency} {amount_xmr}XMR")
|
|
self.last_dash.add(reference)
|
|
|
|
# Purge old trades from cache
|
|
for ref in list(self.last_dash): # We're removing from the list on the fly
|
|
if ref not in current_trades:
|
|
self.last_dash.remove(ref)
|
|
if reference and reference not in current_trades:
|
|
current_trades.append(reference)
|
|
self.tx.cleanup(current_trades)
|
|
return dash_tmp
|
|
|
|
def dashboard_release_urls(self):
|
|
"""
|
|
Get information about our open trades.
|
|
Post new trades to IRC and cache trades for the future.
|
|
:return: human readable list of strings about our trades or False
|
|
:rtype: list or bool
|
|
"""
|
|
dash = self.agora.dashboard_seller()
|
|
dash_tmp = []
|
|
if "data" not in dash["response"]:
|
|
self.log.error("Data not in dashboard response: {content}", content=dash)
|
|
return False
|
|
if dash["response"]["data"]["contact_count"] > 0:
|
|
for contact in dash["response"]["data"]["contact_list"]:
|
|
contact_id = contact["data"]["contact_id"]
|
|
buyer = contact["data"]["buyer"]["username"]
|
|
amount = contact["data"]["amount"]
|
|
amount_xmr = contact["data"]["amount_xmr"]
|
|
currency = contact["data"]["currency"]
|
|
release_url = contact["actions"]["release_url"]
|
|
if not contact["data"]["is_selling"]:
|
|
continue
|
|
reference = self.tx.tx_to_ref(contact_id)
|
|
if not reference:
|
|
reference = "not_set"
|
|
dash_tmp.append(f"{reference}: {buyer} {amount}{currency} {amount_xmr}XMR {release_url}")
|
|
|
|
return dash_tmp
|
|
|
|
def get_messages(self, contact_id, send_irc=True):
|
|
"""
|
|
Get messages for a certain trade ID.
|
|
Post new messages to IRC and cache messages for the future.
|
|
:param contact_id: trade/contact ID
|
|
:return: list of messages
|
|
:rtype: list
|
|
"""
|
|
messages = self.agora.contact_messages(contact_id)
|
|
messages_tmp = []
|
|
reference = self.tx.tx_to_ref(contact_id)
|
|
if not reference:
|
|
reference = "not_set"
|
|
for message in messages["response"]["data"]["message_list"]:
|
|
messages_tmp.append(f"({message['sender']['username']}): {message['msg']}")
|
|
if send_irc:
|
|
if reference in self.last_messages:
|
|
if not len(self.last_messages[reference]) == len(messages_tmp):
|
|
difference = set(messages_tmp) ^ set(self.last_messages[reference])
|
|
for x in difference:
|
|
self.irc.client.msg(self.irc.client.channel, f"AUTO {reference}: {x}")
|
|
else:
|
|
self.last_messages[reference] = messages_tmp
|
|
for x in messages_tmp:
|
|
self.irc.client.msg(self.irc.client.channel, f"NEW {reference}: {x}")
|
|
return messages_tmp
|
|
|
|
def get_all_messages(self, send_irc=True):
|
|
"""
|
|
Get all messages for all open trades.
|
|
Wrapper for passing in dashboard.
|
|
:return: dict of lists keyed by trade/contact ID
|
|
:rtype: dict with lists
|
|
"""
|
|
dash = self.dashboard()
|
|
if dash is None:
|
|
return False
|
|
return self.get_all_messages_hook(dash, send_irc=send_irc)
|
|
|
|
def get_all_messages_hook(self, dash, send_irc=True):
|
|
"""
|
|
Get all messages for all open trades.
|
|
:return: dict of lists keyed by trade/contact ID
|
|
:rtype: dict with lists
|
|
"""
|
|
messages_tmp = {}
|
|
for contact_id in dash:
|
|
reference = self.tx.tx_to_ref(contact_id)
|
|
if not reference:
|
|
reference = "not_set"
|
|
messages = self.get_messages(contact_id, send_irc=send_irc)
|
|
messages_tmp[reference] = messages
|
|
|
|
# Purge old trades from cache
|
|
for ref in list(self.last_messages): # We're removing from the list on the fly
|
|
if ref not in messages_tmp:
|
|
del self.last_messages[ref]
|
|
|
|
return messages_tmp
|
|
|
|
def get_ads(self):
|
|
"""
|
|
Get information about our ads.
|
|
:return: data about our open ads
|
|
:rtype: dict
|
|
"""
|
|
ads = self.agora.ads()
|
|
return ads
|
|
|
|
def get_rates_all(self):
|
|
"""
|
|
Get all rates that pair with USD.
|
|
:return: dictionary of USD/XXX rates
|
|
:rtype: dict
|
|
"""
|
|
rates = self.cr.get_rates("USD")
|
|
return rates
|
|
|
|
def get_acceptable_margins(self, currency, amount):
|
|
"""
|
|
Get the minimum and maximum amounts we would accept a trade for.
|
|
:param currency: currency code
|
|
:param amount: amount
|
|
:return: (min, max)
|
|
:rtype: tuple
|
|
"""
|
|
rates = self.get_rates_all()
|
|
if currency == "USD":
|
|
min_amount = amount - float(settings.Agora.AcceptableUSDMargin)
|
|
max_amount = amount + float(settings.Agora.AcceptableUSDMargin)
|
|
return (min_amount, max_amount)
|
|
amount_usd = amount / rates[currency]
|
|
min_usd = amount_usd - float(settings.Agora.AcceptableUSDMargin)
|
|
max_usd = amount_usd + float(settings.Agora.AcceptableUSDMargin)
|
|
min_local = min_usd * rates[currency]
|
|
max_local = max_usd * rates[currency]
|
|
return (min_local, max_local)
|
|
|
|
def create_ad(self, countrycode, currency):
|
|
"""
|
|
Post an ad in a country with a given currency.
|
|
Convert the min and max amounts from settings to the given currency with CurrencyRates.
|
|
:param countrycode: country code
|
|
:param currency: currency code
|
|
:type countrycode: string
|
|
:type currency: string
|
|
:return: data about created object or error
|
|
:rtype: dict
|
|
"""
|
|
rates = self.get_rates_all()
|
|
if currency == "USD":
|
|
min_amount = float(settings.Agora.MinUSD)
|
|
max_amount = float(settings.Agora.MaxUSD)
|
|
else:
|
|
min_amount = rates[currency] * float(settings.Agora.MinUSD)
|
|
max_amount = rates[currency] * float(settings.Agora.MaxUSD)
|
|
price_formula = f"coingeckoxmrusd*usd{currency.lower()}*{settings.Agora.Margin}"
|
|
# price_formula = f"coingeckoxmrusd*{settings.Agora.Margin}"
|
|
ad = settings.Agora.Ad
|
|
ad = ad.replace("\\t", "\t")
|
|
ad = self.agora.ad_create(
|
|
country_code=countrycode,
|
|
currency=currency,
|
|
trade_type="ONLINE_SELL",
|
|
asset="XMR",
|
|
price_equation=price_formula,
|
|
track_max_amount=False,
|
|
require_trusted_by_advertiser=False,
|
|
# verified_email_required = False,
|
|
online_provider="REVOLUT",
|
|
msg=settings.Agora.Ad,
|
|
min_amount=min_amount,
|
|
max_amount=max_amount,
|
|
payment_method_details=settings.Agora.PaymentMethodDetails,
|
|
# require_feedback_score = 0,
|
|
account_info=settings.Agora.PaymentDetails,
|
|
)
|
|
return ad
|
|
|
|
def dist_countries(self):
|
|
"""
|
|
Distribute our advert into all countries listed in the config.
|
|
:return: True or False
|
|
:rtype: bool
|
|
"""
|
|
for currency, countrycode in loads(settings.Agora.DistList):
|
|
rtrn = self.create_ad(countrycode, currency)
|
|
if not rtrn:
|
|
return False
|
|
return True
|