Add types to some function definitions in ResolutionManager.py

This commit is contained in:
AccentuSoft
2022-07-20 23:14:47 +03:00
parent d8e60c9bbd
commit 4fa771d0a5

View File

@@ -18,7 +18,7 @@ class ResolutionManager:
self.mainWindow = mainWindow self.mainWindow = mainWindow
self.resolutions = {} self.resolutions = {}
def loadResolutionsFromDir(self, directory: Path): def loadResolutionsFromDir(self, directory: Path) -> None:
exceptionsCount = 0 exceptionsCount = 0
for resolution in listdir(directory): for resolution in listdir(directory):
resolution = str(resolution) resolution = str(resolution)
@@ -86,14 +86,14 @@ class ResolutionManager:
return resolutionDescription return resolutionDescription
return None return None
def loadResolutionsFromServer(self, serverRes): def loadResolutionsFromServer(self, serverRes) -> None:
for category in serverRes: for category in serverRes:
if category not in self.resolutions: if category not in self.resolutions:
self.resolutions[category] = {} self.resolutions[category] = {}
for serverResolution in serverRes[category]: for serverResolution in serverRes[category]:
self.resolutions[category][serverResolution] = serverRes[category][serverResolution] self.resolutions[category][serverResolution] = serverRes[category][serverResolution]
def removeServerResolutions(self): def removeServerResolutions(self) -> None:
for category in dict(self.resolutions): for category in dict(self.resolutions):
for resolution in dict(self.resolutions[category]): for resolution in dict(self.resolutions[category]):
# If resolution class does not exist locally, then assume it exists on the server. # If resolution class does not exist locally, then assume it exists on the server.
@@ -103,13 +103,10 @@ class ResolutionManager:
if len(self.resolutions[category]) == 0: if len(self.resolutions[category]) == 0:
self.resolutions.pop(category) self.resolutions.pop(category)
def getResolutionCategories(self): def getResolutionCategories(self) -> list:
result = [] return list(self.resolutions)
for category in self.resolutions:
result.append(category)
return result
def getResolutionsForEntityTypesByCategory(self, eTypes): def getResolutionsForEntityTypesByCategory(self, eTypes) -> dict:
""" """
Gets a set of entity types, and returns a dictionary with all the resolutions that can take all Gets a set of entity types, and returns a dictionary with all the resolutions that can take all
included types as input. included types as input.
@@ -124,12 +121,12 @@ class ResolutionManager:
result[category].append(resolution) result[category].append(resolution)
return result return result
def getResolutionsInCategory(self, category): def getResolutionsInCategory(self, category) -> list:
if category in self.resolutions: if category in self.resolutions:
return list(self.resolutions[category]) return list(self.resolutions[category])
return [] return []
def getAllResolutions(self): def getAllResolutions(self) -> list:
categories = self.getResolutionCategories() categories = self.getResolutionCategories()
result = [] result = []
for category in categories: for category in categories: