diff --git a/scrub_mtl_class.py b/scrub_mtl_class.py index 39120f8..1494a84 100644 --- a/scrub_mtl_class.py +++ b/scrub_mtl_class.py @@ -3,12 +3,25 @@ from basic_functions import * class ScrubMTL: - def __init__(self): - pass - - def define_input_paths(self): - """Can be a part of the constructor""" - pass + def __init__(self, nrcan, geoindex, property_assessment, + montreal_boundary, output_paths_dir): + self.nrcan = nrcan + self.geoindex = geoindex + self.property_assessment = property_assessment + self.montreal_boundary = montreal_boundary + self.output_paths_dir = output_paths_dir + self.output_paths = { + 'Fixed NRCan': '', + 'Fixed GeoIndex': '', + 'Clipped Fixed GeoIndex': '', + 'Splitted NRCan': '', + 'Pairwise Clipped Property Assessment Partitions': '', + 'Pairwise Clipped Merged Property Assessment': '', + 'Property Assessment and NRCan': '', + 'Property Assessment and NRCan and GeoIndex': '', + 'Deleted Duplicates Layer': '', + 'Singled Parts Layer': '' + } @staticmethod def merge_layers(layers_path, mered_layer_path): @@ -22,7 +35,11 @@ class ScrubMTL: processing.run("native:mergevectorlayers", params) def generate_output_paths(self): - pass + for path in self.output_paths.keys(): + new_folder = path.lower().replace(' ', '_') + output_path = self.output_paths_dir + '/' + new_folder + os.mkdir(output_path) + self.output_paths[path] = output_path + f'/{new_folder}.shp' def initialize_layer(self): pass