mirror of
https://github.com/Xtao-Labs/twitter2telegram.git
synced 2024-11-21 14:48:21 +00:00
fix: add invalid user
This commit is contained in:
parent
fd133a7de2
commit
476e768426
@ -4,3 +4,4 @@ rss_hub_host = ["https://rsshub.app"]
|
|||||||
cid = 11
|
cid = 11
|
||||||
tid = None
|
tid = None
|
||||||
owner = 11
|
owner = 11
|
||||||
|
logtail_token = None
|
||||||
|
@ -122,7 +122,6 @@ async def async_get_user(user_data: Dict, username: str) -> None:
|
|||||||
user_data[username] = UsernameNotFound
|
user_data[username] = UsernameNotFound
|
||||||
except Exception:
|
except Exception:
|
||||||
logs.error(f"获取 {username} 的数据失败")
|
logs.error(f"获取 {username} 的数据失败")
|
||||||
traceback.print_exc()
|
|
||||||
user_data[username] = None
|
user_data[username] = None
|
||||||
|
|
||||||
|
|
||||||
|
6
init.py
6
init.py
@ -1,10 +1,11 @@
|
|||||||
import pyrogram
|
import pyrogram
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
from defs.glover import api_id, api_hash
|
from defs.glover import api_id, api_hash, logtail_token
|
||||||
from scheduler import scheduler
|
from scheduler import scheduler
|
||||||
from logging import getLogger, INFO, ERROR, StreamHandler, basicConfig, FileHandler, Formatter
|
from logging import getLogger, INFO, ERROR, StreamHandler, basicConfig, FileHandler, Formatter
|
||||||
from coloredlogs import ColoredFormatter
|
from coloredlogs import ColoredFormatter
|
||||||
|
from logtail import LogtailHandler
|
||||||
|
|
||||||
# Enable logging
|
# Enable logging
|
||||||
logs = getLogger("T2G")
|
logs = getLogger("T2G")
|
||||||
@ -13,10 +14,13 @@ logging_handler = StreamHandler()
|
|||||||
logging_handler.setFormatter(ColoredFormatter(logging_format))
|
logging_handler.setFormatter(ColoredFormatter(logging_format))
|
||||||
file_handler = FileHandler("log.txt", mode="w", encoding="utf-8")
|
file_handler = FileHandler("log.txt", mode="w", encoding="utf-8")
|
||||||
file_handler.setFormatter(Formatter(logging_format))
|
file_handler.setFormatter(Formatter(logging_format))
|
||||||
|
logtail_handler = LogtailHandler(source_token=logtail_token)
|
||||||
root_logger = getLogger()
|
root_logger = getLogger()
|
||||||
root_logger.setLevel(ERROR)
|
root_logger.setLevel(ERROR)
|
||||||
root_logger.addHandler(logging_handler)
|
root_logger.addHandler(logging_handler)
|
||||||
root_logger.addHandler(file_handler)
|
root_logger.addHandler(file_handler)
|
||||||
|
if logtail_token:
|
||||||
|
root_logger.addHandler(logtail_handler)
|
||||||
basicConfig(level=INFO)
|
basicConfig(level=INFO)
|
||||||
logs.setLevel(INFO)
|
logs.setLevel(INFO)
|
||||||
|
|
||||||
|
@ -3,12 +3,21 @@ import re
|
|||||||
from pyrogram import filters
|
from pyrogram import filters
|
||||||
from pyrogram.types import Message
|
from pyrogram.types import Message
|
||||||
|
|
||||||
|
from defs.feed import get_user
|
||||||
from init import bot
|
from init import bot
|
||||||
|
|
||||||
from defs.sqlite import UserDB
|
from defs.sqlite import UserDB
|
||||||
from defs.glover import owner
|
from defs.glover import owner
|
||||||
|
|
||||||
|
|
||||||
|
async def check_user(username: str) -> bool:
|
||||||
|
try:
|
||||||
|
await get_user(username)
|
||||||
|
return True
|
||||||
|
except:
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
@bot.on_message(filters=filters.command("add_user") & filters.user(owner))
|
@bot.on_message(filters=filters.command("add_user") & filters.user(owner))
|
||||||
async def add_user(_, message: Message):
|
async def add_user(_, message: Message):
|
||||||
try:
|
try:
|
||||||
@ -19,6 +28,9 @@ async def add_user(_, message: Message):
|
|||||||
if UserDB.check(username):
|
if UserDB.check(username):
|
||||||
await message.reply("该用户添加过了!")
|
await message.reply("该用户添加过了!")
|
||||||
return
|
return
|
||||||
|
if not await check_user(username):
|
||||||
|
await message.reply("该用户不存在!或者 rss 服务出现问题!")
|
||||||
|
return
|
||||||
UserDB.add(username)
|
UserDB.add(username)
|
||||||
await message.reply("添加成功!")
|
await message.reply("添加成功!")
|
||||||
|
|
||||||
@ -35,6 +47,9 @@ async def add_user_regex(_, message: Message):
|
|||||||
if UserDB.check(username):
|
if UserDB.check(username):
|
||||||
await message.reply("该用户添加过了!")
|
await message.reply("该用户添加过了!")
|
||||||
return
|
return
|
||||||
|
if not await check_user(username):
|
||||||
|
await message.reply("该用户不存在!或者 rss 服务出现问题!")
|
||||||
|
return
|
||||||
UserDB.add(username)
|
UserDB.add(username)
|
||||||
await message.reply(f"添加 {username} 成功!")
|
await message.reply(f"添加 {username} 成功!")
|
||||||
|
|
||||||
|
@ -16,6 +16,6 @@ async def update_all(_, message: Message):
|
|||||||
await msg.edit("检查更新完毕!")
|
await msg.edit("检查更新完毕!")
|
||||||
|
|
||||||
|
|
||||||
@scheduler.scheduled_job("cron", minute="*/30", id="update_all")
|
@scheduler.scheduled_job("cron", minute="*/15", id="update_all")
|
||||||
async def update_all_30_minutes():
|
async def update_all_30_minutes():
|
||||||
await check_update()
|
await check_update()
|
||||||
|
@ -8,3 +8,4 @@ coloredlogs
|
|||||||
apscheduler
|
apscheduler
|
||||||
sqlitedict
|
sqlitedict
|
||||||
pydantic
|
pydantic
|
||||||
|
logtail-python
|
||||||
|
Loading…
Reference in New Issue
Block a user