mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-11 13:08:34 +03:00
26 lines
865 B
Python
26 lines
865 B
Python
# TODO: temporary file for this code, maybe other code belongs here, maybe not. do decide
|
|
|
|
|
|
import datetime
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.shared.db import worker_crud
|
|
|
|
|
|
def get_store_archive_until(db: Session, group_id: str) -> datetime.datetime:
|
|
group = worker_crud.get_group(db, group_id)
|
|
assert group, f"Group {group_id} not found."
|
|
assert group.permissions and type(group.permissions) == dict, f"Group {group_id} has no permissions."
|
|
|
|
max_lifespan = group.permissions.get("max_archive_lifespan_months", -1)
|
|
if max_lifespan == -1: return None
|
|
|
|
return datetime.datetime.now() + datetime.timedelta(days=30 * max_lifespan)
|
|
|
|
|
|
def get_store_archive_until_or_never(db: Session, group_id: str) -> datetime.datetime:
|
|
try:
|
|
return get_store_archive_until(db, group_id)
|
|
except AssertionError as e:
|
|
return None
|