mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-28 06:51:28 +00:00
feature(backups): add a datetime validator function for huey autobackups
This commit is contained in:
parent
aeec3ad0a2
commit
f6402f2394
|
@ -1,5 +1,5 @@
|
||||||
from typing import List, Optional
|
from typing import List, Optional
|
||||||
from datetime import datetime
|
from datetime import datetime, timezone, timedelta
|
||||||
|
|
||||||
from selfprivacy_api.models.backup.snapshot import Snapshot
|
from selfprivacy_api.models.backup.snapshot import Snapshot
|
||||||
from selfprivacy_api.models.backup.provider import BackupProviderModel
|
from selfprivacy_api.models.backup.provider import BackupProviderModel
|
||||||
|
@ -78,21 +78,52 @@ class Backups:
|
||||||
redis.expire(snapshot_key, REDIS_SNAPSHOT_CACHE_EXPIRE_SECONDS)
|
redis.expire(snapshot_key, REDIS_SNAPSHOT_CACHE_EXPIRE_SECONDS)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _redis_autobackup_key(service: Service) -> str:
|
def _redis_autobackup_key(service_name: str) -> str:
|
||||||
return REDIS_AUTOBACKUP_ENABLED_PREFIX + service.get_id()
|
return REDIS_AUTOBACKUP_ENABLED_PREFIX + service_name
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def enable_autobackup(service: Service):
|
def enable_autobackup(service: Service):
|
||||||
redis.set(Backups._redis_autobackup_key(service), 1)
|
redis.set(Backups._redis_autobackup_key(service.get_id()), 1)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def is_time_to_backup(time: datetime) -> bool:
|
||||||
|
"""
|
||||||
|
Intended as a time validator for huey cron scheduler of automatic backups
|
||||||
|
"""
|
||||||
|
for key in redis.keys(REDIS_AUTOBACKUP_ENABLED_PREFIX + "*"):
|
||||||
|
service_id = key.split(":")[-1]
|
||||||
|
if Backups.is_time_to_backup_service(service_id, time):
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def is_time_to_backup_service(service_id: str, time: datetime):
|
||||||
|
period = Backups.autobackup_period_minutes()
|
||||||
|
if period is None:
|
||||||
|
return False
|
||||||
|
if not Backups._is_autobackup_enabled_by_name(service_id) is None:
|
||||||
|
return False
|
||||||
|
|
||||||
|
last_backup = Backups._get_last_backup_time_redis(service_id)
|
||||||
|
if last_backup is None:
|
||||||
|
return True # queue a backup immediately if there are no previous backups
|
||||||
|
|
||||||
|
if time > last_backup + timedelta(minutes=period):
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def disable_autobackup(service: Service):
|
def disable_autobackup(service: Service):
|
||||||
"""also see disable_all_autobackup()"""
|
"""also see disable_all_autobackup()"""
|
||||||
redis.delete(Backups._redis_autobackup_key(service))
|
redis.delete(Backups._redis_autobackup_key(service.get_id()))
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def is_autobackup_enabled(service: Service) -> bool:
|
def is_autobackup_enabled(service: Service) -> bool:
|
||||||
return redis.exists(Backups._redis_autobackup_key(service))
|
return Backups._is_autobackup_enabled_by_name(service.get_id())
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _is_autobackup_enabled_by_name(service_name: str):
|
||||||
|
return redis.exists(Backups._redis_autobackup_key(service_name))
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def autobackup_period_minutes() -> Optional[int]:
|
def autobackup_period_minutes() -> Optional[int]:
|
||||||
|
|
Loading…
Reference in a new issue