def test_read_agency(self): agency = Agency('FDHUKSDI', None) db.session.add(agency) db.session.commit() agency_name = read_agency(agency.id)['name'] db.session.delete(agency) db.session.commit() self.assertEqual(agency_name, 'FDHUKSDI')
def __init__(self): self.database = None self.source_folder = None self.agency = Agency() self.trips = None self.num_routes = None self.routes = {} self.stops = {} self.calendar_dates = {} self.available_files = {} self.shapes = {} self.schedule_exceptions = None
def create_agency(name, parent_id): ''' This function creates a new tupple in Agency table. return 0 means everything accomplished successfully. return 1 means the operation was unsuccessfull. ''' try: agency = Agency(name, parent_id) db.session.add(agency) db.session.commit() return 0 except Exception as e: print('error while creating the agency: ' + str(e)) return 1
from porc import Client import utils, time from agency import Agency client = Client("your api key") agency = Agency("rutgers") relevantTags = ['a', 'b', 'f', 'lx'] relevantRoutes = [] for relevantTag in relevantTags: route = agency.getRoute(relevantTag) if route != None: relevantRoutes.append(route) while True: print("start") predictions = [] for route in relevantRoutes: predictions = predictions + route.getPredictions() # asynchronously post items with client. async () as c: futures = [ c.post('predictions', prediction) for prediction in predictions ] responses = [future.result() for future in futures] [response.raise_for_status() for response in responses] print("end") time.sleep(10)
threadList = [] clientList = [] companyList = [] agencyList = [] #Flight Companies latam = Company('LATAM') ryanair = Company('Ryanair') airFrance = Company('AirFrance') panAm = Company('PanAm') companyList = [latam, ryanair, airFrance, panAm] #Travel Agencies decolar = Agency(companyList, 'Decolar.Com') skyScanner = Agency(companyList, 'Sky Scanner') agencyList = [decolar, skyScanner] for agency in agencyList: agency.run() sleep(2) # Creating Clients for i in range(10): clientList.append(Client(agencyList)) threadList.append(Thread(target=clientList[i].run).start())