Add some missing type annotations for return values

Author: correctmost

archinstall/lib/crypt.py

@@ -108,7 +108,7 @@ def encrypt(password: str, data: str) -> str:
 	return f"$argon2id${encoded_salt}${encoded_token}"
 
 
-def decrypt(data: str, password: str):
+def decrypt(data: str, password: str) -> str:
 	_, algo, encoded_salt, encoded_token = data.split("$")
 	salt = base64.urlsafe_b64decode(encoded_salt)
 	token = base64.urlsafe_b64decode(encoded_token)

archinstall/lib/installer.py

@@ -1513,7 +1513,7 @@ def _config_uki(
 		if not self.mkinitcpio(["-P"]):
 			error("Error generating initramfs (continuing anyway)")
 
-	def add_bootloader(self, bootloader: Bootloader, uki_enabled: bool = False):
+	def add_bootloader(self, bootloader: Bootloader, uki_enabled: bool = False) -> None:
 		"""
 		Adds a bootloader to the installation instance.
 		Archinstall supports one of three types:
@@ -1530,7 +1530,7 @@ def add_bootloader(self, bootloader: Bootloader, uki_enabled: bool = False):
 				# Allow plugins to override the boot-loader handling.
 				# This allows for bot configuring and installing bootloaders.
 				if plugin.on_add_bootloader(self):
-					return True
+					return
 
 		efi_partition = self._get_efi_partition()
 		boot_partition = self._get_boot_partition()
@@ -1560,7 +1560,7 @@ def add_bootloader(self, bootloader: Bootloader, uki_enabled: bool = False):
 	def add_additional_packages(self, packages: str | list[str]) -> None:
 		return self.pacman.strap(packages)
 
-	def enable_sudo(self, user: User, group: bool = False):
+	def enable_sudo(self, user: User, group: bool = False) -> None:
 		info(f"Enabling sudo permissions for {user.username}")
 
 		sudoers_dir = self.target / "etc/sudoers.d"

archinstall/lib/mirrors.py

@@ -336,12 +336,12 @@ def select_mirror_regions(preset: list[MirrorRegion]) -> list[MirrorRegion]:
 			return selected_mirrors
 
 
-def add_custom_mirror_servers(preset: list[CustomServer] = []):
+def add_custom_mirror_servers(preset: list[CustomServer] = []) -> list[CustomServer]:
 	custom_mirrors = CustomMirrorServersList(preset).run()
 	return custom_mirrors
 
 
-def select_custom_mirror(preset: list[CustomRepository] = []):
+def select_custom_mirror(preset: list[CustomRepository] = []) -> list[CustomRepository]:
 	custom_mirrors = CustomMirrorRepositoriesList(preset).run()
 	return custom_mirrors
 

archinstall/lib/models/users.py

@@ -132,7 +132,7 @@ def plaintext(self) -> str:
 		return self._plaintext
 
 	@plaintext.setter
-	def plaintext(self, value: str):
+	def plaintext(self, value: str) -> None:
 		self._plaintext = value
 		self.enc_password = crypt_yescrypt(value)