add ML telegram logs

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():
+    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,8 +226,10 @@ 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('Начинается обновление полей в базе')
     # Загрузка в базу обновленных значений
+    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'),
@@ -223,6 +237,8 @@ def raschet():
             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()