bellingcat/auto-archiver-api
1
0
Fork 0
mirror of https://github.com/bellingcat/auto-archiver-api.git synced 2026-06-10 04:28:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix orchestrator file presence check based on enabled group permissions

Browse Source
This commit is contained in:
msramalho
2025-04-03 15:13:58 +01:00
parent c65f3fe121
commit fd799c8fbc
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
app/shared/user_groups.py
View File

@@ -94,20 +94,31 @@ class GroupPermissions(BaseModel):
class GroupModel(BaseModel):
description: str
orchestrator: str
orchestrator_sheet: str
orchestrator: str | None = None
orchestrator_sheet: str | None = None
permissions: GroupPermissions
@classmethod
@field_validator("orchestrator", "orchestrator_sheet", mode="before")
@field_validator("orchestrator", mode="before")
def validate_orchestrator(cls, v):
if not os.path.exists(v):
# orchestrator is only needed if the group has archive_url permission
if cls.permissions.archive_url and not os.path.exists(v):
raise ValueError(f"Orchestrator file not found with this path: {v}")
return v
@classmethod
@field_validator("orchestrator_sheet", mode="before")
def validate_orchestrator_sheet(cls, v):
# orchestrator_sheet is only needed if the group has archive_sheet permission
if cls.permissions.archive_sheet and not os.path.exists(v):
raise ValueError(f"Orchestrator file not found with this path: {v}")
return v
@computed_field
@property
def service_account_email(self) -> str:
if self.orchestrator_sheet is None:
return ""
if hasattr(self, "_service_account_email"):
return self._service_account_email
orch = yaml.safe_load(open(self.orchestrator_sheet))