warp-beacon 2.2.34__py3-none-any.whl → 2.2.36__py3-none-any.whl

warp_beacon/__version__.py

@@ -1,2 +1,2 @@
-__version__ = "2.2.34"
+__version__ = "2.2.36"
 

warp_beacon/scraper/account_selector.py

@@ -4,7 +4,6 @@ import json
 import re
 
 import multiprocessing
-from itertools import islice, cycle
 
 from warp_beacon.jobs import Origin
 
@@ -12,26 +11,24 @@ import logging
 
 class AccountSelector(object):
 	accounts = []
-	acc_pools = {}
 	current = None
 	current_module_name = None
-	index = None
 	accounts_meta_data = None
 	session_dir = "/var/warp_beacon"
 	manager = None
+	account_index = {}
 
 	def __init__(self, manager: multiprocessing.managers.SyncManager, acc_file_path: str) -> None:
 		self.manager = manager
-		self.index = self.manager.Value('i', 0)
 		self.accounts_meta_data = self.manager.dict()
 		if os.path.exists(acc_file_path):
 			with open(acc_file_path, 'r', encoding="utf-8") as f:
 				self.accounts = json.loads(f.read())
 			if self.accounts:
 				self.__init_meta_data()
-				self.load_yt_sessions()
+				#self.load_yt_sessions()
 				for acc_type, _ in self.accounts.items():
-					self.acc_pools[acc_type] = cycle(self.accounts[acc_type])
+					self.account_index[acc_type] = self.manager.Value('i', 0)
 		else:
 			raise ValueError("Accounts file not found")
 
@@ -42,7 +39,8 @@ class AccountSelector(object):
 	#	for k, v in self.accounts.items():
 
 	def load_yt_sessions(self) -> None:
-		self.accounts["youtube"] = []
+		if "youtube" not in self.accounts:
+			self.accounts["youtube"] = []
 		for f in os.listdir(self.session_dir):
 			if "yt_session" in f and ".json" in f:
 				match = re.search(r'\d+', f)
@@ -64,27 +62,29 @@ class AccountSelector(object):
 		module_name = 'youtube' if next((s for s in ("yt", "youtube", "youtu_be") if s in module_origin.value), None) else 'instagram'
 		self.current_module_name = module_name
 		if self.current is None:
-			self.current = self.accounts[self.current_module_name][self.index.value]
-			#self.acc_pools[self.current_module_name] = next(islice(self.acc_pools[self.current_module_name], self.index.value, None))
+			self.current = self.accounts[self.current_module_name][self.account_index[self.current_module_name].value]
 
 	def next(self) -> dict:
-		self.current = next(self.acc_pools[self.current_module_name])
-		self.index.value = self.accounts[self.current_module_name].index(self.current)
-		logging.info("Selected account index is '%d'", self.index.value)
+		idx = self.account_index[self.current_module_name].value + 1
+		if idx > len(self.accounts[self.current_module_name]):
+			idx = 0
+		self.current = self.accounts[self.current_module_name][idx]
+		self.account_index[self.current_module_name].value = idx
+		logging.info("Selected account index is '%d'", idx)
 		return self.current
 	
 	def bump_acc_fail(self, key: str, amount: int = 1) -> int:
-		self.accounts_meta_data[self.index.value][key] += amount
-		return self.accounts_meta_data[self.index.value][key]
+		self.accounts_meta_data[self.account_index[self.current_module_name].value][key] += amount
+		return self.accounts_meta_data[self.account_index[self.current_module_name].value][key]
 
 	def how_much(self, key: str) -> int:
-		return self.accounts_meta_data[self.current_module_name][self.index.value][key]
+		return self.accounts_meta_data[self.current_module_name][self.account_index[self.current_module_name].value][key]
 	
 	def get_current(self) -> tuple:
-		return (self.index.value, self.current)
+		return (self.account_index[self.current_module_name].value, self.current)
 	
 	def get_meta_data(self) -> dict:
-		return self.accounts_meta_data[self.current_module_name][self.index.value]
+		return self.accounts_meta_data[self.current_module_name][self.account_index[self.current_module_name].value]
 	
 	def count_service_accounts(self, mod_name: Origin) -> int:
 		module_name = 'youtube' if next((s for s in ("yt", "youtube", "youtu_be") if s in mod_name.value), None) else 'instagram'

{warp_beacon-2.2.34.dist-info → warp_beacon-2.2.36.dist-info}/METADATA

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

{warp_beacon-2.2.34.dist-info → warp_beacon-2.2.36.dist-info}/RECORD

@@ -3,7 +3,7 @@ lib/systemd/system/warp_beacon.service,sha256=lPmHqLqcI2eIV7nwHS0qcALQrznixqJuww
 var/warp_beacon/accounts.json,sha256=2QQMS9N_Z-TnfZq4U1vSUe4IKyZGmTFdgo8BoF4hQ7E,1710
 var/warp_beacon/placeholder.gif,sha256=cE5CGJVaop4Sx21zx6j4AyoHU0ncmvQuS2o6hJfEH88,6064
 warp_beacon/__init__.py,sha256=_rThNODmz0nDp_n4mWo_HKaNFE5jk1_7cRhHyYaencI,163
-warp_beacon/__version__.py,sha256=oYqTWRUbR0yBNg1Ucm85He64XgsCurjGgTv4ct-G_S0,24
+warp_beacon/__version__.py,sha256=VrKs5E5SeGI4N9fYl4L9vngIM8ORIydBeuAwozQZqPE,24
 warp_beacon/warp_beacon.py,sha256=7KEtZDj-pdhtl6m-zFLsSojs1ZR4o7L0xbqtdmYPvfE,342
 warp_beacon/compress/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 warp_beacon/compress/video.py,sha256=_PDMVYCyzLYxHv1uZmmzGcG_8rjaZr7BTXsXTTy_oS4,2846
@@ -21,7 +21,7 @@ warp_beacon/scheduler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 warp_beacon/scheduler/scheduler.py,sha256=oKuDP3RqUQmNYBorKxnhz9QqA-eLTJHVTinWbsYB530,2585
 warp_beacon/scraper/__init__.py,sha256=tkl7UyfqFcvPlxomVPB9GpWQvAoADaM6jGb_bmx1zRY,15229
 warp_beacon/scraper/abstract.py,sha256=aNZ9ypF9B8BjflcIwi-7wEzIqF-XPeF0xvfX9CP_iIw,2708
-warp_beacon/scraper/account_selector.py,sha256=jkf8sWncNj37YAK0_KT-5XZ7aHSpSz8Y4LIzb4LXnRE,3387
+warp_beacon/scraper/account_selector.py,sha256=lX9l2RwMbPKhSjmEvWLKTuo-6-QI8kA52eoPMKSCo7o,3510
 warp_beacon/scraper/exceptions.py,sha256=lHsPrYy5iYnHsIdUHqRxZmzlqrkNj_TS4TkiTJfWhTs,1326
 warp_beacon/scraper/instagram/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 warp_beacon/scraper/instagram/instagram.py,sha256=D_r6Kdpe4iBjCeGv-vVIodiVnZps6ZyBbaY0JC7mGWI,13782
@@ -37,9 +37,9 @@ warp_beacon/telegram/handlers.py,sha256=xSNG_v4D4FaA5q1_5wXvJkjcDUhy5GaQnfpijJZp
 warp_beacon/telegram/placeholder_message.py,sha256=32U-R_IMPRJd-S5UwlpemLkArJOxCmO7whcynduSbMA,6418
 warp_beacon/telegram/utils.py,sha256=CeWXMwyzv2l4Af6I-wW2ySU490c-5k82sBhkjabv0n4,2452
 warp_beacon/uploader/__init__.py,sha256=rsUkzzmtaeQY-kNkcIiPsrayS9idFKY3hBC71mR19XE,4743
-warp_beacon-2.2.34.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-warp_beacon-2.2.34.dist-info/METADATA,sha256=-i1V1OCyRp7_knj7oPpmyLARLn1BV1nUzGRuzXECGwQ,21251
-warp_beacon-2.2.34.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-warp_beacon-2.2.34.dist-info/entry_points.txt,sha256=eSB61Rb89d56WY0O-vEIQwkn18J-4CMrJcLA_R_8h3g,119
-warp_beacon-2.2.34.dist-info/top_level.txt,sha256=ALb_Ft_eG-OY4_m0TWUifNUOZsrx483L-zw7G7vqXoc,971
-warp_beacon-2.2.34.dist-info/RECORD,,
+warp_beacon-2.2.36.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+warp_beacon-2.2.36.dist-info/METADATA,sha256=TmWY6ey4APMBelOZVDrgNY9MM4wEsnCkiznCz-OmcOQ,21251
+warp_beacon-2.2.36.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+warp_beacon-2.2.36.dist-info/entry_points.txt,sha256=eSB61Rb89d56WY0O-vEIQwkn18J-4CMrJcLA_R_8h3g,119
+warp_beacon-2.2.36.dist-info/top_level.txt,sha256=ALb_Ft_eG-OY4_m0TWUifNUOZsrx483L-zw7G7vqXoc,971
+warp_beacon-2.2.36.dist-info/RECORD,,