Write docstrings for Agora and Transactions classes
This commit is contained in:
parent
a5bd7051df
commit
e15cec7afe
|
@ -18,6 +18,8 @@ from settings import settings
|
||||||
|
|
||||||
log = Logger("agora.global")
|
log = Logger("agora.global")
|
||||||
|
|
||||||
|
# TODO: move to utils
|
||||||
|
|
||||||
|
|
||||||
def handle_exceptions(func):
|
def handle_exceptions(func):
|
||||||
def inner_function(*args, **kwargs):
|
def inner_function(*args, **kwargs):
|
||||||
|
@ -594,6 +596,7 @@ class Agora(object):
|
||||||
ad = self.agora.ad_create(**form)
|
ad = self.agora.ad_create(**form)
|
||||||
return ad
|
return ad
|
||||||
|
|
||||||
|
# TODO: move to markets
|
||||||
def create_distribution_list(self, filter_asset=None):
|
def create_distribution_list(self, filter_asset=None):
|
||||||
"""
|
"""
|
||||||
Create a list for distribution of ads.
|
Create a list for distribution of ads.
|
||||||
|
|
|
@ -24,6 +24,7 @@ class Transactions(object):
|
||||||
"""
|
"""
|
||||||
self.log = Logger("transactions")
|
self.log = Logger("transactions")
|
||||||
|
|
||||||
|
# TODO: write tests then refactor, this is terribly complicated!
|
||||||
def transaction(self, data):
|
def transaction(self, data):
|
||||||
"""
|
"""
|
||||||
Store details of transaction and post notifications to IRC.
|
Store details of transaction and post notifications to IRC.
|
||||||
|
@ -284,6 +285,7 @@ class Transactions(object):
|
||||||
"""
|
"""
|
||||||
refs = self.get_refs()
|
refs = self.get_refs()
|
||||||
matching_refs = []
|
matching_refs = []
|
||||||
|
# TODO: use get_ref_map in this function instead of calling get_ref multiple times
|
||||||
for ref in refs:
|
for ref in refs:
|
||||||
stored_trade = self.get_ref(ref)
|
stored_trade = self.get_ref(ref)
|
||||||
if stored_trade["currency"] == currency and float(stored_trade["amount"]) == float(amount):
|
if stored_trade["currency"] == currency and float(stored_trade["amount"]) == float(amount):
|
||||||
|
@ -320,9 +322,11 @@ class Transactions(object):
|
||||||
|
|
||||||
def get_ref(self, reference):
|
def get_ref(self, reference):
|
||||||
"""
|
"""
|
||||||
Get a reference ID for a single trade.
|
Get the trade information for a reference.
|
||||||
:return: trade ID
|
:param reference: trade reference
|
||||||
:rtype: string
|
:type reference: string
|
||||||
|
:return: dict of trade information
|
||||||
|
:rtype: dict
|
||||||
"""
|
"""
|
||||||
ref_data = r.hgetall(f"trade.{reference}")
|
ref_data = r.hgetall(f"trade.{reference}")
|
||||||
ref_data = convert(ref_data)
|
ref_data = convert(ref_data)
|
||||||
|
@ -333,12 +337,20 @@ class Transactions(object):
|
||||||
def del_ref(self, reference):
|
def del_ref(self, reference):
|
||||||
"""
|
"""
|
||||||
Delete a given reference from the Redis database.
|
Delete a given reference from the Redis database.
|
||||||
|
:param reference: trade reference to delete
|
||||||
|
:type reference: string
|
||||||
"""
|
"""
|
||||||
tx = self.ref_to_tx(reference)
|
tx = self.ref_to_tx(reference)
|
||||||
r.delete(f"trade.{reference}")
|
r.delete(f"trade.{reference}")
|
||||||
r.delete(f"trade.{tx}.reference")
|
r.delete(f"trade.{tx}.reference")
|
||||||
|
|
||||||
def cleanup(self, references):
|
def cleanup(self, references):
|
||||||
|
"""
|
||||||
|
Reconcile the internal reference database with a given list of references.
|
||||||
|
Delete all internal references not present in the list and clean up artifacts.
|
||||||
|
:param references: list of references to reconcile against
|
||||||
|
:type references: list
|
||||||
|
"""
|
||||||
for tx, reference in self.get_ref_map().items():
|
for tx, reference in self.get_ref_map().items():
|
||||||
if reference not in references:
|
if reference not in references:
|
||||||
self.log.info("Archiving trade reference: {reference} / TX: {tx}", reference=reference, tx=tx)
|
self.log.info("Archiving trade reference: {reference} / TX: {tx}", reference=reference, tx=tx)
|
||||||
|
@ -346,6 +358,13 @@ class Transactions(object):
|
||||||
r.rename(f"trade.{reference}", f"archive.trade.{reference}")
|
r.rename(f"trade.{reference}", f"archive.trade.{reference}")
|
||||||
|
|
||||||
def tx_to_ref(self, tx):
|
def tx_to_ref(self, tx):
|
||||||
|
"""
|
||||||
|
Convert a trade ID to a reference.
|
||||||
|
:param tx: trade ID
|
||||||
|
:type tx: string
|
||||||
|
:return: reference
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
refs = self.get_refs()
|
refs = self.get_refs()
|
||||||
for reference in refs:
|
for reference in refs:
|
||||||
ref_data = convert(r.hgetall(f"trade.{reference}"))
|
ref_data = convert(r.hgetall(f"trade.{reference}"))
|
||||||
|
@ -355,12 +374,24 @@ class Transactions(object):
|
||||||
return reference
|
return reference
|
||||||
|
|
||||||
def ref_to_tx(self, reference):
|
def ref_to_tx(self, reference):
|
||||||
|
"""
|
||||||
|
Convert a reference to a trade ID.
|
||||||
|
:param reference: trade reference
|
||||||
|
:type reference: string
|
||||||
|
:return: trade ID
|
||||||
|
:rtype: string
|
||||||
|
"""
|
||||||
ref_data = convert(r.hgetall(f"trade.{reference}"))
|
ref_data = convert(r.hgetall(f"trade.{reference}"))
|
||||||
if not ref_data:
|
if not ref_data:
|
||||||
return False
|
return False
|
||||||
return ref_data["id"]
|
return ref_data["id"]
|
||||||
|
|
||||||
def get_total_usd(self):
|
def get_total_usd(self):
|
||||||
|
"""
|
||||||
|
Get total USD in all our accounts, bank and trading.
|
||||||
|
:return: value in USD
|
||||||
|
:rtype float:
|
||||||
|
"""
|
||||||
total_usd_revolut = self.revolut.get_total_usd()
|
total_usd_revolut = self.revolut.get_total_usd()
|
||||||
if total_usd_revolut is False:
|
if total_usd_revolut is False:
|
||||||
return False
|
return False
|
||||||
|
@ -389,7 +420,14 @@ class Transactions(object):
|
||||||
total_usd = total_usd_agora + total_usd_revolut
|
total_usd = total_usd_agora + total_usd_revolut
|
||||||
return total_usd
|
return total_usd
|
||||||
|
|
||||||
|
# TODO: possibly refactor this into smaller functions which don't return as much stuff
|
||||||
|
# check if this is all really needed in the corresponding withdraw function
|
||||||
def get_total(self):
|
def get_total(self):
|
||||||
|
"""
|
||||||
|
Get all the values corresponding to the amount of money we hold.
|
||||||
|
:return: ((total SEK, total USD, total GBP), (total XMR USD, total BTC USD), (total XMR, total BTC))
|
||||||
|
:rtype: tuple(tuple(float, float, float), tuple(float, float), tuple(float, float))
|
||||||
|
"""
|
||||||
total_usd_revolut = self.revolut.get_total_usd()
|
total_usd_revolut = self.revolut.get_total_usd()
|
||||||
if total_usd_revolut is False:
|
if total_usd_revolut is False:
|
||||||
self.log.error("Could not get USD total.")
|
self.log.error("Could not get USD total.")
|
||||||
|
@ -433,6 +471,11 @@ class Transactions(object):
|
||||||
) # Total XMR and BTC balance in Agora
|
) # Total XMR and BTC balance in Agora
|
||||||
|
|
||||||
def get_remaining(self):
|
def get_remaining(self):
|
||||||
|
"""
|
||||||
|
Check how much profit we need to make in order to withdraw.
|
||||||
|
:return: profit remaining in USD
|
||||||
|
:rtype: float
|
||||||
|
"""
|
||||||
total_usd = self.get_total_usd()
|
total_usd = self.get_total_usd()
|
||||||
if not total_usd:
|
if not total_usd:
|
||||||
return False
|
return False
|
||||||
|
@ -443,6 +486,11 @@ class Transactions(object):
|
||||||
return remaining
|
return remaining
|
||||||
|
|
||||||
def get_open_trades_usd(self):
|
def get_open_trades_usd(self):
|
||||||
|
"""
|
||||||
|
Get total value of open trades in USD.
|
||||||
|
:return: total trade value
|
||||||
|
:rtype: float
|
||||||
|
"""
|
||||||
dash = self.agora.wrap_dashboard()
|
dash = self.agora.wrap_dashboard()
|
||||||
if dash is False:
|
if dash is False:
|
||||||
return False
|
return False
|
||||||
|
@ -463,6 +511,11 @@ class Transactions(object):
|
||||||
return cumul_usd
|
return cumul_usd
|
||||||
|
|
||||||
def get_total_remaining(self):
|
def get_total_remaining(self):
|
||||||
|
"""
|
||||||
|
Check how much profit we need to make in order to withdraw, taking into account open trade value.
|
||||||
|
:return: profit remaining in USD
|
||||||
|
:rtype: float
|
||||||
|
"""
|
||||||
total_usd = self.get_total_usd()
|
total_usd = self.get_total_usd()
|
||||||
total_trades_usd = self.get_open_trades_usd()
|
total_trades_usd = self.get_open_trades_usd()
|
||||||
if not total_usd:
|
if not total_usd:
|
||||||
|
|
Loading…
Reference in New Issue