def __init__(self, settings): # instantiates Engine() class # and DatabaseOps() class, if writing to CSV (name convention seems counter intuitive) self.settings = settings self.engine = Engine(settings) self.write_to_db = settings.WRITE_TO_DB if settings.WRITE_TO_DB: self.db = Database() else: self.db = None self.new_file = open('%s.csv' % (self.engine.tablename), 'w') dbops = DatabaseOps() dbops.import_classes(self.engine.sql_cols, self.engine.sql_declare) self.new_file.write(dbops.build_csv('|', True))
def process_result(result): #s = Settings() #mostly a static variable class args = get_result_list(KeyArgs.record, result) if args != None: data = KeyArgs.engine.init_cls(args) sql_cols = KeyArgs.engine.init_cls(KeyArgs.engine.sql_cols_list) sql = DatabaseOps() sql.import_classes(sql_cols, data, self.settings.INCLUDE_IDS) if KeyArgs.db != None: KeyArgs.db.query(sql.build_insert()) else: line = sql.build_csv() print line KeyArgs.new_file.write(line) self.counter()