warp-beacon 2.1.10__tar.gz → 2.1.12__tar.gz

{warp_beacon-2.1.10/warp_beacon.egg-info → warp_beacon-2.1.12}/PKG-INFO

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

warp_beacon-2.1.12/warp_beacon/__version__.py

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

{warp_beacon-2.1.10 → warp_beacon-2.1.12}/warp_beacon/jobs/abstract.py

@@ -1,9 +1,11 @@
 import os
-from abc import ABC, abstractmethod
+from abc import ABC
 from typing import TypedDict
 from typing_extensions import Unpack
 import uuid
 
+from pyrogram.enums import ChatType
+
 from warp_beacon.jobs import Origin
 from warp_beacon.jobs.types import JobType
 
@@ -37,6 +39,7 @@ class JobSettings(TypedDict):
 	account_switches: int
 	yt_auth: bool
 	session_validation: bool
+	chat_type: ChatType
 
 class AbstractJob(ABC):
 	job_id: uuid.UUID = None
@@ -68,6 +71,7 @@ class AbstractJob(ABC):
 	account_switches: int = 0
 	yt_auth: bool = False
 	session_validation: bool = False
+	chat_type: ChatType = None
 
 	def __init__(self, **kwargs: Unpack[JobSettings]) -> None:
 		if kwargs:

{warp_beacon-2.1.10 → warp_beacon-2.1.12}/warp_beacon/scheduler/scheduler.py

@@ -11,7 +11,7 @@ class IGScheduler(object):
 	thread = None
 	event = None
 
-	def __init__(self, downloader: "warp_beacon.scraper.AsyncDownloader") -> None:
+	def __init__(self, downloader: warp_beacon.scraper.AsyncDownloader) -> None:
 		self.downloader = downloader
 		self.event = threading.Event()
 
@@ -24,6 +24,7 @@ class IGScheduler(object):
 
 	def stop(self) -> None:
 		self.running = False
+		self.event.set()
 		if self.thread:
 			t_id = self.thread.native_id
 			logging.info("Stopping scheduler thread #'%s'", t_id)

{warp_beacon-2.1.10 → warp_beacon-2.1.12}/warp_beacon/scraper/__init__.py

@@ -15,7 +15,6 @@ from warp_beacon.jobs.download_job import DownloadJob
 from warp_beacon.jobs.upload_job import UploadJob
 from warp_beacon.jobs.types import JobType
 from warp_beacon.scraper.account_selector import AccountSelector
-from warp_beacon.scheduler.scheduler import IGScheduler
 
 import logging
 
@@ -37,8 +36,6 @@ class AsyncDownloader(object):
 		self.uploader = uploader
 		self.workers_count = workers_count
 		self.acc_selector = AccountSelector(ACC_FILE)
-		self.scheduler = IGScheduler(self)
-		self.scheduler.start()
 
 	def __del__(self) -> None:
 		self.stop_all()
@@ -329,7 +326,6 @@ class AsyncDownloader(object):
 
 	def stop_all(self) -> None:
 		self.allow_loop.value = 0
-		self.scheduler.stop()
 		for proc in self.workers:
 			if proc.is_alive():
 				logging.info("stopping process #%d", proc.pid)

{warp_beacon-2.1.10 → warp_beacon-2.1.12}/warp_beacon/scraper/account_selector.py

@@ -80,4 +80,6 @@ class AccountSelector(object):
 	
 	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'
+		if module_name not in self.accounts_meta_data:
+			return 0
 		return len(self.accounts_meta_data[module_name])

{warp_beacon-2.1.10 → warp_beacon-2.1.12}/warp_beacon/telegram/bot.py

@@ -4,7 +4,7 @@ import signal
 import uvloop
 
 from pyrogram import Client, filters
-from pyrogram.enums import ParseMode
+from pyrogram.enums import ParseMode, ChatType
 from pyrogram.handlers import MessageHandler, CallbackQueryHandler
 from pyrogram.types import Message, InputMedia, InputMediaAudio, InputMediaPhoto, InputMediaVideo, InputMediaAnimation, InputMediaDocument, InlineKeyboardButton, InlineKeyboardMarkup
 from pyrogram.errors import RPCError, FloodWait, NetworkMigrate, BadRequest, MultiMediaTooLong, MessageIdInvalid
@@ -18,6 +18,7 @@ from warp_beacon.uploader import AsyncUploader
 from warp_beacon.jobs.upload_job import UploadJob
 from warp_beacon.jobs.types import JobType
 from warp_beacon.telegram.utils import Utils
+from warp_beacon.scheduler.scheduler import IGScheduler
 
 import logging
 
@@ -29,6 +30,7 @@ class Bot(object):
 	client = None
 	handlers = None
 	placeholder = None
+	scheduler = None
 
 	def __init__(self, tg_bot_name: str, tg_token: str, tg_api_id: str, tg_api_hash: str) -> None:
 		# Enable logging
@@ -70,8 +72,11 @@ class Bot(object):
 			uploader=self.uploader
 		)
 
+		self.scheduler = IGScheduler(self.downloader)
+
 		self.downloader.start()
 		self.uploader.start()
+		self.scheduler.start()
 
 		self.handlers = Handlers(self)
 
@@ -94,9 +99,9 @@ class Bot(object):
 
 	def stop(self) -> None:
 		logging.info("Warp Beacon terminating. This may take a while ...")
+		self.scheduler.stop()
 		self.downloader.stop_all()
 		self.uploader.stop_all()
-		#self.client.stop()
 
 	async def send_text(self, chat_id: int, text: str, reply_id: int = None) -> int:
 		try:
@@ -141,12 +146,28 @@ class Bot(object):
 
 		return []
 
+	def build_signature_caption(self, job: UploadJob) -> str:
+		caption = ""
+		if job.chat_type in (ChatType.GROUP, ChatType.SUPERGROUP):
+			if job.source_usename:
+				caption += f"Requested by **@{job.source_usename}**"
+			if job.source_usename and job.url:
+				caption += " | "
+			if job.url:
+				caption += f"[Source link]({job.url})"
+
+		return caption
+
 	def build_tg_args(self, job: UploadJob) -> dict:
 		args = {}
 		if job.media_type == JobType.VIDEO:
 			if job.tg_file_id:
 				if job.placeholder_message_id:
-					args["media"] = InputMediaVideo(media=job.tg_file_id.replace(":video", ''), supports_streaming=True)
+					args["media"] = InputMediaVideo(
+						media=job.tg_file_id.replace(":video", ''),
+						supports_streaming=True,
+						caption=self.build_signature_caption(job)
+					)
 				else:
 					args["video"] = job.tg_file_id.replace(":video", '')
 			else:
@@ -157,7 +178,8 @@ class Bot(object):
 						width=job.media_info["width"],
 						height=job.media_info["height"],
 						duration=round(job.media_info["duration"]),
-						thumb=job.media_info["thumb"]
+						thumb=job.media_info["thumb"],
+						caption=self.build_signature_caption(job)
 					)
 				else:
 					args["video"] = job.local_media_path
@@ -166,18 +188,23 @@ class Bot(object):
 					args["height"] = job.media_info["height"]
 					args["duration"] = round(job.media_info["duration"])
 					args["thumb"] = job.media_info["thumb"]
+					args["caption"] = self.build_signature_caption(job)
 
 				args["file_name"] = "downloaded_via_warp_beacon_bot%s" % (os.path.splitext(job.local_media_path)[-1])
 		elif job.media_type == JobType.IMAGE:
 			if job.tg_file_id:
 				if job.placeholder_message_id:
-					args["media"] = InputMediaPhoto(media=job.tg_file_id.replace(":image", ''))
+					args["media"] = InputMediaPhoto(
+						media=job.tg_file_id.replace(":image", ''),
+						caption=self.build_signature_caption(job)
+					)
 				else:
 					args["photo"] = job.tg_file_id.replace(":image", '')
 			else:
 				if job.placeholder_message_id:
 					args["media"] = InputMediaPhoto(
-						media=job.local_media_path
+						media=job.local_media_path,
+						caption=self.build_signature_caption(job)
 					)
 				else:
 					args["photo"] = job.local_media_path
@@ -185,7 +212,8 @@ class Bot(object):
 			if job.tg_file_id:
 				if job.placeholder_message_id:
 					args["media"] = InputMediaAudio(
-						media=job.tg_file_id.replace(":audio", '')
+						media=job.tg_file_id.replace(":audio", ''),
+						caption=self.build_signature_caption(job)
 					)
 				else:
 					args["audio"] = job.tg_file_id.replace(":audio", '')
@@ -197,6 +225,7 @@ class Bot(object):
 						thumb=job.media_info["thumb"],
 						duration=round(job.media_info["duration"]),
 						title=job.canonical_name,
+						caption=self.build_signature_caption(job)
 					)
 				else:
 					args["audio"] = job.local_media_path
@@ -204,6 +233,7 @@ class Bot(object):
 					args["thumb"] = job.media_info["thumb"]
 					args["duration"] = round(job.media_info["duration"])
 					args["title"] = job.canonical_name
+					args["caption"] = self.build_signature_caption(job)
 				#args["file_name"] = "%s%s" % (job.canonical_name, os.path.splitext(job.local_media_path)[-1]),
 		elif job.media_type == JobType.ANIMATION:
 			if job.tg_file_id:
@@ -220,7 +250,8 @@ class Bot(object):
 						thumb=job.media_info["thumb"],
 						duration=round(job.media_info["duration"]),
 						width=job.media_info["width"],
-						height=job.media_info["height"]
+						height=job.media_info["height"],
+						caption=self.build_signature_caption(job)
 					)
 				else:
 					args["animation"] = job.local_media_path
@@ -228,6 +259,7 @@ class Bot(object):
 					args["height"] = job.media_info["height"]
 					args["duration"] = round(job.media_info["duration"])
 					args["thumb"] = job.media_info["thumb"]
+					args["caption"] = self.build_signature_caption(job)
 		elif job.media_type == JobType.COLLECTION:
 			if job.tg_file_id:
 				args["media"] = []
@@ -238,11 +270,11 @@ class Bot(object):
 						ctype = JobType[mtype.upper()]
 						ptr = None
 						if ctype == JobType.VIDEO:
-							ptr = InputMediaVideo(media=tg_id)
+							ptr = InputMediaVideo(media=tg_id, caption=self.build_signature_caption(job))
 						elif ctype == JobType.IMAGE:
-							ptr = InputMediaPhoto(media=tg_id)
+							ptr = InputMediaPhoto(media=tg_id, caption=self.build_signature_caption(job))
 						elif ctype == JobType.ANIMATION:
-							ptr = InputMediaAnimation(media=tg_id)
+							ptr = InputMediaAnimation(media=tg_id, caption=self.build_signature_caption(job))
 						tg_chunk.append(ptr)
 
 					args["media"].append(tg_chunk)
@@ -259,11 +291,13 @@ class Bot(object):
 								height=j.media_info["height"],
 								duration=round(j.media_info["duration"]),
 								thumb=j.media_info["thumb"],
+								caption=self.build_signature_caption(job)
 							)
 							tg_chunk.append(vid)
 						elif j.media_type == JobType.IMAGE:
 							photo = InputMediaPhoto(
-								media=j.local_media_path
+								media=j.local_media_path,
+								caption=self.build_signature_caption(job)
 							)
 							tg_chunk.append(photo)
 					mediafs.append(tg_chunk)
@@ -329,11 +363,11 @@ class Bot(object):
 							messages = await self.client.send_media_group(
 								chat_id=job.chat_id,
 								reply_to_message_id=job.message_id,
-								media=media_chunk,
+								media=media_chunk
 							)
 							sent_messages += messages
 							if job.media_collection:
-								for j, chunk in enumerate(media_chunk):
+								for j, _ in enumerate(media_chunk):
 									tg_file_id = Utils.extract_file_id(messages[j])
 									if tg_file_id:
 										job.media_collection[i][j].tg_file_id = tg_file_id
@@ -360,7 +394,7 @@ class Bot(object):
 					if retry_amount+1 >= max_retries:
 						msg = ""
 						if hasattr(e, "MESSAGE") and e.MESSAGE:
-							msg = "Telegram error: %s" % str(e.MESSAGE)
+							msg = f"Telegram error: {str(e.MESSAGE)}"
 						else:
 							msg = (f"Unknown Telegram error. Known information:\n```python\n{traceback.format_exc().strip()}```"
 									"\nPlease [create issue](https://github.com/sb0y/warp_beacon/issues) with this info and service logs.")
@@ -374,4 +408,11 @@ class Bot(object):
 		finally:
 			job.remove_files()
 
+		if job.chat_type in (ChatType.GROUP, ChatType.SUPERGROUP):
+			try:
+				await self.client.delete_messages(job.chat_id, (job.message_id,))
+			except Exception as e:
+				logging.warning("Failed to delete source message. Check bot permissions in Telegram chat settings.")
+				logging.exception(e)
+
 		return tg_file_ids

{warp_beacon-2.1.10 → warp_beacon-2.1.12}/warp_beacon/telegram/handlers.py

@@ -1,7 +1,7 @@
 from pyrogram import Client
 from pyrogram.types import Message, CallbackQuery
 from pyrogram.enums import ChatType, ParseMode
-from pyrogram.types import Chat, BotCommand
+from pyrogram.types import BotCommand
 
 from urlextract import URLExtract
 
@@ -37,7 +37,9 @@ class Handlers(object):
 				tg_file_id=d["tg_file_id"],
 				chat_id=message.chat.id,
 				media_type=JobType[d["media_type"].upper()],
-				message_id=message.id
+				message_id=message.id,
+				chat_type=message.chat.type,
+				source_username=message.from_user
 			)
 		)
 
@@ -96,7 +98,9 @@ class Handlers(object):
 								tg_file_id=",".join(tg_file_ids),
 								message_id=effective_message_id,
 								media_type=JobType.COLLECTION,
-								chat_id=chat.id
+								chat_id=chat.id,
+								chat_type=message.chat.type,
+								source_username=message.from_user
 							)
 						)
 					elif ent_len:
@@ -107,7 +111,9 @@ class Handlers(object):
 								tg_file_id=tg_file_ids.pop(),
 								message_id=effective_message_id,
 								media_type=media_type,
-								chat_id=chat.id
+								chat_id=chat.id,
+								chat_type=message.chat.type,
+								source_username=message.from_user
 							)
 						)
 				else:
@@ -141,7 +147,7 @@ class Handlers(object):
 						if not placeholder_message_id:
 							await self.bot.send_text(
 								chat_id=chat.id,
-								reply_id=effective_message_id, 
+								reply_id=effective_message_id,
 								text="Failed to create message placeholder. Please check your bot Internet connection.")
 							return
 
@@ -157,7 +163,9 @@ class Handlers(object):
 							chat_id=chat.id,
 							in_process=self.bot.uploader.is_inprocess(uniq_id),
 							uniq_id=uniq_id,
-							job_origin=origin
+							job_origin=origin,
+							source_username=message.from_user,
+							chat_type=chat.type
 						))
 						self.bot.uploader.set_inprocess(uniq_id)
 					except Exception as e:

{warp_beacon-2.1.10 → warp_beacon-2.1.12/warp_beacon.egg-info}/PKG-INFO

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

warp_beacon-2.1.10/warp_beacon/__version__.py

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