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()
