from quantopian.algorithm import attach_pipeline, pipeline_output
from quantopian.pipeline import Pipeline
from quantopian.pipeline.data.builtin import USEquityPricing
from quantopian.pipeline.factors import AverageDollarVolume
 
def initialize(context):
    set_do_not_order_list(security_lists.leveraged_etf_list)
    # Rebalance every day, 1 hour after market open.
    schedule_function(my_rebalance, date_rules.every_day(), time_rules.market_open(hours=1))
    # Create our dynamic stock selector.
    attach_pipeline(make_pipeline(), 'my_pipeline')
         
def make_pipeline():
    
     
    # Create a dollar volume factor.
    dollar_volume = AverageDollarVolume(window_length=1)
 
    # Pick the top 1% of stocks ranked by dollar volume.
    high_dollar_volume = dollar_volume.percentile_between(99, 100)
     
    pipe = Pipeline(
        screen = high_dollar_volume,
        columns = {
            'dollar_volume': dollar_volume
        }
    )
    return pipe
 
def before_trading_start(context, data):
    """
    Called every day before market open.
    """
    context.output = pipeline_output('my_pipeline').sort('dollar_volume')
  
    # These are the securities that we are interested in trading each day.
    context.secs = context.output.index
     
def my_assign_weights(context, data):
    """
    Assign weights to securities that we want to order.
    """
    pass
 
def my_rebalance(context,data):
    cantrade = []
    for sec in context.secs[-20:]:
        if data.can_trade(sec) and (sec not in security_lists.leveraged_etf_list):
            cantrade.append(sec)
    for sec in context.portfolio.positions:
        if sec not in cantrade:
            order_target_percent(sec, 0)
    for sec in cantrade:
        order_target_percent(sec, 1.0/len(cantrade))
    print 1
    """
    Execute orders according to our schedule_function() timing. 
    """
    pass