Как боту сделать рассылку в телеграмме через python?

Question

[MarkBelousov567]

У меня есть db таблица с 3 столбцами id, user_id и Status. Нажимая на /start пользователь подписывается на бота, а нажимая на /unsubscribe отписывается.
Вот код: (библиотека aiogram)<code lang="python">



# -*- coding: utf8 -*-

from aiogram import Bot, Dispatcher, executor, types
from aiogram.dispatcher import FSMContext
from aiogram.contrib.fsm_storage.memory import MemoryStorage
from aiogram.dispatcher.filters.state import State, StatesGroup
import random
from random import randint
import time

import logging
import asyncio
from datetime import datetime
from aiogram import Bot, Dispatcher, executor, types
from chart import SQLighter

from aiogram.types import ReplyKeyboardRemove, \
	ReplyKeyboardMarkup, KeyboardButton, \
	InlineKeyboardMarkup, InlineKeyboardButton


bot = Bot(token= 'token' )
dp = Dispatcher(bot, storage=MemoryStorage())

@dp.message_handler(commands=['start'])
async def process_start_command(message: types.Message):
	await message.answer_sticker(r'https://tlgrm.ru/_/stickers/061/2ac/0612acc2-f6fd-3470-83df- 
            429ee8ba3d3b/192/25.webp')
	await message.reply("Вы успешно подписаны на этого бота. ВНИМАНИЕ!!! ваш ID не будет 
            отправлен не одному человеку, на вас не будут оформленны никакие платные подписки и                         
            другие платные услуги. Но если вы не хотите быть подписанным на бота, который будет 
            скидывать вам всякие ништяки от проверенных источников, то вы можете отписаться, 
            нажав на /unsubscribe. Если же вы не отписались, то спасибо за понимание. :-)")

	if(not db.subscriber_exists(message.from_user.id)):

		db.add_subscriber(message.from_user.id)
	else:

		db.update_subscription(message.from_user.id, True)


logging.basicConfig(level=logging.INFO)
db = SQLighter('db.db')


@dp.message_handler(commands=['unsubscribe'])
async def unsubscribe(message: types.Message):
	if(not db.subscriber_exists(message.from_user.id)):
		
		await message.answer("Вы итак не подписаны.")
	else:
		await message.answer("Вы успешно отписаны.")

@dp.message_handler(commands=['go'])
async def go(message: types.Message):
	for user in db:
		await bot.send_message("...")

if __name__ == '__main__':
	executor.start_polling(dp,skip_updates=True)

Код таблицы: (chart)

import sqlite3

class SQLighter:

    def __init__(self, database):

        self.connection = sqlite3.connect(database)
        self.cursor = self.connection.cursor()

    def get_subscriptions(self, status = True):
        
        with self.connection:
            return self.cursor.execute("SELECT * FROM `subscriptions` WHERE `status` = ?", (status,)).fetchall()

    def subscriber_exists(self, user_id):

        with self.connection:
            result = self.cursor.execute('SELECT * FROM `subscriptions` WHERE `user_id` = ?', (user_id,)).fetchall()
            return bool(len(result))

    def add_subscriber(self, user_id, status = True):

        with self.connection:
            return self.cursor.execute("INSERT INTO `subscriptions` (`user_id`, `status`) VALUES(?,?)", (user_id,status))

    def update_subscription(self, user_id, status):

        with self.connection:
            return self.cursor.execute("UPDATE `subscriptions` SET `status` = ? WHERE `user_id` = ?", (status, 
                user_id))

    def close(self):

        self.connection.close()

Заранее спасибо)
</code>

Comments

[Михаил Кростелев]

Вот посмотри на свой вопрос со стороны и скажи читаемый ли тут код? Особенно учитывая, что это python-код, где важны отступы. Для правильного оформления есть соответствующий тег. Исправляй.

[MarkBelousov567]

Михаил, Код не читаемый, согласен. Но при редактировании все отступы есть. С чем это связано я не знаю.

[Михаил Кростелев]

MarkBelousov567, С тем, что код надо оформлять в соответствующий тег!

[MarkBelousov567]

Михаил, Исправил

[Alexa2007]

Если честно то непонятно почему вы задаёте настолько простой вопрос, с учётом того что вы написали такой код

[MarkBelousov567]

Alexa2007, Я знаю как отослать сообщение одному пользователю, но вот как вынуть из db ID и устроить массовую рассылку я додуматься не могу.

[Alexa2007]

MarkBelousov567,

from aiogram import Bot, Dispatcher
bot = Bot(token= 'token' )
for user in users:
bot.send_message(user, "...")

[MarkBelousov567]

Alexa2007, Не работает.
Ошибка:
NameError: name 'users' is not defined
INFO:aiogram.dispatcher.dispatcher:Stop polling...
WARNING:aiogram:Goodbye!

[Alexa2007]

MarkBelousov567, ну это понятно, я тебе код как образец набросал..

Я знаю как отослать сообщение

Ну если знаешь, почему фигню делаешь.
Пользователей необходимо из базы достать, и код у тебя для этого есть(почти готовый)
У тебя всё уже готово, осталось до ума довести. Поэтому и спрашиваю почему глупые вопросы задаешь если ТЫ этот код писал........

Answer 1

[o5a]

fetchall() в get_subscriptions в текущем виде будет возвращать вложенные списки со всеми полями. Лучше сразу возвращать только нужную информацию - список самих id

...
    def get_subscriptions(self, status = True):
        with self.connection:
            return [x[0] for x in self.cursor.execute("SELECT user_id FROM `subscriptions` WHERE `status` = ?", (status,)).fetchall()]

тогда можно просто напрямую использовать идентификаторы при отправке

@dp.message_handler(commands=['go'])
async def go(message: types.Message):
  subscriptions = db.get_subscriptions()
  for user in subscriptions:
    await bot.send_message(user, "...")

Comments

[MarkBelousov567]

Не находит чат, что делать?

[o5a]

MarkBelousov567, чтобы бот мог отвечать пользователю, тот сначала должен инициировать чат с ботом. Ну или Вы как-то не так id пользователей сохраняли.

[MarkBelousov567]

o5a, Подскажите пожалуйста, как правильно сохранять id пользователей

[Ян]

Это уже зависит где ты собрался их хранить. Обычно пользуются Redis или Mongo. От нагрузки и нюансов зависит. Если не так много пользователей можешь хранить да же в файлах.