@ -3,13 +3,12 @@
# Группы пользователей # Группы пользователей
from bot_sys import bot_bd , log , config , keyboard , user_access from bot_sys import keyboard , user_access , bot_bd
from bot_modules import start , access from bot_modules import mod_simple_message
from template import simple_message , sql_request from template import simple_message , sql_request , bd_item
from aiogram . dispatcher import FSMContext from aiogram . dispatcher import FSMContext
from aiogram . dispatcher . filters . state import State , StatesGroup from aiogram . dispatcher . filters . state import State , StatesGroup
from aiogram . dispatcher import Dispatcher
class FSMRequestToBD ( StatesGroup ) : class FSMRequestToBD ( StatesGroup ) :
sqlRequest = State ( ) sqlRequest = State ( )
@ -27,29 +26,10 @@ user_id_field = 'user_id'
access_field = ' access ' access_field = ' access '
create_datetime_field = ' createDateTime ' create_datetime_field = ' createDateTime '
init_bd_cmds = [ f """ CREATE TABLE IF NOT EXISTS { table_groups_name } (
{ key_table_groups_name } INTEGER PRIMARY KEY NOT NULL ,
{ name_table_groups_field } TEXT ,
{ access_field } TEXT ,
{ create_datetime_field } TEXT ,
UNIQUE ( { key_table_groups_name } ) ,
UNIQUE ( { name_table_groups_field } )
) ; """ ,
f """ CREATE TABLE IF NOT EXISTS { table_user_in_groups_name } (
{ user_id_field } INTEGER ,
{ key_table_groups_name } INTEGER ,
{ access_field } TEXT ,
{ create_datetime_field } TEXT ,
UNIQUE ( { user_id_field } , { key_table_groups_name } )
) ; """ ,
f " INSERT OR IGNORE INTO module_access (modName, modAccess, itemDefaultAccess) VALUES ( ' { module_name } ' , ' { user_access . user_access_group_new } =- ' , ' { user_access . user_access_group_new } =- ' ); " ,
f " INSERT OR IGNORE INTO { table_groups_name } ( { name_table_groups_field } , { access_field } , { create_datetime_field } ) VALUES ( ' { user_access . user_access_group_new } ' , ' { user_access . user_access_group_new } =- ' , { bot_bd . GetBDDateTimeNow ( ) } ); "
]
# --------------------------------------------------------- # ---------------------------------------------------------
# Сообщения # Сообщения
group_ start_message= ''' start_message = '''
< b > Группы пользователей находятся в стадии разработки < / b > < b > Группы пользователей находятся в стадии разработки < / b >
Пока можете воспользоваться хардкорным способом через запросы к БД Пока можете воспользоваться хардкорным способом через запросы к БД
@ -73,69 +53,73 @@ help_message = '''
` user_in_groups ( user_id , group_id ) ` - содержит соответсвия ID пользователей и групп ` user_in_groups ( user_id , group_id ) ` - содержит соответсвия ID пользователей и групп
''' '''
user_group _button_name= " ️ ️ " start_menu _button_name= " ️ ️ "
sql_request_button_name = " ⛃ Запрос к БД для редактирования групп " sql_request_button_name = " ⛃ Запрос к БД для редактирования групп "
help_button_name = " 📄 Информация по группам " help_button_name = " 📄 Информация по группам "
# --------------------------------------------------------- init_access = ' {user_access.user_access_group_new} =- '
# Работа с кнопками
''' class ModuleGroups ( mod_simple_message . SimpleMessageModule ) :
def GetEditGroupKeyboardButtons ( a_Message , a_UserGroups ) : def __init__ ( self , a_ChildModuleNameList , a_Bot , a_ModuleAgregator , a_BotMessages , a_BotButtons , a_Log ) :
cur_buttons = [ super ( ) . __init__ ( start_message , start_menu_button_name , init_access , a_ChildModuleNameList , a_Bot , a_ModuleAgregator , a_BotMessages , a_BotButtons , a_Log )
keyboard . ButtonWithAccess ( sql_request_button_name , user_access . AccessMode . EDIT , GetAccess ( ) ) , self . m_SqlRequestButtonName = self . CreateButton ( ' sql request ' , sql_request_button_name )
keyboard . ButtonWithAccess ( help_button_name , user_access . AccessMode . VIEW , GetAccess ( ) ) self . m_RequestStartMessage = self . CreateMessage ( ' equest start ' , request_start_message )
]
mods = [ start ] self . m_HelpButtonName = self . CreateButton ( ' help ' , help_button_name )
return keyboard . MakeKeyboard ( keyboard . GetButtons ( mods ) + cur_buttons , a_UserGroups ) self . m_HelpMessage = self . CreateMessage ( ' help ' , help_message )
# --------------------------------------------------------- self . m_HelpMessageHandler = simple_message . InfoMessageTemplate (
# Обработка сообщений self . m_Bot ,
self . m_HelpMessage ,
# --------------------------------------------------------- self . m_GetStartKeyboardButtonsFunc ,
# Работа с базой данных групп None ,
self . m_GetAccessFunc
def GetGroupIDForUser ( a_UserID ) : )
return bot_bd . SQLRequestToBD ( ' SELECT group_id FROM user_in_groups WHERE user_id = ? ' , param = [ a_UserID ] )
def GetInitBDCommands ( self ) :
def GetGroupNamesForUser ( a_UserID ) : return super ( ) . GetInitBDCommands ( ) + [
return bot_bd . SQLRequestToBD ( ' SELECT groupName FROM user_groups WHERE group_id=(SELECT group_id FROM user_in_groups WHERE user_id = ?) ' , param = [ a_UserID ] ) f """ CREATE TABLE IF NOT EXISTS { table_groups_name } (
{ key_table_groups_name } INTEGER PRIMARY KEY NOT NULL ,
# --------------------------------------------------------- { name_table_groups_field } TEXT ,
# API { access_field } TEXT ,
{ create_datetime_field } TEXT ,
def GetUserGroupData ( a_UserID ) : UNIQUE ( { key_table_groups_name } ) ,
r = GetGroupNamesForUser ( a_UserID ) UNIQUE ( { name_table_groups_field } )
groups = [ ] ) ; """ ,
for i in r : f """ CREATE TABLE IF NOT EXISTS { table_user_in_groups_name } (
if len ( i ) > 0 : { user_id_field } INTEGER ,
groups + = [ i [ 0 ] ] { key_table_groups_name } INTEGER ,
return user_access . UserGroups ( a_UserID , groups ) { access_field } TEXT ,
{ create_datetime_field } TEXT ,
# Инициализация БД UNIQUE ( { user_id_field } , { key_table_groups_name } )
def GetInitBDCommands ( ) : ) ; """ ,
return init_bd_cmds f " INSERT OR IGNORE INTO { table_groups_name } ( { name_table_groups_field } , { access_field } , { create_datetime_field } ) VALUES ( ' { user_access . user_access_group_new } ' , ' { user_access . user_access_group_new } =- ' , { bot_bd . GetBDDateTimeNow ( ) } ); "
]
def GetAccess ( ) :
return access . GetAccessForModule ( module_name ) def GetName ( self ) :
return module_name
# Доступные кнопки
def GetModuleButtons ( ) : def GetStartKeyboardButtons ( self , a_Message , a_UserGroups ) :
return [ keyboard . ButtonWithAccess ( user_group_button_name , user_access . AccessMode . VIEW , GetAccess ( ) ) ] mod_buttons = super ( ) . GetStartKeyboardButtons ( a_Message , a_UserGroups )
cur_buttons = [
# Обработка кнопок keyboard . ButtonWithAccess ( self . m_SqlRequestButtonName , user_access . AccessMode . EDIT , self . GetAccess ( ) ) ,
def RegisterHandlers ( dp : Dispatcher ) : keyboard . ButtonWithAccess ( self . m_HelpButtonName , user_access . AccessMode . VIEW , self . GetAccess ( ) )
dp . register_message_handler ( simple_message . InfoMessageTemplateLegacy ( group_start_message , GetEditGroupKeyboardButtons , GetAccess ) , text = user_group_button_name ) ]
dp . register_message_handler ( simple_message . InfoMessageTemplateLegacy ( help_message , GetEditGroupKeyboardButtons , GetAccess ) , text = help_button_name ) return mod_buttons + keyboard . MakeButtons ( cur_buttons , a_UserGroups )
sql_request . RequestToBDRegisterHandlers ( dp , sql_request_button_name , request_start_message , FSMRequestToBD , GetEditGroupKeyboardButtons , user_access . AccessMode . EDIT , GetAccess ) def RegisterHandlers ( self ) :
''' super ( ) . RegisterHandlers ( )
sql_request . RequestToBDRegisterHandlers (
self . m_Bot ,
self . m_SqlRequestButtonName ,
self . m_RequestStartMessage ,
FSMRequestToBD ,
self . m_GetStartKeyboardButtonsFunc ,
user_access . AccessMode . EDIT ,
self . m_GetAccessFunc
)
self . m_Bot . RegisterMessageHandler (
self . m_HelpMessageHandler ,
bd_item . GetCheckForTextFunc ( self . m_HelpButtonName )
)
def GetUserGroupData ( a_Bot , a_UserID ) :
def GetGroupNamesForUser ( a_UserID ) :
return a_Bot . SQLRequest ( ' SELECT groupName FROM user_groups WHERE group_id=(SELECT group_id FROM user_in_groups WHERE user_id = ?) ' , param = [ a_UserID ] )
r = GetGroupNamesForUser ( a_UserID )
groups = [ ]
for i in r :
if len ( i ) > 0 :
groups + = [ i [ 0 ] ]
return user_access . UserGroups ( a_UserID , groups )
Alexei
2 years ago