warp-beacon 2.2.0__tar.gz → 2.2.2__tar.gz

{warp_beacon-2.2.0/warp_beacon.egg-info → warp_beacon-2.2.2}/PKG-INFO

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

warp_beacon-2.2.2/warp_beacon/__version__.py

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

{warp_beacon-2.2.0 → warp_beacon-2.2.2}/warp_beacon/scheduler/scheduler.py

@@ -1,4 +1,7 @@
+import os
+import time
 import threading
+import json
 
 from warp_beacon.jobs import Origin
 import warp_beacon
@@ -6,10 +9,12 @@ import warp_beacon
 import logging
 
 class IGScheduler(object):
+	state_file = "/var/warp_beacon/scheduler_state.json"
 	downloader = None
 	running = True
 	thread = None
 	event = None
+	state = {"remaining": 3600}
 
 	def __init__(self, downloader: warp_beacon.scraper.AsyncDownloader) -> None:
 		self.downloader = downloader
@@ -18,6 +23,23 @@ class IGScheduler(object):
 	def __del__(self) -> None:
 		self.stop()
 
+	def save_state(self) -> None:
+		try:
+			with open(self.state_file, 'w+', encoding="utf-8") as f:
+				f.write(json.dumps(self.state))
+		except Exception as e:
+			logging.error("Failed to save Scheduler state!")
+			logging.exception(e)
+
+	def load_state(self) -> None:
+		try:
+			if os.path.exists(self.state_file):
+				with open(self.state_file, 'r', encoding="utf-8") as f:
+					self.state = json.loads(f.read())
+		except Exception as e:
+			logging.error("Failed to load Scheduler state!")
+			logging.exception(e)
+
 	def start(self) -> None:
 		self.thread = threading.Thread(target=self.do_work)
 		self.thread.start()
@@ -55,11 +77,20 @@ class IGScheduler(object):
 
 	def do_work(self) -> None:
 		logging.info("Scheduler thread started ...")
+		self.load_state()
 		while self.running:
 			try:
+				if self.state["remaining"] <= 0:
+					self.state["remaining"] = 3600
+
+				start_time = time.time()
+				self.event.wait(timeout=self.state["remaining"])
+				elapsed = time.time() - start_time
+				self.state["remaining"] -= elapsed
+
 				logging.info("Scheduler waking up")
 				self.validate_ig_session()
-				self.event.wait(timeout=3600)
 			except Exception as e:
 				logging.error("An error occurred in scheduler thread!")
-				logging.exception(e)
+				logging.exception(e)
+		self.save_state()

{warp_beacon-2.2.0 → warp_beacon-2.2.2}/warp_beacon/telegram/bot.py

@@ -170,6 +170,7 @@ class Bot(object):
 					)
 				else:
 					args["video"] = job.tg_file_id.replace(":video", '')
+					args["caption"] = self.build_signature_caption(job)
 			else:
 				if job.placeholder_message_id:
 					args["media"] = InputMediaVideo(
@@ -200,6 +201,7 @@ class Bot(object):
 					)
 				else:
 					args["photo"] = job.tg_file_id.replace(":image", '')
+					args["caption"] = self.build_signature_caption(job)
 			else:
 				if job.placeholder_message_id:
 					args["media"] = InputMediaPhoto(
@@ -208,6 +210,7 @@ class Bot(object):
 					)
 				else:
 					args["photo"] = job.local_media_path
+					args["caption"] = self.build_signature_caption(job)
 		elif job.media_type == JobType.AUDIO:
 			if job.tg_file_id:
 				if job.placeholder_message_id:
@@ -239,10 +242,12 @@ class Bot(object):
 			if job.tg_file_id:
 				if job.placeholder_message_id:
 					args["media"] = InputMediaAnimation(
-						media=job.tg_file_id.replace(":animation", '')
+						media=job.tg_file_id.replace(":animation", ''),
+						caption=self.build_signature_caption(job)
 					)
 				else:
 					args["animation"] = job.tg_file_id.replace(":animation", '')
+					args["caption"] = self.build_signature_caption(job)
 			else:
 				if job.placeholder_message_id:
 					args["media"] = InputMediaAnimation(

{warp_beacon-2.2.0 → warp_beacon-2.2.2/warp_beacon.egg-info}/PKG-INFO

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

warp_beacon-2.2.0/warp_beacon/__version__.py

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