Reject blank ownerless claim owner (#4929)

This commit is contained in:
red person
2026-06-28 12:57:11 +03:00
committed by GitHub
parent 8066a8e0cd
commit 827a6b2778
2 changed files with 15 additions and 3 deletions
+8 -2
View File
@@ -27,12 +27,18 @@ def claim_json_entries(entries, owner):
return count
def owner_arg(argv):
if len(argv) < 2 or not argv[1].strip():
return None
return argv[1].strip()
def main():
if len(sys.argv) < 2:
owner = owner_arg(sys.argv)
if not owner:
print("Usage: python scripts/claim_ownerless.py <username>")
sys.exit(1)
owner = sys.argv[1]
print(f"Claiming all ownerless data for: {owner}\n")
# 1. Memories (JSON files)
+7 -1
View File
@@ -1,4 +1,4 @@
from scripts.claim_ownerless import claim_json_entries
from scripts.claim_ownerless import claim_json_entries, owner_arg
def test_claim_json_entries_skips_invalid_rows():
@@ -16,3 +16,9 @@ def test_claim_json_entries_skips_invalid_rows():
None,
{"id": "b", "owner": "already"},
]
def test_owner_arg_rejects_blank_owner():
assert owner_arg(["claim_ownerless.py"]) is None
assert owner_arg(["claim_ownerless.py", " "]) is None
assert owner_arg(["claim_ownerless.py", " admin "]) == "admin"