From 4c5b3de556c1955d0f2a666d1b0a98c0105a3786 Mon Sep 17 00:00:00 2001 From: Ceda EI Date: Mon, 5 Nov 2018 15:30:36 +0530 Subject: [PATCH] Close database on sigterm --- bot.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/bot.py b/bot.py index a2d29fd..9cd53c9 100644 --- a/bot.py +++ b/bot.py @@ -7,6 +7,8 @@ import questable import random from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, \ RegexHandler +import signal +import sys try: import config @@ -456,8 +458,15 @@ def message_handling(bot, update, db): edit_quest(bot, update, player, state["extra"], "diff", "side_quest") +def sigterm_handler(signal, frame, db): + db.close() + sys.exit(0) + + db = sqlite3.connect("questable.db", check_same_thread=False) cursor = db.cursor() +signal.signal(signal.SIGTERM, lambda x, y: sigterm_handler(x, y, db)) + # Set up tables queries = [ ("CREATE TABLE IF NOT EXISTS quests(chat_id int NOT NULL, qid int NOT"