Изменён парсинг конфига #10
10
bot.py
10
bot.py
@ -22,11 +22,11 @@ logging.basicConfig(
|
||||
)
|
||||
logger = logging.getLogger("Bot")
|
||||
|
||||
WEBAPP_HOST = config.bot("ip")
|
||||
WEBAPP_PORT = config.bot("port")
|
||||
WEBAPP_HOST = config.ip
|
||||
WEBAPP_PORT = config.port
|
||||
|
||||
WEBHOOK_HOST = f'http://{WEBAPP_HOST}:{WEBAPP_PORT}'
|
||||
WEBHOOK_PATH = f'/bot{config.bot("token")}/'
|
||||
WEBHOOK_PATH = f'/bot{config.token}/'
|
||||
WEBHOOK_URL = f"{WEBHOOK_HOST}{WEBHOOK_PATH}"
|
||||
|
||||
|
||||
@ -40,7 +40,7 @@ async def on_shutdown(dp):
|
||||
|
||||
|
||||
def main() -> None:
|
||||
if config.logging_user:
|
||||
if config.enable_logging:
|
||||
logger.info("Logging enabled!")
|
||||
else:
|
||||
logger.info("Logging disabled!")
|
||||
@ -50,7 +50,7 @@ def main() -> None:
|
||||
asyncio.set_event_loop(loop)
|
||||
loop.create_task(scheduler())
|
||||
|
||||
if config.bot("use_webhook").lower() in ['t', 'true', '1', 'yes', 'y']:
|
||||
if config.use_webhook.lower() in ['t', 'true', '1', 'yes', 'y']:
|
||||
executor.start_webhook(
|
||||
dispatcher=dp,
|
||||
loop=loop,
|
||||
|
@ -1,32 +1,28 @@
|
||||
from configparser import ConfigParser
|
||||
|
||||
from .module import Config
|
||||
from easydict import EasyDict as edict
|
||||
|
||||
|
||||
CONFIG_FILE = 'config.ini'
|
||||
|
||||
|
||||
class Configure(Config):
|
||||
class Configure:
|
||||
|
||||
def __init__(self):
|
||||
|
||||
self.config = ConfigParser()
|
||||
config = ConfigParser()
|
||||
config.read(CONFIG_FILE)
|
||||
self.data = dict()
|
||||
self.__readconfig()
|
||||
|
||||
def __readconfig(self):
|
||||
self.config.read(CONFIG_FILE)
|
||||
for section in self.config.sections():
|
||||
for section in config.sections():
|
||||
self.data[section] = dict()
|
||||
|
||||
for (key, value) in self.config.items(section):
|
||||
for key, value in config.items(section):
|
||||
self.data[section][key] = value
|
||||
|
||||
def bot(self, key):
|
||||
return self.data["Bot"][key]
|
||||
|
||||
def db(self, key):
|
||||
return self.data["DataBase"][key]
|
||||
|
||||
def anons(self, key):
|
||||
return self.data["announcements"][key]
|
||||
|
||||
|
||||
def __getattr__(self, name):
|
||||
for key in self.data.keys():
|
||||
if name not in self.data[key]:
|
||||
continue
|
||||
return self.data[key][name]
|
||||
raise NameError("Config options not found!")
|
||||
|
@ -1,54 +0,0 @@
|
||||
class Config():
|
||||
|
||||
@property
|
||||
def config_folder(self):
|
||||
return self.config.get("Docs_Settings", "Config_folder").rstrip("/")
|
||||
|
||||
@property
|
||||
def documentid(self):
|
||||
return self.config.get("Docs_Settings", 'Document_ID')
|
||||
|
||||
@property
|
||||
def token_file(self):
|
||||
file = self.config.get("Docs_Settings", "token_file")
|
||||
return (self.config_folder + "/" + file)
|
||||
|
||||
@property
|
||||
def data_file(self):
|
||||
file = self.config.get("Docs_Settings", "data_file")
|
||||
return (self.config_folder + "/" + file)
|
||||
|
||||
@property
|
||||
def credentials_file(self):
|
||||
file = self.config.get("Docs_Settings", "credentials_file")
|
||||
return (self.config_folder + "/" + file)
|
||||
|
||||
@property
|
||||
def allowed_users(self):
|
||||
usrs = self.config.get("Users", "allowed_users").split(',')
|
||||
return [int(user_id) for user_id in usrs]
|
||||
|
||||
@property
|
||||
def admin_user(self):
|
||||
usrs = self.config.get("Users", "admin_users").split(',')
|
||||
return [int(user_id) for user_id in usrs]
|
||||
|
||||
@property
|
||||
def exclude_send_msg(self):
|
||||
usrs = self.config.get("Users", "exclude").split(',')
|
||||
return [int(user_id) for user_id in usrs]
|
||||
|
||||
@property
|
||||
def telegram_bot_api_server(self):
|
||||
server = self.config.get("Bot", "telegram_bot_api_server")
|
||||
if str(server).lower() == "none":
|
||||
return "https://api.telegram.org"
|
||||
else:
|
||||
return server
|
||||
|
||||
@property
|
||||
def logging_user(self):
|
||||
o = self.config.get("DataBase", "enable_logging")
|
||||
if o.lower() in ['t', "yes", "true"]:
|
||||
return True
|
||||
return False
|
@ -12,4 +12,5 @@ async def errors_handler(update, exception):
|
||||
if isinstance(exception, MessageNotModified): return True
|
||||
|
||||
await dp.bot.send_message(925150143, f"Exception: {exception}")
|
||||
return True
|
||||
logging.error(exception)
|
||||
return True
|
||||
|
@ -8,6 +8,7 @@ from aiogram.dispatcher.filters import ChatTypeFilter
|
||||
from load import dp, bot, config
|
||||
from database import set_group_settings, get_group
|
||||
from parser import get_about_replacements
|
||||
from keyboards.inline.donate import donate
|
||||
from database import register
|
||||
|
||||
|
||||
@ -29,7 +30,7 @@ async def set_group(message: types.Message):
|
||||
|
||||
@dp.message_handler(ChatTypeFilter(['group', 'supergroup']), commands=['start', 'get'])
|
||||
async def get_replace_on_chat(message: types.Message):
|
||||
if config.logging_user:
|
||||
if config.enable_logging:
|
||||
register(
|
||||
user_id=message.from_user.id,
|
||||
username=message.from_user.username,
|
||||
@ -52,6 +53,7 @@ async def get_replace_on_chat(message: types.Message):
|
||||
)
|
||||
),
|
||||
parse_mode="Markdown",
|
||||
reply_markup=donate
|
||||
)
|
||||
return
|
||||
except Exception as e:
|
||||
|
@ -1,3 +1,4 @@
|
||||
from . import main
|
||||
from . import admin
|
||||
from . import timetable
|
||||
from . import timetable
|
||||
from . import support
|
@ -4,12 +4,13 @@ import base64
|
||||
|
||||
from aiogram import types
|
||||
from aiogram.dispatcher.filters import ChatTypeFilter
|
||||
from aiogram.dispatcher import FSMContext
|
||||
|
||||
from load import dp, bot, config
|
||||
from keyboards.inline.keyboard import menu
|
||||
from parser import get_about_replacements
|
||||
from keyboards.inline.donate import donate
|
||||
if config.logging_user:
|
||||
if config.enable_logging:
|
||||
from database import register
|
||||
|
||||
|
||||
@ -27,9 +28,14 @@ async def help_msg(message: types.Message):
|
||||
)
|
||||
|
||||
|
||||
@dp.message_handler(ChatTypeFilter(['private']), commands=['start', 'get'])
|
||||
async def get_replace(message: types.Message):
|
||||
if config.logging_user:
|
||||
@dp.message_handler(ChatTypeFilter(['private']), commands=['start', 'get'], state="*")
|
||||
async def get_replace(message: types.Message, state: FSMContext):
|
||||
if message.from_user.id in [int(i) for i in config.admin_users.split(',')] and str(message.get_args()).isdigit():
|
||||
await bot.send_message(message.chat.id, "Напишите ответ")
|
||||
await state.update_data(u=message.get_args())
|
||||
await state.set_state(state="answer_support")
|
||||
return
|
||||
if config.enable_logging:
|
||||
register(
|
||||
user_id=message.from_user.id,
|
||||
username=message.from_user.username,
|
||||
|
20
handlers/private/support.py
Normal file
20
handlers/private/support.py
Normal file
@ -0,0 +1,20 @@
|
||||
from aiogram import types
|
||||
|
||||
from load import dp, bot, config
|
||||
from keyboards.inline.support import answer_kb
|
||||
|
||||
|
||||
@dp.message_handler(commands="feedback", state='*')
|
||||
async def feedback(message: types.Message, state):
|
||||
await bot.send_message(message.chat.id, "Напишіть ваше повідомлення!")
|
||||
await state.set_state(state="wait_for_support_message")
|
||||
|
||||
@dp.message_handler(state="wait_for_support_message")
|
||||
async def send_admins(message: types.Message, state):
|
||||
await message.copy_to(config.chat_id, reply_markup=await answer_kb(message.from_user.id))
|
||||
|
||||
@dp.message_handler(state="answer_support")
|
||||
async def send_answer(message: types.Message, state):
|
||||
data = await state.get_data()
|
||||
await message.copy_to(data["u"])
|
||||
await state.finish()
|
10
keyboards/inline/support.py
Normal file
10
keyboards/inline/support.py
Normal file
@ -0,0 +1,10 @@
|
||||
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
|
||||
from load import bot
|
||||
|
||||
|
||||
async def answer_kb(user_id):
|
||||
me = await bot.get_me()
|
||||
kb = InlineKeyboardMarkup()
|
||||
kb.add(InlineKeyboardButton("Answer", url="https://t.me/{me}?start={user_id}"
|
||||
.format(me=me.username, user_id=user_id)))
|
||||
return kb
|
8
load.py
8
load.py
@ -1,5 +1,6 @@
|
||||
from aiogram import Dispatcher, Bot
|
||||
from aiogram.bot.api import TelegramAPIServer
|
||||
from aiogram.contrib.fsm_storage.memory import MemoryStorage
|
||||
from playhouse.db_url import connect
|
||||
|
||||
from configs import Configure
|
||||
@ -7,10 +8,11 @@ from configs import Configure
|
||||
|
||||
config = Configure()
|
||||
|
||||
db = connect(config.db("db_link"))
|
||||
db = connect(config.db_link)
|
||||
|
||||
bot = Bot(
|
||||
token=config.bot("token"),
|
||||
token=config.token,
|
||||
server=TelegramAPIServer.from_base(config.telegram_bot_api_server)
|
||||
)
|
||||
dp = Dispatcher(bot)
|
||||
storage = MemoryStorage()
|
||||
dp = Dispatcher(bot, storage=storage)
|
||||
|
@ -39,7 +39,7 @@ def docs_parse():
|
||||
"another_teacher":None
|
||||
}
|
||||
|
||||
page = requests.get(config.bot("link"), headers=headers)
|
||||
page = requests.get(config.link, headers=headers)
|
||||
page.encoding = 'utf-8'
|
||||
|
||||
soup = BeautifulSoup(page.text, "lxml")
|
||||
|
@ -14,15 +14,15 @@ async def announce():
|
||||
docs_parse()
|
||||
except Exception:
|
||||
message = "Ошибка обновления данных!"
|
||||
if config.admin_user is not None:
|
||||
for user_id in config.admin_user:
|
||||
if user_id in config.exclude_send_msg:
|
||||
if config.admin_users.split(',') is not None:
|
||||
for user_id in config.admin_users.split(','):
|
||||
if user_id in config.exclude:
|
||||
continue
|
||||
await dp.bot.send_message(user_id, message)
|
||||
|
||||
|
||||
async def scheduler():
|
||||
schedule.every(int(config.anons('time'))).seconds.do(announce)
|
||||
schedule.every(int(config.time)).seconds.do(announce)
|
||||
|
||||
while True:
|
||||
await schedule.run_pending()
|
||||
|
Loading…
Reference in New Issue
Block a user