Merge pull request #36 from SHABIN-K/main

add broadcast command

README.md

@@ -69,6 +69,9 @@ python3 main.py
 
 /genlink - create link for one post
 
+/users - view bot statistics
+
+/broadcast - broadcast any messages to bot users
 ```
 
 ### Variables

config.py

@@ -17,6 +17,9 @@
 #OWNER ID
 OWNER_ID = int(os.environ.get("OWNER_ID", ""))
 
+#Database 
+DB_URI = os.environ.get("DATABASE_URL", "")
+
 #force sub channel id, if you want enable force sub
 FORCE_SUB_CHANNEL = int(os.environ.get("FORCE_SUB_CHANNEL", "0"))
 

database/sql.py

@@ -0,0 +1,50 @@
+
+import os
+import threading
+from sqlalchemy import create_engine
+from sqlalchemy import Column, TEXT, Numeric
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker, scoped_session
+from config import DB_URI
+
+def start() -> scoped_session:
+    engine = create_engine(DB_URI, client_encoding="utf8")
+    BASE.metadata.bind = engine
+    BASE.metadata.create_all(engine)
+    return scoped_session(sessionmaker(bind=engine, autoflush=False))
+
+
+BASE = declarative_base()
+SESSION = start()
+
+INSERTION_LOCK = threading.RLock()
+
+class Broadcast(BASE):
+    __tablename__ = "broadcast"
+    id = Column(Numeric, primary_key=True)
+    user_name = Column(TEXT)
+
+    def __init__(self, id, user_name):
+        self.id = id
+        self.user_name = user_name
+
+Broadcast.__table__.create(checkfirst=True)
+
+
+#  Add user details -
+async def add_user(id, user_name):
+    with INSERTION_LOCK:
+        msg = SESSION.query(Broadcast).get(id)
+        if not msg:
+            usr = Broadcast(id, user_name)
+            SESSION.add(usr)
+            SESSION.commit()
+        else:
+            pass
+
+async def query_msg():
+    try:
+        query = SESSION.query(Broadcast.id).order_by(Broadcast.id)
+        return query
+    finally:
+        SESSION.close()

database/support.py

@@ -0,0 +1,21 @@
+import asyncio
+from database.sql import query_msg
+from pyrogram.errors import FloodWait
+
+async def users_info(bot):
+    users = 0
+    blocked = 0
+    identity = await query_msg()
+    for id in identity:
+        name = bool()
+        try:
+            name = await bot.send_chat_action(int(id[0]), "typing")
+        except FloodWait as e:
+            await asyncio.sleep(e.x)
+        except Exception:
+            pass
+        if bool(name):
+            users += 1
+        else:
+            blocked += 1
+    return users, blocked

plugins/channel_post.py

@@ -9,7 +9,7 @@
 from config import ADMINS, CHANNEL_ID, DISABLE_CHANNEL_BUTTON
 from helper_func import encode
 
-@Bot.on_message(filters.private & filters.user(ADMINS) & ~filters.command(['start','batch','genlink']))
+@Bot.on_message(filters.private & filters.user(ADMINS) & ~filters.command(['start','users','broadcast','batch','genlink']))
 async def channel_post(client: Client, message: Message):
     reply_text = await message.reply_text("Please Wait...!", quote = True)
     try:

plugins/start.py

@@ -1,5 +1,5 @@
 #(©)Codexbotz
-
+import os
 import asyncio
 from pyrogram import Client, filters, __version__
 from pyrogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
@@ -8,9 +8,27 @@
 from bot import Bot
 from config import ADMINS, START_MSG, OWNER_ID, CUSTOM_CAPTION, DISABLE_CHANNEL_BUTTON
 from helper_func import subscribed, encode, decode, get_messages
+from database.support import users_info
+from database.sql import add_user, query_msg
+
+
+#=====================================================================================##
+
+USERS_LIST = """<b>⭕️Total:</b>\n\n⭕️Subscribers - {}\n⭕️Blocked- {}"""
+
+WAIT_MSG = """"<b>Processing ...</b>"""
+
+REPLY_ERROR = """<code>Use this command as a replay to any telegram message with out any spaces.</code>"""
+
+
+#=====================================================================================##
+
 
 @Bot.on_message(filters.command('start') & filters.private & subscribed)
 async def start_command(client: Client, message: Message):
+    id = message.from_user.id
+    user_name = '@' + message.from_user.username if message.from_user.username else None
+    await add_user(id, user_name)
     text = message.text
     if len(text)>7:
         try:
@@ -108,3 +126,43 @@ async def not_joined(client: Client, message: Message):
         quote = True,
         disable_web_page_preview = True
     )
+
+@Bot.on_message(filters.private & filters.command('users'))
+async def subscribers_count(bot, m: Message):
+    id = m.from_user.id
+    if id not in ADMINS:
+        return
+    msg = await m.reply_text(WAIT_MSG)
+    messages = await users_info(bot)
+    active = messages[0]
+    blocked = messages[1]
+    await m.delete()
+    await msg.edit(USERS_LIST.format(active, blocked))
+
+
+
+@Bot.on_message(filters.private & filters.command('broadcast'))
+async def send_text(bot, m: Message):
+    id = m.from_user.id
+    if id not in ADMINS:
+        return
+    if (" " not in m.text) and ("broadcast" in m.text) and (m.reply_to_message is not None):
+        query = await query_msg()
+        for row in query:
+            chat_id = int(row[0])
+            try:
+                await bot.copy_message(
+                    chat_id=chat_id,
+                    from_chat_id=m.chat.id,
+                    message_id=m.reply_to_message.message_id,
+                    caption=m.caption,
+                    reply_markup=m.reply_markup
+                )
+            except FloodWait as e:
+                await asyncio.sleep(e.x)
+            except Exception:
+                pass
+    else:
+        msg = await m.reply_text(REPLY_ERROR, m.message_id)
+        await asyncio.sleep(8)
+        await msg.delete()

requirements.txt

@@ -1,3 +1,9 @@
-Pyrogram
+# --- For-Bot-Working --------- #
+pyrogram
 TgCrypto
 Pyromod
+# --- For-Database ------------ #
+sqlalchemy~=1.3.23
+psycopg2-binary
+feedparser
+