def setScript(args):
script_name = args.script_name
if not script_name:
msg = '-s argument is required. Set -s argument to use script. (e.g. [-s st02-045] or [-s script/struts/st02-045.py])'
sys.exit(logger.error(msg))
if script_name.endswith('.py'):
if paths.SCRIPT_PATH in script_name:
script_path = os.path.join(paths.SCRIPT_PATH, script_name)
checkFile(script_path)
conf.SCRIPT_NAME = script_path
else:
checkFile(script_name)
conf.SCRIPT_NAME = script_name
else:
if not os.path.splitext(script_name)[-1]:
script_path = os.path.join(paths.SCRIPT_PATH, script_name + '.py')
checkFile(script_path)
conf.SCRIPT_NAME = script_path
else:
checkFile(script_name)
conf.SCRIPT_NAME = script_name
msg = 'Use script: %s' % conf.SCRIPT_NAME # os.path.split(conf.SCRIPT_NAME)[-1].rstrip('.py')
logger.success(msg)
def loadNetwork(self, network, cost=None, cutoff=None, numToFind=None, searchDist=None, cutoffFld=None, numToFindFld=None, mappings=[]):
common.progress('creating routing layer')
if not numToFind:
numToFind = common.count(self.places)
self.naLayer = self.makeNALayer(common.checkFile(network), self.NA_LAY, cost, cutoff, numToFind)
self.cost = cost
self.placeMapper.addSilentField(cost, float)
common.progress('calculating places\' network locations')
self.calculateLocations(network, self.places, searchDist)
common.progress('loading places to network')
# create mappings
toMappingList = common.NET_FIELDS + [(self.NAME_MAP, self.placesIDField, None)]
for item in mappings:
toMappingList.append(item + [None])
fromMappingList = toMappingList[:]
if cutoffFld:
fromMappingList.append((self.CUTOFF_PREFIX + cost, cutoffFld, None))
if numToFindFld:
fromMappingList.append((self.NUM_TO_FIND_HEADER, numToFindFld, None))
# load locations
arcpy.AddLocations_na(self.NA_LAY, self.OD_SUBLAYERS[0], self.places, self.networkMappings(fromMappingList), '', append='clear')
arcpy.AddLocations_na(self.NA_LAY, self.OD_SUBLAYERS[1], self.places, self.networkMappings(toMappingList), '', append='clear')
self.routeSublayer = common.sublayer(self.naLayer, self.OUTPUT_SUBLAYER)
self.linkMapper = conversion.LinkFieldMapper(self.routeSublayer)
try:
self.linkMapper.addMapField(self.OUT_COST_PREFIX + cost, cost)
except conversion.FieldError:
raise conversion.FieldError, 'cost attribute %s not found in network dataset' % cost
def __init__(self, zones, zoneIDFld, method, location, **kwargs):
zones = common.toFeatureClass(common.checkFile(zones))
if not location:
location = common.location(zones)
Converter.__init__(self, location, **kwargs)
self.zones = zones
self.zoneMapper = ODFieldMapper(self.zones)
self.zoneMapper.addIDField(zoneIDFld, [common.NEIGH_FROM_FLD, common.NEIGH_TO_FLD], transfer=True)
self.method = method
self.tmpTable = self.tablePath(self.TABLE_FILE_NAME)
self.swmFile = self.folderPath(self.SWM_FILE_NAME)
common.overwrite(True)
def loadNetwork(self, network, cost, searchDist, mappings=[]):
common.progress('calculating places\' network locations')
network = common.checkFile(network)
self.calculateLocations(network, self.places, searchDist)
self.mappings = self.networkMappings(common.NET_FIELDS + mappings)
if self.speedup:
self.performSpeedup(network, cost)
common.progress('preparing routing layer')
self.naLayer = self.makeRouteLayer(network, self.ROUTE_LAY, cost)
self.routeSublayer = common.sublayer(self.naLayer, 'Routes')
# common.debug(common.fieldList(self.routeSublayer))
self.cost = cost
self.placeMapper.addSilentField(self.cost, float)
def loadNetwork(self, network, cost=None, searchDist=None, breaks=[1]):
common.progress('creating routing layer')
# if not numToFind:
# numToFind = common.count(self.places)
self.naLayer = self.makeNALayer(common.checkFile(network), self.NA_LAY, cost, breaks)
common.progress('calculating places\' network locations')
self.calculateLocations(network, self.places, searchDist)
common.progress('loading places to network')
# create mappings
toMappingList = common.NET_FIELDS + [(self.NAME_MAP, self.placesIDField, None)]
for item in mappings:
toMappingList.append(item + [None])
# fromMappingList = toMappingList[:]
# if cutoffFld:
# fromMappingList.append((self.CUTOFF_PREFIX + cost, cutoffFld, None))
# if numToFindFld:
# fromMappingList.append((self.NUM_TO_FIND_HEADER, numToFindFld, None))
# load locations
arcpy.AddLocations_na(self.NA_LAY, self.OD_SUBLAYER, self.places, self.networkMappings(toMappingList), '', append='clear')
self.outputSublayer = common.sublayer(self.naLayer, self.OUTPUT_SUBLAYER)
def __init__(self, polygons, inside=False, idFld=None, mapping=None, **kwargs): Converter.__init__(self, **kwargs) self.polygons = common.checkFile(polygons) self.inside = inside
def __init__(self, location=None, **kwargs):
common.progress('initializing paths')
if location:
self.location = common.checkFile(location)
def _file_target(target):
checkFile(filename_target)
targets.file_target = target
targets.handle_mode = targets.FILE_MODE
def __init__(self, table, odIDFields, places, placeIDField, location, **kwargs): PlaceLinker.__init__(self, location, places, placeIDField, **kwargs) self.table = common.checkFile(table) self.linkMapper = conversion.LinkFieldMapper(self.table) self.placeMapper.setJoinIDFields(odIDFields)
def __init__(self, table, odIDFields, places, placeIDField, **kwargs): PlaceLinker.__init__(self, common.location(table), places, placeIDField, **kwargs) self.table = common.checkFile(table) self.placeMapper.setJoinIDFields(odIDFields)
def __init__(self, location, places, placesIDField, **kwargs): conversion.Converter.__init__(self, location, **kwargs) self.places = common.checkFile(places) self.placesIDField = placesIDField self.placeMapper = self.MAPPER_CLASS(self.places) self.placeMapper.addIDField(placesIDField)
