Python FlightsRespDAL.get_all_airports Examples

Example #1

File: random_tests_creator.py Project: natid/cheap-travel

import pickle
import datetime
from db.flights_resp import FlightsRespDAL
from random import randint
import pickle

TESTS_TO_CREATE = 10000

flights_resp_dal = FlightsRespDAL()

airports = flights_resp_dal.get_all_airports()

tests = []
for index in xrange(0, TESTS_TO_CREATE):
    origin = airports[randint(0, len(airports)-1)]
    dest = airports[randint(0, len(airports)-1)]
    departure_date = datetime.date(2014, 8, 22) + datetime.timedelta(days=randint(0,130))
    return_date = departure_date+ datetime.timedelta(days=randint(5,40))

    tests.append((origin, dest, departure_date, return_date))


with open("tests.info", "w") as tests_file:
    for test in tests:
        pickle.dump(test,tests_file)