class QAlgorithm1(AlgorithmBase):
def __init__(self, log):
super().__init__(log)
self.inverse = None
def start(self, broker, capital, interval, loss_cut, fee,
minimum_transaction_amount):
self.inverse = AlgorithmItem('252670')
self.add_item(self.inverse)
self.initialize(broker, capital, interval, loss_cut, fee,
minimum_transaction_amount)
self.shift_interval = loss_cut
# Open Orders cancellation
self.clear_open_orders()
# Charting & Monitoring
broker.go_chart(self.inverse.item_code)
broker.demand_monitoring_items_info(self.inverse)
self.is_running = True
self.post('STARTED')
def update_transaction_info(self, item):
# First time work
if not self.start_price:
self.start_time_text = datetime.now().strftime('%H:%M')
self.start_time = self.to_min_count(self.start_time_text)
self.start_price = item.current_price
reference_price = item.current_price + self.loss_cut
self.set_reference(reference_price)
self.inverse.buy(self.buy_limit,
self.capital // item.current_price)
return
# Block during situation processing
if self.shift_in_progress:
return
# Trade according to current price
if item.current_price > self.reference_price:
self.post('Situation 1')
self.shift_in_progress = True
self.inverse.cancel_purchases()
self.inverse.clear_sales()
self.inverse.init_purchase()
self.inverse.init_sale()
self.shift_reference_up()
elif item.current_price < self.buy_limit:
self.post('Situation 4')
self.shift_in_progress = True
self.inverse.cancel_sales()
self.inverse.clear_purchases()
self.inverse.init_purchase()
self.inverse.init_sale()
self.shift_reference_down()
def update_execution_info(self, order):
# Inverse update execution
self.inverse.update_execution_info(order)
# Order processing
if order.order_position in (PURCHASE, CORRECT_PURCHASE):
self.update_episode_purchase(order)
if order.executed_amount:
self.inverse.sell(self.reference_price, order.executed_amount)
elif order.order_position in (SELL, CORRECT_SELL):
self.update_episode_sale(order)
if order.executed_amount:
self.inverse.buy(self.buy_limit, order.executed_amount)
self.total_profit += order.profit
self.total_fee += order.transaction_fee
self.net_profit += order.net_profit
elif order.order_position == CANCEL_PURCHASE and order.order_state == CONFIRMED:
if not self.inverse.purchases:
order_amount = self.capital // order.current_price - self.inverse.holding_amount
self.inverse.buy(self.buy_limit, order_amount)
self.shift_in_progress = False
elif order.order_position == CANCEL_SELL and order.order_state == CONFIRMED:
if not self.inverse.sales:
self.inverse.sell(self.reference_price,
self.inverse.holding_amount)
self.shift_in_progress = False
self.signal('algorithm_update')
self.broker.draw_chart.start()
def update_episode_purchase(self, order):
executed_amount = abs(order.executed_amount)
if self.episode_purchase.order_price != order.order_price:
self.episode_purchase = copy.deepcopy(order)
self.episode_purchase_number = self.get_episode_purchase_number()
self.episode_purchase.episode_number = self.episode_purchase_number
self.episode_purchase.order_amount = 0
self.episode_purchase.open_amount = 0
self.episode_purchase.executed_time = order.executed_time
self.episode_purchase.order_number = order.order_number
self.episode_purchase.order_state = order.order_state
self.episode_purchase.executed_price_avg = order.executed_price_avg
if order.order_state == RECEIPT:
self.episode_purchase.order_amount += order.order_amount
self.episode_purchase.open_amount += order.open_amount
elif order.order_state == ORDER_EXECUTED:
self.episode_purchase.open_amount -= executed_amount
self.episode_purchase.executed_amount_sum += executed_amount
self.orders[self.episode_purchase_number] = self.episode_purchase
def update_episode_sale(self, order):
executed_amount = abs(order.executed_amount)
if self.episode_sale.order_price != order.order_price:
self.episode_sale = copy.deepcopy(order)
self.episode_sale_number = self.get_episode_sale_number()
self.episode_sale.episode_number = self.episode_sale_number
self.episode_sale.order_amount = 0
self.episode_sale.open_amount = 0
self.episode_sale.executed_time = order.executed_time
self.episode_sale.order_number = order.order_number
self.episode_sale.order_state = order.order_state
self.episode_sale.executed_price_avg = order.executed_price_avg
self.episode_sale.profit += order.profit
if order.order_state == RECEIPT:
self.episode_sale.order_amount += order.order_amount
self.episode_sale.open_amount += order.open_amount
elif order.order_state == ORDER_EXECUTED:
self.episode_sale.open_amount -= executed_amount
self.episode_sale.executed_amount_sum += executed_amount
self.orders[self.episode_sale_number] = self.episode_sale
class QuantumJumpAlgorithm1(AlgorithmBase):
def __init__(self, log):
super().__init__(log)
self.log = log
self.reference_price = 0
self.buy_limit = 0
self.loss_limit = 0
self.situation_processing = False
self.inverse = None
self.purchase = Order()
self.sale = Order()
def start(self, broker, capital, interval, loss_cut, fee):
self.inverse = AlgorithmItem('252670')
self.inverse.set_broker(broker)
self.inverse.set_log(self.log)
self.inverse.fee_ratio = fee
self.broker = broker
self.capital = capital
self.interval = interval
self.loss_cut = loss_cut
self.fee = fee
self.is_running = True
# Charting & Monitoring
broker.go_chart(self.inverse.item_code)
broker.demand_monitoring_items_info(self.inverse)
# Open Orders cancellation
open_orders = list(broker.open_orders.values())
for order in open_orders:
self.broker.cancel(order)
self.post('STARTED')
def stop(self):
if not self.is_running:
return
self.post('STOPPED')
# Open Orders cancellation
open_orders = list(self.broker.open_orders.values())
for order in open_orders:
self.broker.cancel(order)
# Init Fields
self.orders.clear()
self.broker = None
self.inverse = None
self.is_running = False
self.capital = 0
self.interval = 0
self.loss_cut = 0
self.fee = 0
self.total_profit = 0
self.net_profit = 0
self.start_time_text = ''
self.start_time = 0
self.start_price = 0
self.reference_price = 0
self.buy_limit = 0
self.loss_limit = 0
self.situation_processing = False
self.previous_situation = ''
self.previous_msg = ()
def resume(self):
self.is_running = True
def halt(self):
self.is_running = False
def set_reference(self, price):
self.reference_price = price
self.buy_limit = self.reference_price - self.interval
self.loss_limit = self.buy_limit - self.loss_cut
self.episode_count += 1
self.purchase_episode = self.get_episode_purchase_number()
self.sale_episode = self.get_episode_sale_number()
def shift_reference_up(self):
self.set_reference(self.reference_price + self.loss_cut)
def shift_reference_down(self):
self.set_reference(self.reference_price - self.loss_cut)
def update_transaction_info(self, item):
# First time work
if not self.start_price:
self.start_time_text = datetime.now().strftime('%H:%M')
self.start_time = self.to_min_count(self.start_time_text)
self.start_price = item.current_price
reference_price = item.current_price + self.loss_cut
self.set_reference(reference_price)
self.inverse.buy(self.buy_limit,
self.capital // item.current_price)
return
# Block during situation processing
if self.situation_processing:
return
# Trade according to current price
if item.current_price > self.reference_price:
self.post('Situation 1')
self.situation_processing = True
self.inverse.cancel_purchases()
self.inverse.clear_sales()
self.inverse.init_purchase()
self.inverse.init_sale()
self.shift_reference_up()
elif item.current_price < self.buy_limit:
self.post('Situation 4')
self.situation_processing = True
self.inverse.cancel_sales()
self.inverse.clear_purchases()
self.inverse.init_purchase()
self.inverse.init_sale()
self.shift_reference_down()
def update_execution_info(self, order):
# Inverse update execution
self.inverse.update_execution_info(order)
# Order processing
executed_amount = abs(order.executed_amount)
if order.order_position in (PURCHASE, CORRECT_PURCHASE):
# Algorithm history update
if self.purchase.order_price != order.order_price:
self.purchase = copy.deepcopy(order)
self.purchase.order_amount = 0
self.purchase.open_amount = 0
self.purchase.executed_time = order.executed_time
self.purchase.episode_number = self.purchase_episode
self.purchase.order_number = order.order_number
self.purchase.order_state = order.order_state
self.purchase.executed_price_avg = order.executed_price_avg
if order.order_state == RECEIPT:
self.purchase.order_amount += order.order_amount
self.purchase.open_amount += order.open_amount
elif order.order_state == ORDER_EXECUTED:
self.purchase.open_amount -= executed_amount
self.purchase.executed_amount_sum += executed_amount
self.orders[self.purchase_episode] = self.purchase
# Order
if order.executed_amount:
self.inverse.sell(self.reference_price, order.executed_amount)
elif order.order_position in (SELL, CORRECT_SELL):
# Algorithm history update
if self.sale.order_price != order.order_price:
self.sale = copy.deepcopy(order)
self.sale.order_amount = 0
self.sale.open_amount = 0
self.sale.executed_time = order.executed_time
self.sale.episode_number = self.sale_episode
self.sale.order_number = order.order_number
self.sale.order_state = order.order_state
self.sale.executed_price_avg = order.executed_price_avg
self.sale.profit += order.profit
if order.order_state == RECEIPT:
self.sale.order_amount += order.order_amount
self.sale.open_amount += order.open_amount
elif order.order_state == ORDER_EXECUTED:
self.sale.open_amount -= executed_amount
self.sale.executed_amount_sum += executed_amount
self.orders[self.sale_episode] = self.sale
# Order
if order.executed_amount:
self.inverse.buy(self.buy_limit, order.executed_amount)
self.total_profit += order.profit
self.total_fee += order.transaction_fee
self.net_profit += order.net_profit
elif order.order_position == CANCEL_PURCHASE and order.order_state == CONFIRMED:
if not self.inverse.purchases:
order_amount = self.capital // order.current_price - self.inverse.holding_amount
self.inverse.buy(self.buy_limit, order_amount)
self.situation_processing = False
elif order.order_position == CANCEL_SELL and order.order_state == CONFIRMED:
if not self.inverse.sales:
self.inverse.sell(self.reference_price,
self.inverse.holding_amount)
self.situation_processing = False
self.signal('algorithm_update')
self.broker.draw_chart.start()
def display_situation(self, current_situation):
if current_situation != self.previous_situation:
self.post(current_situation)
self.previous_situation = current_situation
def post(self, *args):
if args != self.previous_msg:
self.debug('\033[93mALGORITHM', *args, '\033[97m')
self.previous_msg = args
def post(self, *args):
self.debug('\033[93mALGORITHM', *args, '\033[97m')