def get_batch_data()

in assets/functions/batch_anomaly_detection/app.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def get_batch_data(meter_start, meter_end, data_end, db_schema, connection):
    query = '''select meter_id, date_trunc('day', reading_date_time) as ds, sum(reading_value) as y 
  from "{}".daily
    where meter_id between '{}' and '{}'
  and reading_date_time > date_add('year', -1, timestamp '{}')
  and reading_date_time <= timestamp '{}'
    group by 2,1
    order by 2,1;
    '''.format(db_schema,meter_start, meter_end, data_end, data_end)

    df_daily = pd.read_sql(query, connection)
    df_daily['weekend'] = df_daily['ds'].apply(weekend)
    return df_daily