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 create_busline_terminal_relation(self):
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.')
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.')
def import_bus_lines(self):
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.')
except IntegrityError:
print('Busline', row[0], 'already registered.')
