Refactor sleep logic in main.py and update status_send.py

This commit is contained in:
sHa
2024-04-08 13:10:29 +03:00
parent d890781080
commit 1e95d9e72a
2 changed files with 6 additions and 2 deletions

View File

@@ -38,7 +38,10 @@ def main(source: str = "autodetect", destination: str = "mattermost"):
status_curr = status status_curr = status
if not status_result: if not status_result:
continue continue
time.sleep(status.get("duration") or SLEEP_TIME) sleep = status.get("remaining_time") or status.get("duration")
if sleep is None or sleep <= 0:
sleep = SLEEP_TIME
time.sleep(sleep)
if __name__ == "__main__": if __name__ == "__main__":

View File

@@ -44,7 +44,8 @@ class StatusSend:
"expires_at": expires_at.isoformat() if expires_at else None, "expires_at": expires_at.isoformat() if expires_at else None,
} }
now = time.strftime("%H:%M:%S", time.localtime()) now = time.strftime("%H:%M:%S", time.localtime())
print(f"{now} Setting status to {self.destination.capitalize()} of {emoji.get("icon")} {text} ⏱️ for {duration} seconds") remaining_time = round(kwargs.get("remaining_time", 0))
print(f"{now} Setting status to {self.connector.NAME} of {emoji.get("icon")} {text} ⏱️ for {remaining_time} seconds")
try: try:
self.connector.send(data=data) self.connector.send(data=data)
except Exception as e: except Exception as e: