142 lines
4.4 KiB
Python
142 lines
4.4 KiB
Python
import sys
|
|
import requests
|
|
import fin_depo
|
|
from . import secrets
|
|
from fin_defs import CryptoCurrency, AssetAmount, MPC, Asset, USDT
|
|
from decimal import Decimal
|
|
from collections import deque
|
|
from fin_depo.data import *
|
|
import datetime
|
|
import dataclasses
|
|
|
|
KUCOIN_CLIENT = fin_depo.defi_kucoin.KucoinDepoFetcher(
|
|
secrets.KUCOIN_KEY,
|
|
secrets.KUCOIN_SECRET,
|
|
secrets.KUCOIN_PASS,
|
|
)
|
|
|
|
KRAKEN_CLIENT = fin_depo.defi_kraken.KrakenDepoFetcher(
|
|
secrets.KRAKEN_KEY,
|
|
secrets.KRAKEN_SECRET,
|
|
)
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class BoughtAndSold:
|
|
amount: AssetAmount
|
|
time_bought: datetime.datetime
|
|
time_sold: datetime.datetime
|
|
|
|
@dataclasses.dataclass
|
|
class BoughtAndNotYetSold:
|
|
amount: AssetAmount
|
|
time_bought: datetime.datetime
|
|
|
|
|
|
def compute_tax(transfers: list):
|
|
transfers.sort(key=lambda t:t.executed_time)
|
|
|
|
|
|
bought_and_sold_for: list[BoughtAndSold] = []
|
|
bought_and_spent_for: list[BoughtAndSold] = []
|
|
|
|
prices_bought_for: dict[Asset, deque[BoughtAndNotYetSold]] = {}
|
|
if False:
|
|
prices_bought_for[MPC] = deque()
|
|
prices_bought_for[MPC].append(BoughtAndNotYetSold(AssetAmount(MPC, Decimal(100)), datetime.datetime(2000, 1,1,1,1,1,1)))
|
|
prices_bought_for[USDT] = deque()
|
|
prices_bought_for[USDT].append(BoughtAndNotYetSold(AssetAmount(USDT, Decimal(100)), datetime.datetime(2000, 1,1,1,1,1,1)))
|
|
|
|
def sell(amount: AssetAmount, executed_time: datetime.datetime, spent=False):
|
|
print(f'Selling: {amount}')
|
|
while amount.amount > 0:
|
|
if len(prices_bought_for[amount.asset]) == 0:
|
|
raise Exception('Miscalculation: ' + str(amount))
|
|
partial = prices_bought_for[amount.asset].popleft()
|
|
|
|
print(f'Fuck : {amount} - {partial.amount}')
|
|
|
|
amount_covered_by_partial = min(partial.amount, amount)
|
|
amount -= amount_covered_by_partial
|
|
new_partial_amount = partial.amount - amount_covered_by_partial
|
|
|
|
print(f' => {amount} ({new_partial_amount}')
|
|
|
|
# Re-add partial
|
|
if new_partial_amount.amount != 0:
|
|
new_partial = BoughtAndNotYetSold(new_partial_amount, partial.time_bought)
|
|
print(f' => new partial: {new_partial}')
|
|
prices_bought_for[amount.asset].appendleft(new_partial)
|
|
del new_partial
|
|
|
|
# Add FIFO like
|
|
if spent:
|
|
bought_and_spent_for.append(BoughtAndSold(amount_covered_by_partial, partial.time_bought, executed_time))
|
|
else:
|
|
bought_and_sold_for.append(BoughtAndSold(amount_covered_by_partial, partial.time_bought, executed_time))
|
|
|
|
del partial, amount_covered_by_partial
|
|
|
|
def spend(amount: AssetAmount, executed_time: datetime.datetime):
|
|
print(f'Spending: {amount}')
|
|
sell(amount, executed_time, spent=True)
|
|
|
|
for transfer in transfers:
|
|
print(transfer)
|
|
# Buy
|
|
if output := transfer.output:
|
|
print('Buy')
|
|
if output.asset not in prices_bought_for:
|
|
prices_bought_for[output.asset] = deque()
|
|
prices_bought_for[output.asset].append(BoughtAndNotYetSold(output, transfer.executed_time))
|
|
del output
|
|
|
|
# Sell
|
|
if _input := transfer.input:
|
|
print('Sell')
|
|
sell(_input, transfer.executed_time)
|
|
del _input
|
|
|
|
if fee := transfer.fee:
|
|
print('Sell')
|
|
spend(fee, transfer.executed_time)
|
|
del transfer, fee
|
|
|
|
if True:
|
|
print('Bought for:')
|
|
for asset, prices in prices_bought_for.items():
|
|
price_sum = sum((p.amount for p in prices), start=AssetAmount.ZERO)
|
|
sys.stdout.write(f' - {asset.raw_short_name():10} ({price_sum}): ')
|
|
for price in prices:
|
|
sys.stdout.write(str(price.amount.amount))
|
|
sys.stdout.write(', ')
|
|
del price
|
|
sys.stdout.write('\n')
|
|
del asset, prices
|
|
|
|
|
|
print('-'*80)
|
|
|
|
|
|
for bas in bought_and_sold_for:
|
|
print(f'{bas.amount} ({bas.time_bought} ----> {bas.time_sold})')
|
|
|
|
|
|
for c, l in prices_bought_for.items():
|
|
print(c)
|
|
print(l)
|
|
|
|
def main():
|
|
"""Main function."""
|
|
|
|
TRANSFERS = list(KRAKEN_CLIENT._get_double_registers())
|
|
for k in TRANSFERS:
|
|
print(k)
|
|
#TRANSFERS += list(KUCOIN_CLIENT._get_double_registers())
|
|
compute_tax(TRANSFERS)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|