distance_getter_factory.py

from tsp.gps.gpsdistancegetter import GpsDistanceGetter
from tsp.osrm.osrmdistancegetter import OsrmDistanceGetter

class DistanceGetterFactory(object):

    def __init__(self, is_test_mode=False):

        self.is_test_mode = is_test_mode

    def get_instance(self):
        
        if self.is_test_mode:
            return GpsDistanceGetter('gps')
        else:
            return OsrmDistanceGetter('osrm')