104 lines
3.5 KiB
Python
104 lines
3.5 KiB
Python
import krakenex
|
|
import os
|
|
import pandas
|
|
import time
|
|
|
|
from dotenv import load_dotenv
|
|
from pykrakenapi import KrakenAPI
|
|
from scipy.stats import linregress
|
|
from time import sleep
|
|
|
|
load_dotenv()
|
|
|
|
R_VALUE_TARGET = float(os.getenv('R_VALUE_TARGET'))
|
|
INVESTMENT_COUNT = int(os.getenv('INVESTMENT_COUNT'))
|
|
INVESTMENT_VOLUME = float(os.getenv('INVESTMENT_VOLUME'))
|
|
|
|
k = krakenex.API(
|
|
os.getenv('KRAKEN_API_TOKEN'),
|
|
os.getenv('KRAKEN_API_SEC')
|
|
)
|
|
kraken = KrakenAPI(k, tier='Pro')
|
|
|
|
def filter_asset_pairs(tradable_asset_pairs):
|
|
return tradable_asset_pairs[
|
|
(tradable_asset_pairs['quote'] == 'ZUSD') &
|
|
(tradable_asset_pairs['status'] == 'online')
|
|
]
|
|
|
|
def model_asset_pair(asset_pair):
|
|
ohlc = kraken.get_ohlc_data(asset_pair)[0]
|
|
sleep(kraken.factor)
|
|
model = pandas.DataFrame({
|
|
'open': linregress(ohlc['time'].astype(float), ohlc['open'].astype(float)),
|
|
'high': linregress(ohlc['time'].astype(float), ohlc['high'].astype(float)),
|
|
'low': linregress(ohlc['time'].astype(float), ohlc['low'].astype(float)),
|
|
'close': linregress(ohlc['time'].astype(float), ohlc['close'].astype(float))
|
|
}).transpose()
|
|
model.columns = ['slope', 'intercept', 'r_value', 'p_value', 'stderr']
|
|
return model
|
|
|
|
def grade_models(models):
|
|
model_averages = pandas.DataFrame({
|
|
model: {
|
|
'slope': models[model].slope.mean(),
|
|
'intercept': models[model].intercept.mean(),
|
|
'r_value': models[model].r_value.mean(),
|
|
'p_value': models[model].p_value.mean(),
|
|
'stderr': models[model].stderr.mean()
|
|
} for model in models
|
|
}).transpose()
|
|
filtered_models = model_averages[
|
|
(model_averages['r_value'] >= R_VALUE_TARGET) &
|
|
(model_averages['slope'] > 0)
|
|
]
|
|
return filtered_models.sort_values(by='slope', ascending=False)
|
|
|
|
def sell_asset(market, volume):
|
|
kraken.add_standard_order(
|
|
ordertype = 'market',
|
|
type = 'sell',
|
|
pair = market,
|
|
volume = volume,
|
|
validate = False
|
|
)
|
|
sleep(kraken.factor)
|
|
print(f"Sold {volume} on {market}")
|
|
|
|
def buy_asset(market, volume):
|
|
kraken.add_standard_order(
|
|
ordertype = 'market',
|
|
type = 'buy',
|
|
pair = market,
|
|
volume = volume,
|
|
validate = False
|
|
)
|
|
print(f"Purchased {volume} on {market}")
|
|
sleep(kraken.factor)
|
|
|
|
def main():
|
|
asset_pairs = filter_asset_pairs(kraken.get_tradable_asset_pairs())
|
|
sleep(kraken.factor)
|
|
graded_models = grade_models(
|
|
{asset_pair: model_asset_pair(asset_pair) for asset_pair in asset_pairs.index}
|
|
)
|
|
|
|
account_balance = kraken.get_account_balance()
|
|
for asset in account_balance.index.drop('ZUSD'):
|
|
market = asset_pairs[asset_pairs['base'] == asset]
|
|
if market.index[0] not in graded_models[:5].index:
|
|
sell_asset(market, account_balance.loc[asset].vol)
|
|
|
|
while len(account_balance := kraken.get_account_balance()) - 1 < INVESTMENT_COUNT:
|
|
if account_balance.loc['ZUSD'].vol <= INVESTMENT_VOLUME:
|
|
break
|
|
for market in graded_models.index:
|
|
if asset_pairs.loc[market].base not in account_balance.index:
|
|
ohlc, last = kraken.get_ohlc_data(market)
|
|
volume = INVESTMENT_VOLUME/ohlc.iloc[-1].close
|
|
buy_asset(market, volume)
|
|
break
|
|
|
|
if __name__ == '__main__':
|
|
main()
|