Update app.py

app.py

@@ -15,8 +15,8 @@ api_key = os.getenv('api_key')  # Ваш API ключ
 
 app = Flask(__name__, template_folder="./")
 
-def init_db():
-    conn = sqlite3.connect('data1.db')
+def init_db(db_name):
+    conn = sqlite3.connect(db_name)
     cursor = conn.cursor()
     cursor.execute('''
     CREATE TABLE IF NOT EXISTS contacts (
@@ -36,16 +36,18 @@ def init_db():
     conn.commit()
     conn.close()
 
-init_db()
+init_db('data1.db')
+init_db('data2.db')
+init_db('data3.db')
 
-def save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1):
+def save_to_database(db_name, name, email, phone, pr5, pr4, pr3, pr2, pr1):
     try:
         moscow_timezone = pytz.timezone('Europe/Moscow')
         current_datetime = datetime.datetime.now(moscow_timezone)
         date = current_datetime.strftime('%Y-%m-%d')
         time = current_datetime.strftime('%H:%M:%S')
         
-        conn = sqlite3.connect('data1.db')
+        conn = sqlite3.connect(db_name)
         cursor = conn.cursor()
         cursor.execute('''
         INSERT INTO contacts (name, email, phone, pr1, pr2, pr3, pr4, pr5, date, time)
@@ -213,7 +215,7 @@ def handle_incoming_request():
 
     if current_day in [0, 4, 5, 6]:  # Понедельник, Пятница, Суббота, Воскресенье
         # Отправляем запросы в листы подряд
-        save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1)
+        db_name = 'data1.db'
         response_1 = send_additional_request1(email, name, phone, pr1, pr2, pr3, pr4, pr5)
         # Возвращаем список ответов
         return jsonify({'responses': [response_1]})
@@ -224,7 +226,7 @@ def handle_incoming_request():
             # Утром во Понедельник
             # Добавьте здесь код, который должен выполняться утром
         
-             save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1)
+             db_name = 'data1.db'
              response_1 = send_additional_request1(email, name, phone, pr1, pr2, pr3, pr4, pr5)
              # Возвращаем список ответов
              return jsonify({'responses': [response_1]})
@@ -237,7 +239,7 @@ def handle_incoming_request():
             # Утром во Понедельник
             # Добавьте здесь код, который должен выполняться утром
         
-             save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1)
+             db_name = 'data1.db'
              response_1 = send_additional_request1(email, name, phone, pr1, pr2, pr3, pr4, pr5)
              # Возвращаем список ответов
              return jsonify({'responses': [response_1]})
@@ -250,7 +252,7 @@ def handle_incoming_request():
             # Утром во Понедельник
             # Добавьте здесь код, который должен выполняться утром
         
-             save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1)
+             db_name = 'data1.db'
              response_1 = send_additional_request1(email, name, phone, pr1, pr2, pr3, pr4, pr5)
              # Возвращаем список ответов
              return jsonify({'responses': [response_1]})
@@ -261,7 +263,7 @@ def handle_incoming_request():
             # Утром во Понедельник
             # Добавьте здесь код, который должен выполняться утром
         
-             save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1)
+             db_name = 'data1.db'
              response_1 = send_additional_request1(email, name, phone, pr1, pr2, pr3, pr4, pr5)
              # Возвращаем список ответов
              return jsonify({'responses': [response_1]})
@@ -276,7 +278,7 @@ def handle_incoming_request():
             # Утром во Понедельник
             # Добавьте здесь код, который должен выполняться утром
         
-             save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1)
+             db_name = 'data1.db'
              response_1 = send_additional_request1(email, name, phone, pr1, pr2, pr3, pr4, pr5)
              # Возвращаем список ответов
              return jsonify({'responses': [response_1]})
@@ -288,7 +290,7 @@ def handle_incoming_request():
             # Утром во Понедельник
             # Добавьте здесь код, который должен выполняться утром
         
-             save_to_database1(name, email, phone, pr5, pr4, pr3, pr2, pr1)
+             db_name = 'data1.db'
              response_1 = send_additional_request1(email, name, phone, pr1, pr2, pr3, pr4, pr5)
              # Возвращаем список ответов
              return jsonify({'responses': [response_1]})
@@ -468,7 +470,6 @@ def show_contacts():
 
     except Exception as e:
         print(f"Error showing contacts: {e}")
-        return "Internal Server Error", 500