Commited backup

KeyZenD/modules/notes.py

@@ -0,0 +1,138 @@
+#	Friendly Telegram (telegram userbot)
+#	Copyright (C) 2018-2019 The Authors
+
+#	This program is free software: you can redistribute it and/or modify
+#	it under the terms of the GNU Affero General Public License as published by
+#	the Free Software Foundation, either version 3 of the License, or
+#	(at your option) any later version.
+
+#	This program is distributed in the hope that it will be useful,
+#	but WITHOUT ANY WARRANTY; without even the implied warranty of
+#	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#	GNU Affero General Public License for more details.
+
+#	You should have received a copy of the GNU Affero General Public License
+#	along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+import logging
+
+from .. import loader, utils
+
+logger = logging.getLogger("friendly-telegram.modules.notes")
+
+
+@loader.tds
+class NotesMod(loader.Module):
+	"""Stores global notes (aka snips)"""
+	strings = {"name": "Notes",
+			   "what_note": "<b>Какую заметку нужно показать?</b>",
+			   "no_note": "<b>Заметка не найдена</b>",
+			   "save_what": "<b>А что сохранить?</b>",
+			   "what_name": "<b>А как будет называться заметка?</b>",
+			   "saved": "<b>Заметка сохранена как:</b> <code>{}</code>",
+			   "notes_header": "<b>Сохранённые заметки:</b>\n\n",
+			   "notes_item": "<b>▷</b> <code>{}</code>",
+			   "delnote_args": "<b>А какую заметку нужно удалить?</b>",
+			   "delnote_done": "<b>Заметка удалена!</b>",
+			   "delnotes_none": "<b>А заметок-то нету...</b>",
+			   "delnotes_done": "<b>ВСЕ ЗАМЕТКИ УДАЛЕНЫ</b>",
+			   "notes_none": "<b>А заметок-то нету...</b>"}
+	
+	
+	async def findnotecmd(self, message):
+		"""Gets the note specified"""
+		args = utils.get_args(message)
+		if not args:
+			await utils.answer(message, self.strings("what_note", message))
+			return
+		asset_id = self._db.get("friendly-telegram.modules.notes", "notes", {}).get(args[0], None)
+		logger.debug(asset_id)
+		if asset_id is not None:
+			asset = await self._db.fetch_asset(asset_id)
+		else:
+			asset = None
+		if asset is None:
+			self.del_note(args[0])
+			await utils.answer(message, self.strings("no_note", message))
+			return
+		link = "https://t.me/c/{}/{}".format(asset.chat.id, asset.id)
+		await message.edit(f'<b>Заметка</b> "<code>{args[0]}</code>" <a href="{link}">находится здесь.</a>')
+		
+
+
+	async def notecmd(self, message):
+		"""Gets the note specified"""
+		args = utils.get_args(message)
+		if not args:
+			await utils.answer(message, self.strings("what_note", message))
+			return
+		asset_id = self._db.get("friendly-telegram.modules.notes", "notes", {}).get(args[0], None)
+		logger.debug(asset_id)
+		if asset_id is not None:
+			asset = await self._db.fetch_asset(asset_id)
+		else:
+			asset = None
+		if asset is None:
+			self.del_note(args[0])
+			await utils.answer(message, self.strings("no_note", message))
+			return
+		await message.delete()
+		await message.client.send_message(message.to_id, await self._db.fetch_asset(asset_id), reply_to=await message.get_reply_message())
+
+	async def delallnotescmd(self, message):
+		"""Deletes all the saved notes"""
+		if not self._db.get("friendly-telegram.modules.notes", "notes", {}):
+			await utils.answer(message, self.strings("delnotes_none", message))
+			return
+		self._db.get("friendly-telegram.modules.notes", "notes", {}).clear()
+		await utils.answer(message, self.strings("delnotes_done", message))
+
+	async def savecmd(self, message):
+		"""Save a new note. Must be used in reply with one parameter (note name)"""
+		args = utils.get_args(message)
+		if not args:
+			await utils.answer(message, self.strings("what_name", message))
+			return
+		if not message.is_reply:
+			if len(args) < 2:
+				await utils.answer(message, self.strings("save_what", message))
+				return
+			else:
+				message.entities = None
+				message.message = args[1]
+				target = message
+				logger.debug(target.message)
+		else:
+			target = await message.get_reply_message()
+		asset_id = await self._db.store_asset(target)
+		self._db.set("friendly-telegram.modules.notes", "notes", {**self._db.get("friendly-telegram.modules.notes", "notes", {}), args[0]: asset_id})
+		await utils.answer(message, str(self.strings("saved", message)).format(args[0]))
+
+	async def delnotecmd(self, message):
+		"""Deletes a note, specified by note name"""
+		args = utils.get_args(message)
+		if not args:
+			await utils.answer(message, self.strings("delnote_args", message))
+		self.del_note(args[0])
+		await utils.answer(message, self.strings("delnote_done", message))
+
+	def del_note(self, note):
+		old = self._db.get("friendly-telegram.modules.notes", "notes", {})
+		try:
+			del old[note]
+		except KeyError:
+			pass
+		else:
+			self._db.set("friendly-telegram.modules.notes", "notes", old)
+
+	async def notescmd(self, message):
+		"""List the saved notes"""
+		if not self._db.get("friendly-telegram.modules.notes", "notes", {}):
+			await utils.answer(message, self.strings("notes_none", message))
+			return
+		await utils.answer(message, self.strings("notes_header", message)
+						   + "\n".join(self.strings("notes_item", message).format(key)
+						   for key in self._db.get("friendly-telegram.modules.notes", "notes", {})))
+
+	async def client_ready(self, client, db):
+		self._db = db