示例#1
0
    def export(self):
        tableNames = self.getTables()
        fileNames = [t + "_results.csv" for t in tableNames]

        for t in tableNames:
            results_file = codecs.open(t + "_results.csv",
                                       encoding='utf-8',
                                       mode='w+')
            rows = self.session.execute('SELECT * FROM ' + t)
            writer = csv.writer(results_file)
            writer.writerow(self.getColumnsForTable(t))
            writer.writerows([(e for e in row) for row in rows])

        cypher_queries_gen = CypherQueriesGenerator(self.keyspace_metadata,
                                                    schema_file)
        cypher_queries_gen.generate()
        cypher_queries_gen.build_queries(tableNames, fileNames)