From abb299271058475f458e9c1acbec750f80b3f2c9 Mon Sep 17 00:00:00 2001
From: timofejmalinin <tmalinin@endwork.today>
Date: Mon, 17 Apr 2023 16:54:37 +0400
Subject: [PATCH] add ML telegram logs

---
 service/tasks.py | 41 +++++++++++++++++++++++++++++------------
 1 file changed, 29 insertions(+), 12 deletions(-)

diff --git a/service/tasks.py b/service/tasks.py
index 37ef992..6cb9396 100644
--- a/service/tasks.py
+++ b/service/tasks.py
@@ -15,17 +15,29 @@ from sklearn import metrics
 from sklearn import model_selection as ms
 from sqlalchemy import text
 
+import requests
+
 from postamates.settings import AGE_DAY_LIMIT
 from postamates.settings import DB_URL
 from service.models import PlacementPoint
 
 
+def log_to_telegram(msg):
+    requests.post('https://api.telegram.org/bot6275517704:AAHVp_qv9d9NU740JJdOM2fJdgS4r1AgJrw/sendMessage', json={"chat_id": "-555238820", "text": msg})
+
+
 @shared_task()
 def raschet():
-    conn = sqlalchemy.create_engine(
-        DB_URL,
-        connect_args={'options': '-csearch_path=public'},
-    )
+    log_to_telegram('start raschet')
+    try:
+        log_to_telegram('try connect to db')
+        conn = sqlalchemy.create_engine(
+            DB_URL,
+            connect_args={'options': '-csearch_path=public'},
+        )
+    except:
+        log_to_telegram('error connect to db')
+
     query = text('select * from service_placementpoint')
     connection = conn.connect()
     pts = pd.read_sql(query, connection)
@@ -214,15 +226,19 @@ def raschet():
 
     update_fields_working = update_fields.loc[update_fields.status == 'Working'].reset_index(drop=True)
     update_fields_working = update_fields_working.fillna(0)
-
+    log_to_telegram('Начинается обновление полей в базе')
     # Загрузка в базу обновленных значений
-    conn2 = psycopg2.connect(
-        database=os.getenv('POSTGRES_DB', 'postgres'), user=os.getenv('POSTGRES_USER', 'postgres'),
-        password=os.getenv('POSTGRES_PASSWORD', 'postgres'),
-        host=os.getenv('POSTGRES_HOST', 'postgres'), port=os.getenv('POSTGRES_PORT', 'postgres'),
-        options='-c search_path=public',
-    )
-    cursor = conn2.cursor()
+    try:
+        log_to_telegram('Подключение к базе данных 2')
+        conn2 = psycopg2.connect(
+            database=os.getenv('POSTGRES_DB', 'postgres'), user=os.getenv('POSTGRES_USER', 'postgres'),
+            password=os.getenv('POSTGRES_PASSWORD', 'postgres'),
+            host=os.getenv('POSTGRES_HOST', 'postgres'), port=os.getenv('POSTGRES_PORT', 'postgres'),
+            options='-c search_path=public',
+        )
+        cursor = conn2.cursor()
+    except:
+        log_to_telegram('Не удалось подключиться к базе данных')
 
     # prediction_current
     update_records1 = []
@@ -288,6 +304,7 @@ def raschet():
         cursor.execute('ROLLBACK')
         psycopg2.extras.execute_batch(cursor, sql_update_query, update_records5)
         conn2.commit()
+    log_to_telegram('end raschet')
 
 
 @shared_task()