def close_db(error):
"""Closes the database session at the end of the request.
:param error: unused
"""
db_session.remove()
def shutdown_session(exception=None):
db_session.remove()
def teardown_appcontext(exception=None):
# remove database session at the end of the request,
# or when the application shuts down
db_session.remove()
def shutdown_session(exception=None):
db_session.remove()
def close_db(error):
"""
Closes the database again at the end of the request.
"""
db_session.remove()
def teardown(bepis):
""" close the database connection after request is done """
db_session.remove()
def remove_session(exception=None):
db_session.remove()
def session_closes(request):
db_session.remove()
return request
def shutdown_session(f=None):
db_session.remove()
def close_db(error):
""" Closes the databse at the end of the request """
db_session.remove()
def shutdown_event():
db_session.remove()
def tearDownClass(cls):
db_session.remove()
Base.metadata.drop_all()
def close_db(e):
db_session.remove()
def after_request(response):
db_session.remove()
return response
def after_request(response):
db_session.remove()
return response
def closeDb(e):
""" closes db connection """
db_session.remove()
def bot_func(message):
logger.info("[%s] %s " % (message.bot.bot_name, message))
bot = message.bot
bot.last_msg = time.time()
# session = db_session()
# msg = BotMessage(bot_name=bot.self.name, sender=message.member.name if message.member else message.sender.name,
# chat=message.chat.name, type=message.type, message=message.text,
# url=message.url, created_at=datetime.now())
# session.add(msg)
# session.commit()
message_type = '%s' % message.type
logger.info('[%s][%s][%s][%s] %s' % (message_type, message.chat.name, message.member, message.chat, message.text))
if message_type == 'Note':
requests.post('http://10.9.21.184/api/mp/addGroup', {
'group': '%s' % message.chat.name,
'text': message.text
}).close()
if bot.master:
bot_master_handler(message)
return
elif bot.self == message.sender and message.chat in (bot.self, bot.file_helper):
bot_command_handler(message)
return
elif '京东菁英站' in '%s' % message.chat:
logger.info('Message from %s type %s' % (message.chat, message_type))
if message_type == 'Text':
logger.info('Request tg return %s' % requests.post('https://' + settings.TG_DOMAIN + '/api/tg/save', {
'text': message.text,
'type': message_type,
'sender': '%s' % message.sender.name
}).json())
elif message_type in ['Picture', 'Video']:
raw = message.get_file()
the_id = cutt.upload_raw(raw)
logger.info('Request tg return %s' % requests.post('https://' + settings.TG_DOMAIN + '/api/tg/save', {
'text': the_id,
'type': message_type,
'sender': '%s' % message.sender.name
}).json())
return
elif message.member:
text = message.text
if message_type in ['Picture', 'Video']:
raw = message.get_file()
text = cutt.upload_raw(raw)
data = {
'text': text,
'type': message_type,
'sender': '%s' % message.member.name,
'chat': '%s' % message.chat.name
}
logger.info('Request tg %s return %s' % (data,
requests.post('https://' + settings.TG_DOMAIN + '/api/tg/save',
data).json()))
logger.info('Request ww %s return %s' % (data,
requests.post('https://' + settings.WW_DOMAIN + '/ww/group/recv',
json=data).json()))
articles = message.articles
if articles:
if bot.crawler_articles:
try:
crawler.crawler(message)
finally:
db_session.remove()
else:
for article in articles:
logger.info('[%s] %s %s' % (message.sender.name, article.title, article.url))
if message.type == 'Sharing':
logger.info('[%s] %s %s' % (message.sender.name, message.text, message.url))
if message.type == 'Friends' and bot.auto_accept:
# 好友申请
bot.accept_friend(message.card)
if bot.notify_dingding:
cutt.send_dingding_msg('%s 自动接受了好友请求 %s' % (bot.self.name, message.card.name), bot.master_phone)
if isinstance(message.chat, Group):
requests.post('http://10.9.21.184/api/mp/addMessage', {
'group': '%s' % message.chat.name,
'from': '%s' % message.member.name,
'text': '%s: %s' % (message.type, message.text)
}).close()
