mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-06-30 00:22:10 -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
|
return count
|
||||||
|
|
||||||
|
|
||||||
|
def owner_arg(argv):
|
||||||
|
if len(argv) < 2 or not argv[1].strip():
|
||||||
|
return None
|
||||||
|
return argv[1].strip()
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
if len(sys.argv) < 2:
|
owner = owner_arg(sys.argv)
|
||||||
|
if not owner:
|
||||||
print("Usage: python scripts/claim_ownerless.py <username>")
|
print("Usage: python scripts/claim_ownerless.py <username>")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
owner = sys.argv[1]
|
|
||||||
print(f"Claiming all ownerless data for: {owner}\n")
|
print(f"Claiming all ownerless data for: {owner}\n")
|
||||||
|
|
||||||
# 1. Memories (JSON files)
|
# 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():
|
def test_claim_json_entries_skips_invalid_rows():
|
||||||
@@ -16,3 +16,9 @@ def test_claim_json_entries_skips_invalid_rows():
|
|||||||
None,
|
None,
|
||||||
{"id": "b", "owner": "already"},
|
{"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