From ce220b1ff7735df5453c8ef7ede89262b091db81 Mon Sep 17 00:00:00 2001
From: "Alfonso E.M" <alfonso@el-magnifico.org>
Date: Tue, 16 Mar 2021 19:51:35 +0100
Subject: [PATCH] Timer to avoid saturating Telegram api

---
 telegram-download-daemon.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/telegram-download-daemon.py b/telegram-download-daemon.py
index 6c83f00..36ece73 100644
--- a/telegram-download-daemon.py
+++ b/telegram-download-daemon.py
@@ -7,6 +7,7 @@ from os import getenv
 from shutil import move
 import subprocess
 import math
+import time
 
 from sessionManager import getSession, saveSession
 
@@ -22,7 +23,7 @@ import argparse
 import asyncio
 
 
-TDD_VERSION="1.3"
+TDD_VERSION="1.4"
 
 TELEGRAM_DAEMON_API_ID = getenv("TELEGRAM_DAEMON_API_ID")
 TELEGRAM_DAEMON_API_HASH = getenv("TELEGRAM_DAEMON_API_HASH")
@@ -81,6 +82,9 @@ channel_id = args.channel
 downloadFolder = args.dest
 tempFolder = args.temp
 worker_count = multiprocessing.cpu_count()
+updateFrequency = 10
+lastUpdate = 0
+#multiprocessing.Value('f', 0)
 
 if not tempFolder:
     tempFolder = downloadFolder
@@ -112,6 +116,9 @@ def getFilename(event: events.NewMessage.Event):
 in_progress={}
 
 async def set_progress(filename, message, received, total):
+    global lastUpdate
+    global updateFrequency
+
     if received >= total:
         try: in_progress.pop(filename)
         except: pass
@@ -121,8 +128,10 @@ async def set_progress(filename, message, received, total):
     progress_message= "{0} % ({1} / {2})".format(percentage, received, total)
     in_progress[filename] = progress_message
 
-    if (int(percentage) % 5) == 0:
+    currentTime=time.time()
+    if (currentTime - lastUpdate) > updateFrequency:
         await log_reply(message, progress_message)
+        lastUpdate=currentTime
 
 
 with TelegramClient(getSession(), api_id, api_hash,
@@ -200,6 +209,7 @@ with TelegramClient(getSession(), api_id, api_hash,
                 print('Queue worker error: ', e)
  
     async def start():
+
         tasks = []
         loop = asyncio.get_event_loop()
         for i in range(worker_count):