def create_busline_terminal_relation(self): """This method is used for read in parser file the fields referring a terminals and define the relationship between terminal and busline. """ csv_file = self.read_file('importer/data/bus_lines.csv') print('Creating Busline-Terminal relation...') for row in csv_file: bus_line = Busline() try: bus_line = Busline.objects.get(line_number=row[0]) except ObjectDoesNotExist: print('Busline', row[0], 'does not exist.') try: terminal1 = Terminal.objects.get(description=row[5]) if terminal1 not in bus_line.terminals.all(): bus_line.terminals.add(terminal1) bus_line.save() except ObjectDoesNotExist: print('Error for Busline ', row[0]) print('Terminal', row[5], 'does not exist.') try: terminal2 = Terminal.objects.get(description=row[6]) if terminal2 not in bus_line.terminals.all(): bus_line.terminals.add(terminal2) bus_line.save() except ObjectDoesNotExist: print('Error for Busline ', row[0]) print('Terminal', row[6], 'does not exist.')
def import_bus_lines(self): """This method is used for save the objects created in database.""" csv_file = self.read_file('importer/data/bus_lines.csv') print('Importing Buslines...') for row in csv_file: try: bus_line = Busline() bus_line.line_number = row[0] bus_line.description = row[1] bus_line.fee = row[2] bus_line.route_size = row[3] bus_line.company = Company.objects.get(name=row[4]) bus_line.via = row[7] bus_line.save() except ObjectDoesNotExist: print('Busline', row[0], 'has incomplete data.')