def set_welcome(bot: Bot, update: Update, session):
if update.message.chat.type in ['group', 'supergroup']:
group = update_group(update.message.chat, session)
welcome_msg = session.query(WelcomeMsg).filter_by(
chat_id=group.id).first()
if welcome_msg is None:
welcome_msg = WelcomeMsg(chat_id=group.id,
message=update.message.text.split(' ',
1)[1])
else:
welcome_msg.message = update.message.text.split(' ', 1)[1]
session.add(welcome_msg)
session.commit()
send_async(bot, chat_id=update.message.chat.id, text=MSG_WELCOME_SET)
def welcome(bot: Bot, update: Update, session):
newbie(bot, update)
global last_welcome
print('welcome')
if update.message.chat.type in ['group', 'supergroup']:
group = update_group(update.message.chat, session)
for new_chat_member in update.message.new_chat_members:
user = add_user(new_chat_member, session)
print(user)
administrator = session.query(Admin).filter_by(
user_id=user.id).all()
allow_anywhere = False
for adm in administrator:
if adm.admin_type == AdminType.FULL.value:
allow_anywhere = True
break
print(update.message.chat.id)
print(VENTURE_CHAT_ID == update.message.chat.id)
if str(update.message.chat.id) == VENTURE_CHAT_ID:
print('equal')
if group.welcome_enabled:
print('enable_welcome')
welcome_msg = session.query(WelcomeMsg).filter_by(
chat_id=group.id).first()
send_async(bot,
chat_id=update.message.chat.id,
text=fill_template(welcome_msg.message, user))
elif user is None and not allow_anywhere and user.id != bot.id:
send_async(bot, chat_id=update.message.chat.id)
bot.restrictChatMember(update.message.chat.id,
new_chat_member.id)
else:
if group.welcome_enabled:
welcome_msg = session.query(WelcomeMsg).filter_by(
chat_id=group.id).first()
if welcome_msg is None:
welcome_msg = WelcomeMsg(chat_id=group.id,
message=MSG_WELCOME_DEFAULT)
session.add(welcome_msg)
welcomed = session.query(Wellcomed).filter_by(
user_id=new_chat_member.id,
chat_id=update.message.chat.id).first()
if welcomed is None:
if time() - last_welcome > 30:
send_async(bot,
chat_id=update.message.chat.id,
text=fill_template(
welcome_msg.message, user))
last_welcome = time()
welcomed = Wellcomed(user_id=new_chat_member.id,
chat_id=update.message.chat.id)
session.add(welcomed)
session.commit()
def show_welcome(bot: Bot, update, session):
if update.message.chat.type in ['group', 'supergroup']:
group = update_group(update.message.chat, session)
welcome_msg = session.query(WelcomeMsg).filter_by(
chat_id=group.id).first()
if welcome_msg is None:
welcome_msg = WelcomeMsg(chat_id=group.id,
message=MSG_WELCOME_DEFAULT)
session.add(welcome_msg)
session.commit()
send_async(bot, chat_id=group.id, text=welcome_msg.message)
def welcome(bot: Bot, update: Update):
newbie(bot, update)
global last_welcome
if update.message.chat.type in ['group', 'supergroup']:
group = update_group(update.message.chat)
for new_chat_member in update.message.new_chat_members:
session = Session()
user = add_user(new_chat_member)
administrator = session.query(Admin).filter_by(
user_id=user.id).all()
allow_anywhere = False
for adm in administrator:
if adm.admin_type == AdminType.FULL.value:
allow_anywhere = True
break
if len(group.squad) == 1 and group.squad[0].thorns_enabled and user.id != 386494081 and \
(user.member or user.member not in group.squad[0].members) and not allow_anywhere:
send_async(bot,
chat_id=update.message.chat.id,
text=MSG_THORNS.format(str(user)))
bot.kickChatMember(update.message.chat.id, new_chat_member.id)
bot.unbanChatMember(update.message.chat.id, new_chat_member.id)
else:
if group.welcome_enabled:
welcome_msg = session.query(WelcomeMsg).filter_by(
chat_id=group.id).first()
if welcome_msg is None:
welcome_msg = WelcomeMsg(chat_id=group.id,
message=MSG_WELCOME_DEFAULT)
session.add(welcome_msg)
welcomed = session.query(Wellcomed).filter_by(
user_id=new_chat_member.id,
chat_id=update.message.chat.id).first()
if welcomed is None:
if time() - last_welcome > 30:
send_async(bot,
chat_id=update.message.chat.id,
text=fill_template(
welcome_msg.message, user))
last_welcome = time()
welcomed = Wellcomed(user_id=new_chat_member.id,
chat_id=update.message.chat.id)
session.add(welcomed)
session.commit()