def test_location_5(self): location = Routes.get_id(52) p = {c.name: getattr(location, c.name) for c in location.__table__.columns} self.assertEqual(p["ID"], 52) self.assertEqual(p["ROUTE_NAME"], "Rock Tunnel") self.assertEqual(p["ROUTE_REGION"], "Kanto") self.assertEqual(p["ROUTE_WEST_EXIT"], "Kanto Route 7 ")
def test_location_4(self): location = Routes.get_id(26) p = {c.name: getattr(location, c.name) for c in location.__table__.columns} self.assertEqual(p["ID"], 26) self.assertEqual(p["ROUTE_NAME"], "Kanto Route 15") self.assertEqual(p["ROUTE_REGION"], "Kanto") self.assertEqual(p["ROUTE_WEST_EXIT"], "Fuchsia City ")
def test_location_6(self): location = Routes.get_id(279) p = {c.name: getattr(location, c.name) for c in location.__table__.columns} self.assertEqual(p["ID"], 279) self.assertEqual(p["ROUTE_NAME"], "Sinnoh Route 224") self.assertEqual(p["ROUTE_REGION"], "Sinnoh") self.assertEqual(p["ROUTE_NORTH_EXIT"], "Seabreak Path ") self.assertEqual(p["ROUTE_WEST_EXIT"], "Victory Road (Sinnoh) ")
def test_location_3(self): locations = Routes.get_all() p = [] for route in locations: p.append({c.name: getattr(route, c.name) for c in route.__table__.columns}) self.assertEqual(p[0]["ID"], 1) self.assertEqual(p[0]["ROUTE_NAME"], "Pallet Town") self.assertEqual(p[0]["ROUTE_REGION"], "Kanto") self.assertEqual(p[1]["ID"], 2) self.assertEqual(p[1]["ROUTE_NAME"], "Viridian City") self.assertEqual(p[1]["ROUTE_REGION"], "Kanto")
def test_location_2(self): location = Routes.get_id(1) p = {c.name: getattr(location, c.name) for c in location.__table__.columns} self.assertEqual(p["ID"], 1) self.assertEqual(p["ROUTE_NAME"], "Pallet Town") self.assertEqual(p["ROUTE_REGION"], "Kanto")
import sys import csv import json sys.path.append("..") from functions import generate_random_string from app import app from models import db, Stops, Links, Edges, Routes with app.app_context(): #ADD ROUTES TO DATABASE with open("routes.json", "r") as file_to_read: data = json.load(file_to_read) for row in data['d']['results']: new_route = Routes(row['RouteId'], row['LongName'], row['RouteName'], row['RouteType']) db.session.add(new_route) db.session.commit() #ADD STOPS TO DATABASE with open("stops.csv", "r") as file_to_read: csv_to_read = csv.reader(file_to_read) for row in csv_to_read: current_stop = Stops.query.filter_by(id=row[1]).first() if current_stop != None: continue new_stop = Stops(row[1], row[0], row[3], row[2]) db.session.add(new_stop) db.session.commit() #ADD LINKS TO DATABASE
journeys = response['journeys'] for j in range(3): print "Journey: %s %s" % (j+1, journeys[j]) print "these" origCode = journeys[j]['flights'][0]['departure'] print "are" destCode = journeys[j]['flights'][0]['arrival'] print "the" orig = getCity(origCode) print 'orig ' + orig print "times" dest = getCity(destCode) print "that" deptTime = journeys[j]['flights'][0]['departureDate'] print "try" arrivTime = journeys[j]['flights'][0]['arrivalDate'] print "men" airline = journeys[j]['flights'][0]['carrier'] print "souls" price = journeys[j]['price'] print "clara" now = 'lalalala' print "is" duration = (float(journeys[j]['flights'][0]['miles'])/500) * 60 print "cool" d = Routes(origin=orig,destination=dest,origin_address=deptCode,origin_time=deptTime,destination_address=arrivalCode,destination_time=arrivTime,brand=airline,types='plane',cost=price,time=duration, current_time=now) print "catz" d.save()