From 0ac9fd2d255c9ae548e5d89a9a42b16efee04509 Mon Sep 17 00:00:00 2001 From: AlexP077 Date: Mon, 27 Mar 2023 19:13:25 +0300 Subject: [PATCH] add_env --- service/tasks.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/service/tasks.py b/service/tasks.py index 457b405..cc0e4e6 100644 --- a/service/tasks.py +++ b/service/tasks.py @@ -1,3 +1,5 @@ +import os + import catboost import pandas as pd import psycopg2 @@ -9,6 +11,7 @@ from sklearn import model_selection as ms from sqlalchemy import text from postamates.settings import AGE_DAY_LIMIT +from postamates.settings import DB_URL from service.models import PlacementPoint @@ -20,7 +23,7 @@ from service.models import PlacementPoint @shared_task() def raschet(): conn = sqlalchemy.create_engine( - 'postgresql://sst_postamates_user:sst_postamates_pass@postnet-dev.selftech.ru:5487/sst_postamates_db', + DB_URL, connect_args={'options': '-csearch_path=public'}, ) query = text('select * from service_placementpoint') @@ -72,8 +75,10 @@ def raschet(): # Загрузка в базу обновленных значений conn2 = psycopg2.connect( - database='sst_postamates_db', user='sst_postamates_user', password='sst_postamates_pass', - host='postnet-dev.selftech.ru', port='5487', options='-c search_path=public', + 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() update_records1 = []