mirror of
https://github.com/bellingcat/auto-archiver-api.git
synced 2026-06-08 03:28:35 +03:00
7e48f706df
* Update pyproject.toml * add pre-commit * Create .pre-commit-config.yaml * Comment out ruff * Update .pre-commit-config.yaml * General formatting * Create format-and-fail.yml * Update ci.yml * Add pre-commit to dev dependencies * Update pyproject.toml
27 lines
866 B
Python
27 lines
866 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
|