def bootstrap():
try:
internal_database.connect()
internal_database.create_tables([
TickerModel, IndexInfoModel, IndexedCoinModel,
SupportedCoinModel, CoinBalanceModel, RebalanceTickModel,
RealizedGainModel, CoinLockModel
])
IndexInfoModel.create(Active=False,
CoinCount=0,
TotalBTCVal=0.0,
TotalUSDVal=0.0,
TotalRealizedGain=0.0,
TotalUnrealizedGain=0.0,
BalanceThreshold=25.0,
RebalanceTickSetting=1,
OrderTimeout=1,
OrderRetryAmount=3)
RebalanceTickModel.create(TickCount=0)
DatabaseManager.create_index_coin_model("BTC", float(100.0), 0.0,
0.0, False)
DatabaseManager.create_realized_gain_model("BTC", 0.0)
except Exception as e:
#print e
pass
def create_rebalance_tick_model(tickCount):
with internal_database.execution_context():
try:
RebalanceTickModel.create(TickCount=tickCount)
return True
except IntegrityError:
return False
except Exception as e:
logger.exception(e)
def get_rebalance_tick_model():
try:
return RebalanceTickModel.get(id=1)
except Exception as e:
# Model dosen't exist
#logger.exception(e)
return None
def get_rebalance_tick_model():
with internal_database.execution_context():
try:
return RebalanceTickModel.get(id=1)
except Exception as e:
# Model dosen't exist
#logger.exception(e)
return None
def update_rebalance_tick_model(tickCount):
try:
rebalanceTick = RebalanceTickModel.get(id=1)
rebalanceTick.TickCount = tickCount
rebalanceTick.save()
return True
except IntegrityError:
return False
except Exception as e:
logger.exception(e)
def update_rebalance_tick_model(tickCount):
with internal_database.execution_context():
try:
rebalanceTick = RebalanceTickModel.get(id=1)
rebalanceTick.TickCount = tickCount
rebalanceTick.save()
return True
except IntegrityError:
return False
except Exception as e:
logger.exception(e)