mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-08 03:28:35 +03:00
36 lines
955 B
Python
36 lines
955 B
Python
import datetime
|
|
from typing import Union
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.shared.db import worker_crud
|
|
|
|
|
|
# TODO: temporary file for this code, maybe other code belongs here, maybe not. do
|
|
# decide
|
|
|
|
|
|
def get_store_archive_until(
|
|
db: Session, group_id: str
|
|
) -> Union[datetime.datetime, None]:
|
|
group = worker_crud.get_group(db, group_id)
|
|
assert group, f"Group {group_id} not found."
|
|
assert group.permissions and isinstance(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
|
|
) -> Union[datetime.datetime, None]:
|
|
try:
|
|
return get_store_archive_until(db, group_id)
|
|
except AssertionError:
|
|
return None
|