diff --git a/src/database/sql_database.py b/src/database/sql_database.py index 1653314..e4c3461 100644 --- a/src/database/sql_database.py +++ b/src/database/sql_database.py @@ -68,10 +68,8 @@ class SQL_Database: """ Insert current generations population """ # Structure the insert data - db_chromosome_list = [ - (ga.current_generation, chromosome.fitness, '[chromosome]') - for chromosome in ga.population.get_chromosome_list() - ] + db_chromosome_list = [(ga.current_generation, chromosome.fitness, chromosome.__str__()) + for chromosome in ga.population.get_chromosome_list() ] # Create sql query structure sql = ''' INSERT INTO data(generation,fitness,chromosome) diff --git a/src/structure/chromosome.py b/src/structure/chromosome.py index e9106ae..458ed6b 100644 --- a/src/structure/chromosome.py +++ b/src/structure/chromosome.py @@ -59,4 +59,16 @@ class Chromosome: output_str = '' for gene in self.gene_list: output_str += gene.__repr__() + + return output_str + + def __str__(self): + """Create a string of the chromosome. Ex chromsome = 1,2,3""" + output_str = '' + for gene in self.gene_list: + if gene == self.gene_list[0]: + output_str += str(gene.get_value()) + else: + output_str += ',' + str(gene.get_value()) + return output_str