Skip to content

Commit b6675c8

Browse files
committed
Rename variables and method names according pep8
1 parent 150b44f commit b6675c8

4 files changed

+61
-61
lines changed

nlsgpkgloader/nls_geopackage_loader.py

+46-46
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def add_action(
194194

195195
return action
196196

197-
def initGui(self):
197+
def initGui(self): # noqa N802
198198
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
199199

200200
self.add_action(
@@ -236,23 +236,23 @@ def run(self):
236236
)
237237

238238
if self.nls_user_key == "":
239-
res = self.showSettingsDialog()
239+
res = self.show_settings_dialog()
240240
if not res:
241241
return
242-
if not self.loadLayers():
242+
if not self.load_layers():
243243
QMessageBox.critical(
244244
self.iface.mainWindow(),
245245
self.tr(u"Failed to load data"),
246246
self.tr(u"Check that necessary files exist in data folder"),
247247
)
248248
return
249249

250-
self.product_types = self.downloadNLSProductTypes()
250+
self.product_types = self.download_nls_product_types()
251251

252252
self.municipalities_dialog = NLSGeoPackageLoaderMunicipalitySelectionDialog()
253253

254254
self.municipalities_dialog.settingsPushButton.clicked.connect(
255-
self.showSettingsDialog
255+
self.show_settings_dialog
256256
)
257257
self.municipalities_dialog.fileNameEdit.setValue(self.fileName)
258258
self.municipalities_dialog.loadLayers.setChecked(self.addDownloadedDataAsLayer)
@@ -261,17 +261,17 @@ def run(self):
261261
)
262262
self.municipalities_dialog.loadUtmGrids.setChecked(self.showUTMGridsAsLayer)
263263
self.municipalities_dialog.loadSeaGrids.setChecked(self.showSeatilesAsLayer)
264-
self.municipalities_dialog.loadLayers.stateChanged.connect(self.toggleLayers)
265-
self.municipalities_dialog.loadMunLayer.stateChanged.connect(self.toggleLayers)
266-
self.municipalities_dialog.loadUtmGrids.stateChanged.connect(self.toggleLayers)
267-
self.municipalities_dialog.loadSeaGrids.stateChanged.connect(self.toggleLayers)
268-
self.toggleLayers()
264+
self.municipalities_dialog.loadLayers.stateChanged.connect(self.toggle_layers)
265+
self.municipalities_dialog.loadMunLayer.stateChanged.connect(self.toggle_layers)
266+
self.municipalities_dialog.loadUtmGrids.stateChanged.connect(self.toggle_layers)
267+
self.municipalities_dialog.loadSeaGrids.stateChanged.connect(self.toggle_layers)
268+
self.toggle_layers()
269269

270270
for feature in self.municipality_layer.getFeatures():
271271
item = QListWidgetItem(feature["NAMEFIN"])
272272
self.municipalities_dialog.municipalityListWidget.addItem(item)
273273

274-
for key, value in self.product_types.items():
274+
for value in self.product_types.values():
275275
item = QListWidgetItem(value)
276276
self.municipalities_dialog.productListWidget.addItem(item)
277277
if value in MTK_PRESELECTED_PRODUCTS:
@@ -365,20 +365,20 @@ def run(self):
365365
if len(product_types) > 0 and len(self.selected_geoms) > 0:
366366
QCoreApplication.processEvents()
367367

368-
self.getIntersectingFeatures(
368+
self.get_intersecting_features(
369369
self.municipality_layer.selectedFeatures(),
370370
self.utm25lr_layer,
371371
selected_mun_names,
372372
)
373-
self.getIntersectingFeatures(
373+
self.get_intersecting_features(
374374
self.seatile_layer.selectedFeatures(), self.utm25lr_layer
375375
)
376376
for grid in grids:
377-
self.getIntersectingFeatures(
377+
self.get_intersecting_features(
378378
grid.selectedFeatures(), self.utm25lr_layer
379379
)
380380

381-
self.downloadData(product_types)
381+
self.download_data(product_types)
382382

383383
else:
384384
self.progress_dialog.hide()
@@ -389,7 +389,7 @@ def run(self):
389389
)
390390
return
391391

392-
def toggleLayers(self):
392+
def toggle_layers(self):
393393
"""Load municipality and map tile layers"""
394394
self.addDownloadedDataAsLayer = (
395395
self.municipalities_dialog.loadLayers.isChecked()
@@ -424,7 +424,7 @@ def toggleLayers(self):
424424
found_utm100_layer
425425
) = found_utm200_layer = found_seatiles_layer = found_municipality_layer = False
426426

427-
current_layers = self.getLayers(self.instance.layerTreeRoot())
427+
current_layers = self.get_layers(self.instance.layerTreeRoot())
428428

429429
for current_layer in current_layers:
430430
if current_layer.layer() == self.utm5_layer:
@@ -472,7 +472,7 @@ def toggleLayers(self):
472472
if not found_utm5_layer and self.utm5_layer:
473473
self.instance.addMapLayer(self.utm5_layer)
474474
except:
475-
self.loadLayers()
475+
self.load_layers()
476476
if not found_utm200_layer and self.utm200_layer:
477477
self.instance.addMapLayer(self.utm200_layer)
478478
if not found_utm100_layer and self.utm100_layer:
@@ -492,17 +492,17 @@ def toggleLayers(self):
492492
try:
493493
self.instance.addMapLayer(self.seatile_layer)
494494
except:
495-
self.loadLayers()
495+
self.load_layers()
496496
self.instance.addMapLayer(self.seatile_layer)
497497

498498
if self.showMunicipalitiesAsLayer and not found_municipality_layer:
499499
try:
500500
self.instance.addMapLayer(self.municipality_layer)
501501
except:
502-
self.loadLayers()
502+
self.load_layers()
503503
self.instance.addMapLayer(self.municipality_layer)
504504

505-
def loadLayers(self):
505+
def load_layers(self):
506506
self.municipality_layer = QgsVectorLayer(
507507
resources_path("data", "SuomenKuntajako_2018_10k.shp"),
508508
"municipalities",
@@ -610,7 +610,7 @@ def loadLayers(self):
610610
self.seatile_layer = False
611611

612612
self.instance = QgsProject.instance()
613-
current_layers = self.getLayers(self.instance.layerTreeRoot())
613+
current_layers = self.get_layers(self.instance.layerTreeRoot())
614614

615615
for lnode in current_layers:
616616
if (
@@ -661,16 +661,16 @@ def loadLayers(self):
661661

662662
return True
663663

664-
def getLayers(self, root):
664+
def get_layers(self, root):
665665
layers = []
666666
for node in root.children():
667667
if isinstance(node, QgsLayerTreeGroup):
668-
layers.extend(self.getLayers(node))
668+
layers.extend(self.get_layers(node))
669669
else:
670670
layers.append(node)
671671
return layers
672672

673-
def getIntersectingFeatures(self, features, layer, selected_mun_names=None):
673+
def get_intersecting_features(self, features, layer, selected_mun_names=None):
674674
if selected_mun_names:
675675
expression = ""
676676
for mun in selected_mun_names:
@@ -694,15 +694,15 @@ def getIntersectingFeatures(self, features, layer, selected_mun_names=None):
694694
if feature not in self.utm25lr_features:
695695
self.utm25lr_features.append(layer_feature)
696696

697-
def downloadData(self, product_types):
697+
def download_data(self, product_types):
698698

699699
self.all_urls = []
700700
self.total_download_count = 0
701701
self.download_count = 0
702702
self.layers_added_count = 0
703703

704704
for product_key, product_title in product_types.items():
705-
urls = self.createDownloadURLS(product_key, product_title)
705+
urls = self.create_download_urls(product_key, product_title)
706706
self.all_urls.extend(urls)
707707
self.total_download_count += len(urls)
708708

@@ -721,9 +721,9 @@ def downloadData(self, product_types):
721721
self.progress_dialog.progressBar.reset()
722722
self.progress_dialog.progressBar.show()
723723
self.progress_dialog.label.setText("Downloading data...")
724-
QTimer.singleShot(1000, self.downloadOneFile)
724+
QTimer.singleShot(1000, self.download_one_file)
725725

726-
def downloadNLSProductTypes(self):
726+
def download_nls_product_types(self):
727727
products = {}
728728

729729
url = (
@@ -759,7 +759,7 @@ def downloadNLSProductTypes(self):
759759

760760
return products
761761

762-
def downloadOneFile(self):
762+
def download_one_file(self):
763763
if (
764764
self.download_count == self.total_download_count
765765
or self.download_count >= len(self.all_urls)
@@ -814,42 +814,42 @@ def downloadOneFile(self):
814814

815815
if self.download_count == self.total_download_count:
816816
QgsMessageLog.logMessage("done downloading data", "NLSgpkgloader", 0)
817-
self.createGeoPackage()
817+
self.create_geopackage()
818818
else:
819-
QTimer.singleShot(10, self.downloadOneFile)
819+
QTimer.singleShot(10, self.download_one_file)
820820

821-
def createGeoPackage(self):
821+
def create_geopackage(self):
822822
"""Creates a GeoPackage from the downloaded MTK data"""
823823
self.progress_dialog.progressBar.reset()
824824
self.progress_dialog.label.setText("Writing layers to GeoPackage...")
825825

826-
writeTask = CreateGeoPackageTask(
826+
write_task = CreateGeoPackageTask(
827827
"Write GML to GPKG",
828828
self.all_urls,
829829
self.total_download_count,
830830
self.selected_mtk_product_types,
831831
self.data_download_dir,
832832
self.gpkg_path,
833833
)
834-
dissolveTask = DissolveFeaturesTask("Dissolve features", self.gpkg_path)
835-
clipTask = ClipLayersTask("Clip layers", self.selected_geoms, self.gpkg_path)
836-
cleanupTask = CleanUpTask("Delete temporary tables", self.path, self.gpkg_path)
834+
dissolve_task = DissolveFeaturesTask("Dissolve features", self.gpkg_path)
835+
clip_task = ClipLayersTask("Clip layers", self.selected_geoms, self.gpkg_path)
836+
cleanup_task = CleanUpTask("Delete temporary tables", self.path, self.gpkg_path)
837837

838-
writeTask.taskCompleted.connect(lambda: self.runTask(dissolveTask))
839-
dissolveTask.taskCompleted.connect(lambda: self.runTask(clipTask))
840-
clipTask.taskCompleted.connect(lambda: self.runTask(cleanupTask))
841-
cleanupTask.taskCompleted.connect(lambda: self.finishProcessing())
838+
write_task.taskCompleted.connect(lambda: self.run_task(dissolve_task))
839+
dissolve_task.taskCompleted.connect(lambda: self.run_task(clip_task))
840+
clip_task.taskCompleted.connect(lambda: self.run_task(cleanup_task))
841+
cleanup_task.taskCompleted.connect(lambda: self.finish_processing())
842842

843-
self.runTask(writeTask)
843+
self.run_task(write_task)
844844

845-
def runTask(self, task):
845+
def run_task(self, task):
846846
self.progress_dialog.label.setText(task.description())
847847
task.progressChanged.connect(
848848
lambda: self.progress_dialog.progressBar.setValue(task.progress())
849849
)
850850
QgsApplication.taskManager().addTask(task)
851851

852-
def finishProcessing(self):
852+
def finish_processing(self):
853853
if self.addDownloadedDataAsLayer:
854854
self.progress_dialog.label.setText("Adding layers to QGIS")
855855
self.progress_dialog.progressBar.hide()
@@ -875,7 +875,7 @@ def finishProcessing(self):
875875
self.progress_dialog.hide()
876876
return True
877877

878-
def showSettingsDialog(self):
878+
def show_settings_dialog(self):
879879
self.nls_user_key_dialog.dataLocationQgsFileWidget.setStorageMode(
880880
QgsFileWidget.GetDirectory
881881
)
@@ -919,7 +919,7 @@ def showSettingsDialog(self):
919919
)
920920
return False
921921

922-
def createDownloadURLS(self, product_key, product_title):
922+
def create_download_urls(self, product_key, product_title):
923923
urls = []
924924
if (
925925
product_key

nlsgpkgloader/nls_geopackage_loader_tasks.py

+12-12
Original file line numberDiff line numberDiff line change
@@ -47,17 +47,17 @@ def __init__(self, description, urls, dlcount, products, dlpath, path):
4747
self.gpkg_path = path
4848

4949
def run(self):
50-
for dlIndex in range(0, self.total_download_count):
51-
url = self.all_urls[dlIndex][0]
50+
for dl_index in range(0, self.total_download_count):
51+
url = self.all_urls[dl_index][0]
5252
url_parts = url.split("/")
5353
file_name = url_parts[-1].split("?")[0]
54-
data_dir_name = self.all_urls[dlIndex][1]
54+
data_dir_name = self.all_urls[dl_index][1]
5555
data_dir_name = data_dir_name.replace(":", "_suhde_")
5656
dir_path = os.path.join(self.data_download_dir, data_dir_name)
5757
dir_path = os.path.join(dir_path, file_name.split(".")[0])
58-
data_type = self.all_urls[dlIndex][3]
58+
data_type = self.all_urls[dl_index][3]
5959

60-
percentage = dlIndex / float(self.total_download_count) * 100.0
60+
percentage = dl_index / float(self.total_download_count) * 100.0
6161
self.setProgress(percentage)
6262

6363
if not os.path.exists(dir_path):
@@ -203,7 +203,7 @@ def __init__(self, description, selected_geoms, path):
203203
self.gpkg_path = path
204204

205205
def run(self):
206-
combinedGeomLayer = QgsVectorLayer(
206+
combined_geom_layer = QgsVectorLayer(
207207
"MultiPolygon?crs=EPSG:3067", "clipLayer", "memory"
208208
)
209209
geom_union = None
@@ -212,17 +212,17 @@ def run(self):
212212
geom_union = geom
213213
else:
214214
geom_union = geom_union.combine(geom)
215-
dp = combinedGeomLayer.dataProvider()
215+
dp = combined_geom_layer.dataProvider()
216216

217-
combinedGeomLayer.startEditing()
217+
combined_geom_layer.startEditing()
218218
feat = QgsFeature()
219219
feat.setGeometry(geom_union)
220220
dp.addFeature(feat)
221-
combinedGeomLayer.commitChanges()
221+
combined_geom_layer.commitChanges()
222222

223-
params = {"INPUT": combinedGeomLayer, "OUTPUT": "memory:geomUnionLayer"}
223+
params = {"INPUT": combined_geom_layer, "OUTPUT": "memory:geomUnionLayer"}
224224
result = processing.run("native:dissolve", params)
225-
geomUnionLayer = result["OUTPUT"]
225+
geom_union_layer = result["OUTPUT"]
226226

227227
conn = ogr.Open(self.gpkg_path)
228228
total_tables = len(conn)
@@ -243,7 +243,7 @@ def run(self):
243243
layer_name = "zz_" + layer_name
244244
params = {
245245
"INPUT": self.gpkg_path + "|layername=" + table_name,
246-
"OVERLAY": geomUnionLayer,
246+
"OVERLAY": geom_union_layer,
247247
"OUTPUT": "ogr:dbname='"
248248
+ self.gpkg_path
249249
+ "' table=\""

nlsgpkgloader/ui/nls_geopackage_loader_user_key_dialog.py

-2
Original file line numberDiff line numberDiff line change
@@ -9,5 +9,3 @@ class NLSGeoPackageLoaderUserKeyDialog(QDialog, FORM_CLASS):
99
def __init__(self, parent=None):
1010
super().__init__(parent)
1111
self.setupUi(self)
12-
13-
print(dir(self))

tests/test_resources.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,11 @@
1515

1616
from qgis.PyQt.QtGui import QIcon
1717

18+
from nlsgpkgloader.qgis_plugin_tools.tools.resources import resources_path
19+
1820

1921
def test_icon_png():
2022
"""Test we can click OK."""
21-
path = ":/plugins/NLSGeoPackageLoader/icon.png"
23+
path = resources_path("icons", "icon.png")
2224
icon = QIcon(path)
2325
assert icon is not None

0 commit comments

Comments
 (0)