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)
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)
cluster = Cluster()
session = cluster.connect('playlist')
meta_str = session.cluster.metadata.export_schema_as_string()
keyspace = session.cluster.metadata.keyspaces["playlist"]
if(len(sys.argv) <= 1):
parser = SchemaParser(keyspace.export_as_string())
parser.parse()
sys.exit("Generated schema.yaml file.")
music_results_file = codecs.open('music_results.csv', encoding='utf-8', mode='w+')
rows = session.execute('SELECT * FROM track_by_id')
writer = csv.writer(music_results_file)
writer.writerow(['track_id', 'artist', 'genre', 'music_file', 'track', 'track_length_in_seconds'])
writer.writerows([(track.track_id, track.artist, track.genre, track.music_file, track.track, track.track_length_in_seconds) for track in rows])
artists_names_results_file = codecs.open('artists_names_results.csv', encoding='utf-8', mode='w+')
rows = session.execute('SELECT * FROM artists_by_first_letter')
writer = csv.writer(artists_names_results_file)
writer.writerow(['first_letter', 'artist'])
writer.writerows([(artistt.first_letter, artistt.artist) for artistt in rows])
cypher_queries_gen = CypherQueriesGenerator(keyspace)
cypher_queries_gen.generate()
cypher_queries_gen.build_queries(["track_by_id", "artists_by_first_letter"], ["music_results.csv", "artists_names_results.csv"])
class Neo4jNodeBuilder(object):
def __init__(self, keyspace, schema):
self.keyspace = keyspace
self.schema = schema