createDistribution.py

# -*- coding: utf-8 -*-


import os, shutil

def copyFiles(sourceDir,targetDir,includes=None):
    
    if not os.path.exists(targetDir): os.makedirs(targetDir)
    
    
    for  f in os.listdir(sourceDir):
        
        base,ext = os.path.splitext(f)
        
        if (ext =='.py') and (True if not includes else (base in includes)): 
            s = os.path.join(sourceDir, f)
            d = os.path.join(targetDir, f)
            print s, '->', d
            shutil.copyfile(s,d)


sourceDir = 'lib'
targetDir = 'dist\\tradingWithPython\\lib'



includes = ['__init__','cboe','csvDatabase','functions','yahooFinance','extra','bats','backtest']

print '-----------init file---------'
shutil.copyfile('__init__.py','dist\\tradingWithPython\\__init__.py')
print '-----------lib files---------'
copyFiles(sourceDir,targetDir,includes)
print '-----------IB files----------'
copyFiles(sourceDir+'\\interactiveBrokers',targetDir+'\\interactiveBrokers')