refactor(services): PARTIAL migrate get_all_services

selfprivacy_api/backup/__init__.py

@@ -8,10 +8,8 @@ import os
 from os import statvfs
 from typing import Callable, List, Optional
 
-from selfprivacy_api.services import (
+from selfprivacy_api.services import ServiceManager
-    get_service_by_id,
+
-    get_all_services,
-)
 from selfprivacy_api.services.service import (
     Service,
     ServiceStatus,
@@ -376,7 +374,7 @@ class Backups:
 
     @staticmethod
     def prune_all_autosnaps() -> None:
-        for service in get_all_services():
+        for service in ServiceManager.get_all_services():
             Backups._prune_auto_snaps(service)
 
     # Restoring
@@ -431,7 +429,7 @@ class Backups:
         snapshot: Snapshot, strategy=RestoreStrategy.DOWNLOAD_VERIFY_OVERWRITE
     ) -> None:
         """Restores a snapshot to its original service using the given strategy"""
-        service = get_service_by_id(snapshot.service_name)
+        service = ServiceManager.get_service_by_id(snapshot.service_name)
         if service is None:
             raise ValueError(
                 f"snapshot has a nonexistent service: {snapshot.service_name}"
@@ -475,7 +473,7 @@ class Backups:
     def _assert_restorable(
         snapshot: Snapshot, strategy=RestoreStrategy.DOWNLOAD_VERIFY_OVERWRITE
     ) -> None:
-        service = get_service_by_id(snapshot.service_name)
+        service = ServiceManager.get_service_by_id(snapshot.service_name)
         if service is None:
             raise ValueError(
                 f"snapshot has a nonexistent service: {snapshot.service_name}"
@@ -646,7 +644,7 @@ class Backups:
         """Returns a list of services that should be backed up at a given time"""
         return [
             service
-            for service in get_all_services()
+            for service in ServiceManager.get_all_services()
             if Backups.is_time_to_backup_service(service, time)
         ]
 

selfprivacy_api/graphql/queries/services.py

@@ -8,12 +8,12 @@ from selfprivacy_api.graphql.common_types.service import (
     Service,
     service_to_graphql_service,
 )
-from selfprivacy_api.services import get_all_services
+from selfprivacy_api.services import ServiceManager
 
 
 @strawberry.type
 class Services:
     @strawberry.field
     def all_services(self) -> typing.List[Service]:
-        services = get_all_services()
+        services = ServiceManager.get_all_services()
         return [service_to_graphql_service(service) for service in services]

selfprivacy_api/services/__init__.py

@@ -26,30 +26,32 @@ services: list[Service] = [
 ]
 
 
-def get_all_services() -> list[Service]:
+class ServiceManager(Service):
+    @staticmethod
+    def get_all_services() -> list[Service]:
         return services
 
-
+    @staticmethod
-def get_service_by_id(service_id: str) -> typing.Optional[Service]:
+    def get_service_by_id(service_id: str) -> typing.Optional[Service]:
         for service in services:
             if service.get_id() == service_id:
                 return service
         return None
 
-
+    @staticmethod
-def get_enabled_services() -> list[Service]:
+    def get_enabled_services() -> list[Service]:
         return [service for service in services if service.is_enabled()]
 
-
+    @staticmethod
-def get_disabled_services() -> list[Service]:
+    def get_disabled_services() -> list[Service]:
         return [service for service in services if not service.is_enabled()]
 
-
+    @staticmethod
-def get_services_by_location(location: str) -> list[Service]:
+    def get_services_by_location(location: str) -> list[Service]:
         return [service for service in services if service.get_drive() == location]
 
-
+    @staticmethod
-def get_all_required_dns_records() -> list[ServiceDnsRecord]:
+    def get_all_required_dns_records() -> list[ServiceDnsRecord]:
         ip4 = network_utils.get_ip4()
         ip6 = network_utils.get_ip6()
         dns_records: list[ServiceDnsRecord] = [

tests/test_autobackup.py

@@ -4,7 +4,7 @@ from copy import copy
 from datetime import datetime, timezone, timedelta
 
 from selfprivacy_api.jobs import Jobs
-from selfprivacy_api.services import Service, get_all_services
+from selfprivacy_api.services import Service, ServiceManager
 
 from selfprivacy_api.graphql.common_types.backup import (
     BackupReason,
@@ -23,7 +23,11 @@ from tests.test_graphql.test_services import only_dummy_service
 
 
 def backuppable_services() -> list[Service]:
-    return [service for service in get_all_services() if service.can_be_backed_up()]
+    return [
+        service
+        for service in ServiceManager.get_all_services()
+        if service.can_be_backed_up()
+    ]
 
 
 def dummy_snapshot(date: datetime):