From be18cfda369de4227873ea25403810a5166e1bc0 Mon Sep 17 00:00:00 2001 From: Marc Michalsky Date: Wed, 16 Dec 2020 23:09:01 +0100 Subject: [PATCH] add return types --- models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/models.py b/models.py index 8d4b46b..152d7aa 100644 --- a/models.py +++ b/models.py @@ -40,7 +40,7 @@ class Container: self.restore_tar_file = "" # Create backup dir if it does not yet exist - def __create_backup_dir(self): + def __create_backup_dir(self) -> bool: if not os.path.isdir(self.backup_dir): try: os.makedirs(self.backup_dir) @@ -291,7 +291,7 @@ class Container: _print(F"{Fore.RED}Could not create temporary folder or backup folder. Backup aborted.{Style.RESET_ALL}") return False - def restore_backup(self, backup_file_path): + def restore_backup(self, backup_file_path) -> bool: self.backup_file_path = backup_file_path self.restore_dump_file = os.path.basename(backup_file_path[:-6] + "sql") @@ -321,7 +321,7 @@ class Container: _print(F"{Fore.RED}Could not create temporary folder. Restore aborted.{Style.RESET_ALL}") return False - def upgrade(self): + def upgrade(self) -> int: upgrade_functions = [ self.__enable_maintenance_mode,