Compare commits
No commits in common. "cd5b73e7ff8f055686309170d0b71e7e23a614eb" and "11255bbbd007781898a0c21e0157a7e911ba22b4" have entirely different histories.
cd5b73e7ff
...
11255bbbd0
159
poster.py
159
poster.py
@ -6,37 +6,23 @@ from threading import Thread
|
||||
from time import time
|
||||
from traceback import format_exc
|
||||
from pathlib import Path
|
||||
from argparse import ArgumentParser
|
||||
|
||||
from modules.logging import logWrite
|
||||
from modules.utils import configGet, jsonLoad, jsonSave, killProc, locale
|
||||
|
||||
# Args =====================================================================================================================================
|
||||
parser = ArgumentParser(
|
||||
prog = "Telegram Poster",
|
||||
description = "Bot for posting some of your stuff and also receiving submissions."
|
||||
)
|
||||
|
||||
parser.add_argument("-m", "--move-sent", action="store_true")
|
||||
parser.add_argument("-c", "--cleanup", action="store_true")
|
||||
parser.add_argument("--confirm", action="store_true")
|
||||
parser.add_argument("-i", "--cleanup-index", action="store_true")
|
||||
parser.add_argument("-n", "--norun", action="store_true")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.move_sent:
|
||||
if "--move-sent" in argv:
|
||||
for entry in jsonLoad(configGet("index", "locations"))["sent"]:
|
||||
try:
|
||||
move(configGet("queue", "locations")+sep+entry, configGet("sent", "locations")+sep+entry)
|
||||
except FileNotFoundError:
|
||||
logWrite(locale("move_sent_doesnt_exist", "console", locale=configGet("locale")).format(entry))
|
||||
logWrite(locale("move_sent_doesnt_exist", "console", locale=configGet("locale_log")).format(entry))
|
||||
except Exception as exp:
|
||||
logWrite(locale("move_sent_doesnt_exception", "console", locale=configGet("locale")).format(entry, exp))
|
||||
logWrite(locale("move_sent_completed", "console", locale=configGet("locale")))
|
||||
logWrite(locale("move_sent_doesnt_exception", "console", locale=configGet("locale_log")).format(entry, exp))
|
||||
logWrite(locale("move_sent_completed", "console", locale=configGet("locale_log")))
|
||||
|
||||
if args.cleanup:
|
||||
if args.confirm:
|
||||
if "--cleanup" in argv:
|
||||
if "--confirm" in argv:
|
||||
index = jsonLoad(configGet("index", "locations"))
|
||||
for entry in index["sent"]:
|
||||
try:
|
||||
@ -49,22 +35,22 @@ if args.cleanup:
|
||||
except FileNotFoundError:
|
||||
pass
|
||||
except Exception as exp:
|
||||
logWrite(locale("cleanup_exception", "console", locale=configGet("locale")).format(entry, exp))
|
||||
logWrite(locale("cleanup_exception", "console", locale=configGet("locale_log")).format(entry, exp))
|
||||
jsonSave(index, jsonLoad(configGet("index", "locations")))
|
||||
logWrite(locale("cleanup_completed", "console", locale=configGet("locale")))
|
||||
logWrite(locale("cleanup_completed", "console", locale=configGet("locale_log")))
|
||||
else:
|
||||
logWrite(locale("cleanup_unathorized", "console", locale=configGet("locale")))
|
||||
logWrite(locale("cleanup_unathorized", "console", locale=configGet("locale_log")))
|
||||
|
||||
if args.cleanup_index:
|
||||
if args.confirm:
|
||||
if "--cleanup-index" in argv:
|
||||
if "--confirm" in argv:
|
||||
index = jsonLoad(configGet("index", "locations"))
|
||||
index["sent"] = []
|
||||
jsonSave(index, jsonLoad(configGet("index", "locations")))
|
||||
logWrite(locale("cleanup_index_completed", "console", locale=configGet("locale")))
|
||||
logWrite(locale("cleanup_index_completed", "console", locale=configGet("locale_log")))
|
||||
else:
|
||||
logWrite(locale("cleanup_index_unathorized", "console", locale=configGet("locale")))
|
||||
logWrite(locale("cleanup_index_unathorized", "console", locale=configGet("locale_log")))
|
||||
|
||||
if args.norun:
|
||||
if "--norun" in argv:
|
||||
logWrite(locale("passed_norun", "console", locale=configGet("locale_log")))
|
||||
exit()
|
||||
#===========================================================================================================================================
|
||||
@ -72,13 +58,13 @@ if args.norun:
|
||||
|
||||
# Import ===================================================================================================================================
|
||||
try:
|
||||
import schedule # type: ignore
|
||||
from pyrogram import Client, filters, idle # type: ignore
|
||||
from pyrogram.types import ChatPermissions, ReplyKeyboardMarkup, InlineKeyboardMarkup, InlineKeyboardButton, BotCommand, BotCommandScopeChat # type: ignore
|
||||
from pyrogram.raw.types import UpdateChannelMessageForwards, InputChannel, InputPeerChannel
|
||||
from pyrogram.raw.functions.stats import GetMessagePublicForwards
|
||||
import schedule
|
||||
from pyrogram.sync import idle
|
||||
from pyrogram.client import Client
|
||||
from pyrogram import filters
|
||||
from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, BotCommand, BotCommandScopeChat
|
||||
except ModuleNotFoundError:
|
||||
print(locale("deps_missing", "console", locale=configGet("locale")), flush=True)
|
||||
print(locale("deps_missing", "console", locale=configGet("locale_log")), flush=True)
|
||||
exit()
|
||||
#===========================================================================================================================================
|
||||
|
||||
@ -152,8 +138,11 @@ def send_content():
|
||||
if len(list_queue) > 0:
|
||||
candidate_file = choice(list_queue)
|
||||
candidate = configGet("queue", "locations")+sep+candidate_file
|
||||
if len(list_queue) <= 10:
|
||||
if configGet("error", "reports"):
|
||||
app.send_message(configGet("admin"), locale("post_low", "message", locale=configGet("locale"))) # type: ignore
|
||||
else:
|
||||
logWrite(locale("post_empty", "console", locale=configGet("locale")))
|
||||
logWrite(locale("post_empty", "console", locale=configGet("locale_log")))
|
||||
if configGet("error", "reports"):
|
||||
app.send_message(configGet("admin"), locale("post_empty", "message", locale=configGet("locale"))) # type: ignore
|
||||
return
|
||||
@ -171,38 +160,44 @@ def send_content():
|
||||
else:
|
||||
caption = caption
|
||||
|
||||
if ext_type == "photo": # type: ignore
|
||||
try:
|
||||
if ext_type == "photo": # type: ignore
|
||||
|
||||
if configGet("enabled", "caption"):
|
||||
if configGet("link", "caption") != None:
|
||||
sent = app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
if configGet("enabled", "caption"):
|
||||
if configGet("link", "caption") != None:
|
||||
sent = app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
else:
|
||||
sent = app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
else:
|
||||
sent = app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
else:
|
||||
sent = app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
|
||||
elif ext_type == "video": # type: ignore
|
||||
elif ext_type == "video": # type: ignore
|
||||
|
||||
if configGet("enabled", "caption"):
|
||||
if configGet("link", "caption") != None:
|
||||
sent = app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
if configGet("enabled", "caption"):
|
||||
if configGet("link", "caption") != None:
|
||||
sent = app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
else:
|
||||
sent = app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
else:
|
||||
sent = app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
else:
|
||||
sent = app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting")) # type: ignore
|
||||
|
||||
else:
|
||||
else:
|
||||
return
|
||||
except Exception as exp:
|
||||
logWrite(locale("post_exception", "console", locale=configGet("locale_log")).format(str(exp), format_exc()))
|
||||
if configGet("error", "reports"):
|
||||
app.send_message(configGet("admin"), locale("post_exception", "message", locale=configGet("locale")).format(exp, traceback.format_exc())) # type: ignore
|
||||
return
|
||||
|
||||
index["sent"].append(candidate_file)
|
||||
index["last_id"] = sent.id
|
||||
index["last_id"] = sent.id # type: ignore
|
||||
|
||||
jsonSave(index, configGet("index", "locations"))
|
||||
|
||||
if configGet("move_sent", "posting"):
|
||||
move(candidate, configGet("sent", "locations")+sep+candidate_file)
|
||||
|
||||
logWrite(locale("post_sent", "console", locale=configGet("locale")).format(candidate, ext_type, str(configGet("channel", "posting")), caption.replace("\n", "%n"), str(configGet("silent", "posting")))) # type: ignore
|
||||
logWrite(locale("post_sent", "console", locale=configGet("locale_log")).format(candidate, ext_type, str(configGet("channel", "posting")), caption.replace("\n", "%n"), str(configGet("silent", "posting")))) # type: ignore
|
||||
|
||||
if configGet("sent", "reports"):
|
||||
app.send_message(configGet("admin"), f"Posted `{candidate_file}`", disable_web_page_preview=True, reply_markup=InlineKeyboardMarkup([
|
||||
@ -210,10 +205,10 @@ def send_content():
|
||||
])) # type: ignore
|
||||
|
||||
except Exception as exp:
|
||||
logWrite(locale("post_exception", "console", locale=configGet("locale")).format(str(exp), format_exc()))
|
||||
logWrite(locale("post_exception", "console", locale=configGet("locale_log")).format(str(exp), format_exc()))
|
||||
if configGet("error", "reports"):
|
||||
app.send_message(configGet("admin"), locale("post_exception", "message", locale=configGet("locale")).format(exp, traceback.format_exc())) # type: ignore
|
||||
pass
|
||||
return
|
||||
|
||||
|
||||
# Work in progress
|
||||
@ -241,7 +236,7 @@ if configGet("submit", "mode"):
|
||||
def cmd_kill(app, msg):
|
||||
|
||||
if msg.from_user.id == configGet("admin"):
|
||||
logWrite(locale("shutdown", "console", locale=configGet("locale")).format(str(pid)))
|
||||
logWrite(locale("shutdown", "console", locale=configGet("locale_log")).format(str(pid)))
|
||||
msg.reply_text(locale("shutdown", "message", locale=configGet("locale")).format(str(pid)))
|
||||
killProc(pid)
|
||||
|
||||
@ -287,14 +282,17 @@ if configGet("submit", "mode"):
|
||||
|
||||
if msg.document != None:
|
||||
if msg.document.mime_type not in configGet("mime_types", "submission"):
|
||||
logWrite(locale("sub_mime_not_allowed", "console", locale=configGet("locale_log")).format(str(msg.from_user.id), msg.document.mime_type))
|
||||
msg.reply_text(locale("mime_not_allowed", "message", locale=user_locale), quote=True)
|
||||
return
|
||||
if msg.document.file_size > configGet("file_size", "submission"):
|
||||
logWrite(locale("sub_document_too_large", "console", locale=configGet("locale_log")).format(str(msg.from_user.id), str(msg.document.file_size), str(configGet("file_size", "submission"))))
|
||||
msg.reply_text(locale("document_too_large", "message", locale=user_locale).format(str(configGet("file_size", "submission")/1024/1024)), quote=True)
|
||||
return
|
||||
|
||||
if msg.video != None:
|
||||
if msg.video.file_size > configGet("file_size", "submission"):
|
||||
logWrite(locale("sub_document_too_large", "console", locale=configGet("locale_log")).format(str(msg.from_user.id), str(msg.document.file_size), str(configGet("file_size", "submission"))))
|
||||
msg.reply_text(locale("document_too_large", "message", locale=user_locale).format(str(configGet("file_size", "submission")/1024/1024)), quote=True)
|
||||
return
|
||||
|
||||
@ -329,25 +327,28 @@ if configGet("submit", "mode"):
|
||||
if msg.from_user.phone_number != None:
|
||||
caption += f" ({msg.from_user.phone_number})"
|
||||
|
||||
msg.copy(configGet("admin"), caption=caption, reply_markup=InlineKeyboardMarkup(buttons))
|
||||
|
||||
if msg.from_user.id != configGet("admin"):
|
||||
buttons += [
|
||||
[
|
||||
InlineKeyboardButton(text=locale("sub_block", "button", locale=configGet("locale")), callback_data=f"sub_block_{msg.from_user.id}")
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(text=locale("sub_unblock", "button", locale=configGet("locale")), callback_data=f"sub_unblock_{msg.from_user.id}")
|
||||
]
|
||||
# [
|
||||
# InlineKeyboardButton(text=locale("sub_unblock", "button", locale=configGet("locale")), callback_data=f"sub_unblock_{msg.from_user.id}")
|
||||
# ]
|
||||
]
|
||||
|
||||
msg.reply_text(locale("sub_sent", "message", locale=user_locale), quote=True)
|
||||
subLimit(msg.from_user)
|
||||
|
||||
logWrite(locale("sub_received", "console", locale=configGet("locale_log")).format(str(msg.from_user.id), str(msg.caption)))
|
||||
msg.copy(configGet("admin"), caption=caption, reply_markup=InlineKeyboardMarkup(buttons))
|
||||
|
||||
else:
|
||||
logWrite(locale("sub_cooldown", "console", locale=configGet("locale_log")).format(str(msg.from_user.id)))
|
||||
msg.reply_text(locale("sub_cooldown", "message", locale=user_locale).format(str(configGet("timeout", "submission"))))
|
||||
|
||||
except AttributeError:
|
||||
logWrite(f"from_user in function get_submission does not seem to contain id")
|
||||
logWrite(locale("sub_received", "console", locale=configGet("locale_log")))
|
||||
|
||||
@app.on_callback_query(filters.regex("sub_yes_[\s\S]*_[\s\S]*")) # type: ignore
|
||||
def callback_query_yes(app, clb): # type: ignore
|
||||
@ -356,18 +357,26 @@ def callback_query_yes(app, clb): # type: ignore
|
||||
try:
|
||||
submission = app.get_messages(int(fullclb[2]), int(fullclb[3]))
|
||||
except:
|
||||
logWrite(locale("sub_msg_unavail", "console", locale=configGet("locale_log")).format(fullclb[3], fullclb[2]))
|
||||
clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True)
|
||||
return
|
||||
try:
|
||||
logWrite(locale("sub_media_downloading", "console", locale=configGet("locale_log")).format(fullclb[3], fullclb[2]))
|
||||
media = app.download_media(submission, file_name=configGet("queue", "locations")+sep)
|
||||
if clb.data.endswith("_caption"):
|
||||
index = jsonLoad(configGet("index", "locations"))
|
||||
index["captions"][Path(media).name] = submission.caption
|
||||
jsonSave(index, configGet("index", "locations"))
|
||||
logWrite(locale("sub_media_downloaded", "console", locale=configGet("locale_log")).format(fullclb[3], fullclb[2]))
|
||||
except:
|
||||
logWrite(locale("sub_media_unavail", "console", locale=configGet("locale_log")).format(fullclb[3], fullclb[2]))
|
||||
clb.answer(text=locale("sub_media_unavail", "message", locale=user_locale), show_alert=True)
|
||||
return
|
||||
logWrite(locale("sub_accepted", "console", locale=configGet("locale_log")).format(fullclb[3], fullclb[2]))
|
||||
submission.reply_text(locale("sub_yes", "message", locale=submission.from_user.language_code), quote=True)
|
||||
edited_markup = clb.message.reply_markup.inline_keyboard
|
||||
edited_markup[0] = [InlineKeyboardButton(text=locale("accepted", "button", locale=configGet("locale")), callback_data=f"sub_done")]
|
||||
clb.message.edit(text=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
clb.answer(text=locale("sub_yes", "callback", locale=user_locale).format(fullclb[2]), show_alert=True)
|
||||
|
||||
@app.on_callback_query(filters.regex("sub_no_[\s\S]*_[\s\S]*")) # type: ignore
|
||||
@ -377,26 +386,44 @@ def callback_query_no(app, clb): # type: ignore
|
||||
try:
|
||||
submission = app.get_messages(int(fullclb[2]), int(fullclb[3]))
|
||||
except:
|
||||
logWrite(locale("sub_msg_unavail", "console", locale=configGet("locale_log")).format(fullclb[3], fullclb[2]))
|
||||
clb.answer(text=locale("sub_msg_unavail", "message", locale=user_locale), show_alert=True)
|
||||
return
|
||||
logWrite(locale("sub_declined", "console", locale=configGet("locale_log")).format(fullclb[3], fullclb[2]))
|
||||
submission.reply_text(locale("sub_no", "message", locale=submission.from_user.language_code), quote=True)
|
||||
edited_markup = clb.message.reply_markup.inline_keyboard
|
||||
edited_markup[0] = [InlineKeyboardButton(text=locale("declined", "button", locale=configGet("locale")), callback_data=f"sub_done")]
|
||||
clb.message.edit(text=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
clb.answer(text=locale("sub_no", "callback", locale=user_locale).format(fullclb[2]), show_alert=True)
|
||||
|
||||
@app.on_callback_query(filters.regex("sub_block_[\s\S]*")) # type: ignore
|
||||
def callback_query_block(app, clb): # type: ignore
|
||||
fullclb = clb.data.split("_")
|
||||
user_locale = clb.from_user.language_code
|
||||
app.send_message(int(fullclb[2]), locale("sub_msg_unavail", "message", locale=configGet("locale")))
|
||||
app.send_message(int(fullclb[2]), locale("sub_blocked", "message", locale=configGet("locale")))
|
||||
subBlock(int(fullclb[2]))
|
||||
logWrite(locale("sub_blocked", "console", locale=configGet("locale_log")).format(fullclb[2]))
|
||||
edited_markup = clb.message.reply_markup.inline_keyboard
|
||||
edited_markup[1] = [InlineKeyboardButton(text=locale("sub_unblock", "button", locale=configGet("locale")), callback_data=f"sub_unblock_{fullclb[2]}")]
|
||||
clb.message.edit(text=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
clb.answer(text=locale("sub_block", "callback", locale=user_locale).format(fullclb[2]), show_alert=True)
|
||||
|
||||
@app.on_callback_query(filters.regex("sub_unblock_[\s\S]*")) # type: ignore
|
||||
def callback_query_unblock(app, clb): # type: ignore
|
||||
fullclb = clb.data.split("_")
|
||||
user_locale = clb.from_user.language_code
|
||||
app.send_message(int(fullclb[2]), locale("sub_msg_unavail", "message", locale=configGet("locale")))
|
||||
app.send_message(int(fullclb[2]), locale("sub_unblocked", "message", locale=configGet("locale")))
|
||||
subUnblock(int(fullclb[2]))
|
||||
logWrite(locale("sub_unblocked", "console", locale=configGet("locale_log")).format(fullclb[2]))
|
||||
edited_markup = clb.message.reply_markup.inline_keyboard
|
||||
edited_markup[1] = [InlineKeyboardButton(text=locale("sub_block", "button", locale=configGet("locale")), callback_data=f"sub_block_{fullclb[2]}")]
|
||||
clb.message.edit(text=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
clb.answer(text=locale("sub_unblock", "callback", locale=user_locale).format(fullclb[2]), show_alert=True)
|
||||
|
||||
@app.on_callback_query(filters.regex("sub_done")) # type: ignore
|
||||
def callback_query_block(app, clb): # type: ignore
|
||||
user_locale = clb.from_user.language_code
|
||||
clb.answer(text=locale("sub_done", "callback", locale=user_locale), show_alert=True)
|
||||
#===========================================================================================================================================
|
||||
|
||||
# Work in progress
|
||||
@ -424,23 +451,23 @@ if configGet("post", "mode"):
|
||||
except:
|
||||
pass
|
||||
except Exception as exp:
|
||||
logWrite(locale("exception_occured", "console", locale=configGet("locale")).format(exp))
|
||||
logWrite(locale("exception_occured", "console", locale=configGet("locale_log")).format(exp))
|
||||
except KeyboardInterrupt:
|
||||
logWrite(locale("keyboard_interrupt", "console", locale=configGet("locale")))
|
||||
logWrite(locale("keyboard_interrupt", "console", locale=configGet("locale_log")))
|
||||
if configGet("shutdown", "reports"):
|
||||
app.send_message(configGet("admin"), locale("shutdown", "message", locale=configGet("locale")).format(str(pid))) # type: ignore
|
||||
killProc(pid)
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
logWrite(locale("startup", "console", locale=configGet("locale")).format(str(pid)))
|
||||
logWrite(locale("startup", "console", locale=configGet("locale_log")).format(str(pid)))
|
||||
|
||||
app.start() # type: ignore
|
||||
if configGet("startup", "reports"):
|
||||
app.send_message(configGet("admin"), locale("startup", "message", locale=configGet("locale")).format(str(pid))) # type: ignore
|
||||
|
||||
if configGet("post", "mode"):
|
||||
t = Thread(target=background_task)
|
||||
t = Thread(target=background_task) # type: ignore
|
||||
t.start()
|
||||
|
||||
if configGet("submit", "mode"):
|
||||
@ -470,6 +497,6 @@ if __name__ == "__main__":
|
||||
idle()
|
||||
|
||||
app.send_message(configGet("admin"), locale("shutdown", "message", locale=configGet("locale")).format(str(pid))) # type: ignore
|
||||
logWrite(locale("shutdown", "console", locale=configGet("locale")).format(str(pid)))
|
||||
logWrite(locale("shutdown", "console", locale=configGet("locale_log")).format(str(pid)))
|
||||
|
||||
killProc(pid)
|
||||
Reference in New Issue
Block a user