norohind/jubilant-system-core
1
0
Fork 0
mirror of https://github.com/norohind/jubilant-system-core.git synced 2025-04-12 09:50:01 +03:00
Code Issues Projects Releases Wiki Activity

Implement force-update cli command for suppress_absence=False

Browse Source
This commit is contained in:
norohind 2022-05-03 23:08:18 +03:00
parent 3791074eb4
commit 1dfaa3cde2
Signed by: norohind
GPG Key ID: 01C3BECC26FB59E1
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
main.py
View File

@ -126,9 +126,10 @@ def discover(back_count: int = 0):
tries = tries + 1
def update(squad_id: int = None, amount_to_update: int = 1):
def update(squad_id: int = None, amount_to_update: int = 1, suppress_absence=True):
"""
:param suppress_absence: if we're manually updating a squadron which we know for sure are deleted
:param squad_id: update specified squad, updates only that squad
:param amount_to_update: update specified amount, ignores when squad_id specified
:return:
@ -136,7 +137,7 @@ def update(squad_id: int = None, amount_to_update: int = 1):
if isinstance(squad_id, int):
logger.debug(f'Going to update one specified squadron: {squad_id} ID')
FAPI.update_squad(squad_id, suppress_absence=True)
FAPI.update_squad(squad_id, suppress_absence=suppress_absence)
# suppress_absence is required because if we're manually updating squad with some high id it may just don't exist yet
return
@ -170,6 +171,7 @@ def main():
main.py update
main.py update amount <amount: int>
main.py update id <id: int>
main.py force-update id <id: int>
main.py daemon
main.py hooks notify <inserted;deleted> <operation_id: int>"""
@ -221,8 +223,8 @@ def main():
exit(1)
elif len(sys.argv) == 4:
if sys.argv[1] == 'update':
if sys.argv[2] == 'amount':
if sys.argv[1] in ('update', 'force-update'):
if sys.argv[2] == 'amount' and sys.argv[1] == 'update':
# main.py update amount <amount: int>
try:
@ -239,6 +241,13 @@ def main():
elif sys.argv[2] == 'id':
# main.py update id <id: int>
# main.py force-update id <id: int>
if sys.argv[1] == 'force-update':
suppress_absence = False
else:
suppress_absence = True
try:
id_for_update: int = int(sys.argv[3])
@ -248,7 +257,7 @@ def main():
exit(1)
logger.info(f'Entering update specified squad: {id_for_update} ID')
update(squad_id=id_for_update)
update(squad_id=id_for_update, suppress_absence=suppress_absence)
exit(0)
else: