warp-beacon 2.6.16__tar.gz → 2.6.17__tar.gz

{warp_beacon-2.6.16/warp_beacon.egg-info → warp_beacon-2.6.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warp_beacon
-Version: 2.6.16
+Version: 2.6.17
 Summary: Telegram bot for expanding external media links
 Home-page: https://github.com/sb0y/warp_beacon
 Author: Andrey Bagrintsev

warp_beacon-2.6.17/warp_beacon/__version__.py

@@ -0,0 +1,2 @@
+__version__ = "2.6.17"
+

{warp_beacon-2.6.16 → warp_beacon-2.6.17}/warp_beacon/scraper/__init__.py

@@ -397,6 +397,7 @@ class AsyncDownloader(object):
 
 	def stop_all(self) -> None:
 		self.allow_loop.value = 0
+		self.acc_selector.save_ig_request_count()
 		for proc in self.workers:
 			if proc.is_alive():
 				logging.info("stopping process #%d", proc.pid)

{warp_beacon-2.6.16 → warp_beacon-2.6.17}/warp_beacon/scraper/account_selector.py

@@ -40,9 +40,33 @@ class AccountSelector(object):
 			if proxy_file_path:
 				with open(proxy_file_path, 'r', encoding="utf-8") as f:
 					self.proxies = json.loads(f.read())
+
+			self.load_ig_request_count()
 		else:
 			raise ValueError("Accounts file not found")
 
+	def save_ig_request_count(self) -> None:
+		try:
+			state = {"ig_count": self.ig_request_count.value}
+			with open(f"{self.session_dir}/ig_request_counter.json", "w+", encoding="utf-8") as f:
+				f.write(json.dumps(state))
+		except Exception as e:
+			logging.error("Failed to save accounts states!")
+			logging.exception(e)
+
+	def load_ig_request_count(self) -> None:
+		try:
+			state = {}
+			filepath = f"{self.session_dir}/ig_request_counter.json"
+			if os.path.exists(filepath):
+				with open(f"{self.session_dir}/ig_request_counter.json", "r", encoding="utf-8") as f:
+					state = json.loads(f.read())
+				if state:
+					self.ig_request_count.value = int(state.get("ig_count", 0))
+		except Exception as e:
+			logging.error("Failed to save accounts states!")
+			logging.exception(e)
+
 	def get_current_proxy(self) -> Optional[dict]:
 		return self.current_proxy
 

{warp_beacon-2.6.16 → warp_beacon-2.6.17}/warp_beacon/scraper/fail_handler.py

@@ -1,12 +1,10 @@
-import os
-
 import pickle
 
+import logging
+
 from warp_beacon.storage.mongo import DBClient
 from warp_beacon.jobs.download_job import DownloadJob
 
-import logging
-
 class FailHandler(object):
 	client = None
 	db = None

{warp_beacon-2.6.16 → warp_beacon-2.6.17/warp_beacon.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: warp_beacon
-Version: 2.6.16
+Version: 2.6.17
 Summary: Telegram bot for expanding external media links
 Home-page: https://github.com/sb0y/warp_beacon
 Author: Andrey Bagrintsev

warp_beacon-2.6.16/warp_beacon/__version__.py

@@ -1,2 +0,0 @@
-__version__ = "2.6.16"
-