2 # Put your poloniex api key in market.py
3 from market
import market
5 def static_var(varname
, value
):
7 setattr(func
, varname
, value
)
13 repartition_pertenthousand
= {
22 def formatted_price(value
):
23 return round(value
/ 10**max_digits
, 8)
25 @static_var("cache", {})
26 def get_ticker(c1
, c2
, market
):
30 "bid": float(1/ticker
["ask"]),
31 "ask": float(1/ticker
["bid"]),
32 "bidA": float(1/ticker
["askA"]),
33 "askA": float(1/ticker
["bidA"]),
35 def augment_ticker(ticker
):
36 # FIXME: need to do better than just a multiplier
39 "bidA": ticker
["bid"] * 0.99,
40 "askA": ticker
["ask"] * 1.01,
43 if (c1
, c2
, market
.__class
__) in get_ticker
.cache
:
44 return get_ticker
.cache
[(c1
, c2
, market
.__class
__)]
45 if (c2
, c1
, market
.__class
__) in get_ticker
.cache
:
46 return invert(get_ticker
.cache
[(c2
, c1
, market
.__class
__)])
49 get_ticker
.cache
[(c1
, c2
, market
.__class
__)] = market
.fetch_ticker("{}/{}".format(c1
, c2
))
50 augment_ticker(get_ticker
.cache
[(c1
, c2
, market
.__class
__)])
51 except ccxt
.ExchangeError
:
53 get_ticker
.cache
[(c2
, c1
, market
.__class
__)] = market
.fetch_ticker("{}/{}".format(c2
, c1
))
54 augment_ticker(get_ticker
.cache
[(c2
, c1
, market
.__class
__)])
55 except ccxt
.ExchangeError
:
56 get_ticker
.cache
[(c1
, c2
, market
.__class
__)] = None
57 return get_ticker(c1
, c2
, market
)
59 def fetch_balances(market
):
61 fetched_balance
= market
.fetch_balance()
62 for key
, value
in fetched_balance
["total"].items():
64 balances
[key
] = int(value
* 10**max_digits
)
67 def assets_value(assets
, market
, base_currency
="BTC"):
68 repartition_in_base_currency
= {}
69 for currency
, asset_value
in assets
.items():
70 if currency
== base_currency
:
71 repartition_in_base_currency
[currency
] = asset_value
73 asset_ticker
= get_ticker(currency
, base_currency
, market
)
74 if asset_ticker
is None:
75 raise Exception("This asset is not available in the chosen market")
76 repartition_in_base_currency
[currency
] = int(asset_ticker
["bidA"] * asset_value
)
77 return repartition_in_base_currency
79 def dispatch_assets(base_currency_value
, repartition_pertenthousand
, market
, base_currency
="BTC"):
80 sum_pertenthousand
= sum([v
for k
, v
in repartition_pertenthousand
.items()])
81 repartition_in_base_currency
= {}
82 for currency
, ptt
in repartition_pertenthousand
.items():
83 repartition_in_base_currency
[currency
] = int(ptt
* base_currency_value
/ sum_pertenthousand
)
84 return repartition_in_base_currency
86 def give_orders(current_assets
, repartition_pertenthousand
, market
, base_currency
="BTC"):
87 value_in_base
= assets_value(current_assets
, market
, base_currency
=base_currency
)
88 total_base_value
= sum([ v
for k
, v
in value_in_base
.items()])
90 new_repartition
= dispatch_assets(total_base_value
, repartition_pertenthousand
, market
, base_currency
=base_currency
)
93 for key
in set(value_in_base
.keys()).union(set(new_repartition
.keys())):
94 mouvements
[key
] = value_in_base
.get(key
, 0) - new_repartition
.get(key
, 0)
96 print("assets before repartition:")
97 for currency
, value
in current_assets
.items():
98 print("holding {} {}".format(
99 formatted_price(value
),
101 print("------------")
102 for currency
, value
in mouvements
.items():
103 if currency
== base_currency
:
105 asset_ticker
= get_ticker(currency
, base_currency
, market
)
108 currency_price
= int(value
/ asset_ticker
["bid"])
111 currency_price
= int(value
/ asset_ticker
["ask"])
113 print("need to {} {} {}'s worth of {}, i.e. {} {}".format(
115 formatted_price(abs(value
)),
118 formatted_price(abs(currency_price
)),
120 print("------------\nassets after repartition:")
121 for currency
, value
in new_repartition
.items():
122 print("holding {} {}".format(formatted_price(value
), currency
))
124 current_assets
= fetch_balances(market
)
125 give_orders(current_assets
, repartition_pertenthousand
, market
)