warp-beacon 2.2.7__tar.gz → 2.2.9__tar.gz

{warp_beacon-2.2.7/warp_beacon.egg-info → warp_beacon-2.2.9}/PKG-INFO

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

warp_beacon-2.2.9/warp_beacon/__version__.py

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

{warp_beacon-2.2.7 → warp_beacon-2.2.9}/warp_beacon/storage/__init__.py

@@ -65,7 +65,13 @@ class Storage(object):
 			logging.debug("uniq_id to search is '%s'", uniq_id)
 			cursor = self.db.find({"uniq_id": uniq_id})
 			for document in cursor:
-				ret.append({"uniq_id": document["uniq_id"], "tg_file_id": document["tg_file_id"], "media_type": document["media_type"]})
+				ret.append(
+				{
+					"uniq_id": document["uniq_id"],
+					"tg_file_id": document["tg_file_id"],
+					"media_type": document["media_type"],
+					"canonical_name": document.get("canonical_name")
+				})
 		except Exception as e:
 			logging.error("Error occurred while trying to read from the database!")
 			logging.exception(e)
@@ -79,7 +85,7 @@ class Storage(object):
 	def db_lookup_id(self, uniq_id: str) -> list[dict]:
 		return self.db_find(uniq_id)
 	
-	def add_media(self, tg_file_ids: list[str], media_url: str, media_type: str, origin: str) -> list[int]:
+	def add_media(self, tg_file_ids: list[str], media_url: str, media_type: str, origin: str, canonical_name: str = "") -> list[int]:
 		uniq_id = self.compute_uniq(media_url)
 		media_ids = []
 		for tg_file_id in tg_file_ids:
@@ -89,7 +95,14 @@ class Storage(object):
 			if self.db_lookup_id(uniq_id):
 				logging.info("Detected existing uniq_id, skipping storage write operation")
 				continue
-			media_ids += str(self.db.insert_one({"uniq_id": uniq_id, "media_type": media_type, "tg_file_id": tg_file_id, "origin": origin}).inserted_id)
+			media_ids += str(self.db.insert_one(
+			{
+				"uniq_id": uniq_id,
+				"media_type": media_type,
+				"tg_file_id": tg_file_id,
+				"origin": origin,
+				"canonical_name": canonical_name
+			}).inserted_id)
 
 		return media_ids
 	

{warp_beacon-2.2.7 → warp_beacon-2.2.9}/warp_beacon/telegram/bot.py

@@ -148,13 +148,17 @@ class Bot(object):
 
 	def build_signature_caption(self, job: UploadJob) -> str:
 		caption = ""
+		if job.canonical_name:
+			caption = f"**{job.canonical_name}**"
 		if job.chat_type in (ChatType.GROUP, ChatType.SUPERGROUP):
+			if job.canonical_name:
+				caption += "\n—\n"
 			if job.source_username:
 				caption += f"Requested by **@{job.source_username}**"
 			if job.source_username and job.url:
 				caption += " | "
 			if job.url:
-				caption += f"[Source link]({job.url})"
+				caption += f"[source link]({job.url})"
 
 		return caption
 
@@ -367,7 +371,7 @@ class Bot(object):
 						for i, media_chunk in enumerate(col_job_args["media"]):
 							messages = await self.client.send_media_group(
 								chat_id=job.chat_id,
-								reply_to_message_id=job.message_id,
+								#reply_to_message_id=job.message_id,
 								media=media_chunk
 							)
 							sent_messages += messages

{warp_beacon-2.2.7 → warp_beacon-2.2.9}/warp_beacon/telegram/handlers.py

@@ -105,8 +105,8 @@ class Handlers(object):
 							)
 						)
 					elif ent_len:
-						logging.info(entities[0])
 						media_type = JobType[entities[0]["media_type"].upper()]
+						canonical_name = entities[0]["canonical_name"]
 						await self.bot.upload_job(
 							UploadJob(
 								url=url,
@@ -115,7 +115,8 @@ class Handlers(object):
 								media_type=media_type,
 								chat_id=chat.id,
 								chat_type=message.chat.type,
-								source_username=Utils.extract_message_author(message)
+								source_username=Utils.extract_message_author(message),
+								canonical_name=canonical_name
 							)
 						)
 				else:
@@ -132,9 +133,21 @@ class Handlers(object):
 								if job.media_type == JobType.COLLECTION and job.save_items:
 									for chunk in job.media_collection:
 										for i in chunk:
-											self.storage.add_media(tg_file_ids=[i.tg_file_id], media_url=i.effective_url, media_type=i.media_type.value, origin=origin.value)
+											self.storage.add_media(
+												tg_file_ids=[i.tg_file_id],
+												media_url=i.effective_url,
+												media_type=i.media_type.value,
+												origin=job.job_origin.value,
+												canonical_name=job.canonical_name
+											)
 								else:
-									self.storage.add_media(tg_file_ids=[','.join(tg_file_ids)], media_url=job.url, media_type=job.media_type.value, origin=origin.value)
+									self.storage.add_media(
+										tg_file_ids=[','.join(tg_file_ids)],
+										media_url=job.url,
+										media_type=job.media_type.value,
+										origin=job.job_origin.value,
+										canonical_name=job.canonical_name
+									)
 						except Exception as e:
 							logging.error("Exception occurred while performing upload callback!")
 							logging.exception(e)

{warp_beacon-2.2.7 → warp_beacon-2.2.9/warp_beacon.egg-info}/PKG-INFO

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

warp_beacon-2.2.7/warp_beacon/__version__.py

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