mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-06-28 15:45:22 -04:00
Reject blank ownerless claim owner (#4929)
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user