def test_read_save(self): """ Open TA file data then save all fields into db """ for key, ta_file in enumerate(self.ta_files, start=1): print '%d. run filename: %s' % (key, ta_file) print 'starting...\n' date = os.path.basename(ta_file)[0:10] data = open(ta_file).read() statement = models.Statement( date=date, account_statement='None', position_statement='None', trade_activity=data, ) statement.save() trade_activity = models.TradeSummary(statement=statement, date=date) trade_activity.save() print 'statement id: %d' % statement.id print 'trade_activity id: %d' % trade_activity.id print 'using open_ta to make dict data...' ta_data = OpenTA(data=data).read() print 'keys: %s\n' % ta_data.keys() self.insert_db(no=key, trade_activity=trade_activity, test_model=models.FilledOrder, data_list=ta_data['filled_order']) self.insert_db(no=key, trade_activity=trade_activity, test_model=models.CancelledOrder, data_list=ta_data['cancelled_order']) self.insert_db(no=key, trade_activity=trade_activity, test_model=models.WorkingOrder, data_list=ta_data['working_order']) self.insert_db(no=key, trade_activity=trade_activity, test_model=models.RollingStrategy, data_list=ta_data['rolling_strategy']) print '\n' + '-' * 100
def test_read_save(self): """ Open TA file data then save all fields into db """ for key, ta_file in enumerate(self.ta_files, start=1): print '%d. run filename: %s' % (key, ta_file) print 'starting...\n' date = os.path.basename(ta_file)[0:10] data = open(ta_file).read() statement = models.Statement( date=date, account_statement='None', position_statement='None', trade_activity=data, ) statement.save() trade_activity = models.TradeSummary( statement=statement, date=date ) trade_activity.save() print 'statement id: %d' % statement.id print 'trade_activity id: %d' % trade_activity.id print 'using open_ta to make dict data...' ta_data = OpenTA(data=data).read() print 'keys: %s\n' % ta_data.keys() self.insert_db( no=key, trade_activity=trade_activity, test_model=models.FilledOrder, data_list=ta_data['filled_order'] ) self.insert_db( no=key, trade_activity=trade_activity, test_model=models.CancelledOrder, data_list=ta_data['cancelled_order'] ) self.insert_db( no=key, trade_activity=trade_activity, test_model=models.WorkingOrder, data_list=ta_data['working_order'] ) self.insert_db( no=key, trade_activity=trade_activity, test_model=models.RollingStrategy, data_list=ta_data['rolling_strategy'] ) print '\n' + '-' * 100