# Author: Lee Wu Love Lele
# Datetime: 2024/9/7 21:28
import db_connection

# 创建游标对象
connection = db_connection.get_connection()
cursor = connection.cursor()
weight_group_name = "subjectCallback"
rfid_group_name = "rfidCallback"
# 根据起止日期获取牛只体重信息
def get_cattle_weight_by_datetime(start_datetime, end_datetime, reader):
    query = f"SELECT biz_time, biz_data FROM rhm_insure_dev.notify_log WHERE notify_group = '{weight_group_name}' AND reader_name = '{reader}' AND biz_time BETWEEN '{start_datetime}' AND '{end_datetime}' ORDER BY biz_time ASC"
    cursor.execute(query)
    result = cursor.fetchall()
    return result

def get_rfid_by_datetime(start_datetime,end_datetime,reader):
    query = f"SELECT biz_time, biz_data FROM rhm_insure_dev.notify_log WHERE notify_group = '{rfid_group_name}' AND reader_name = '{reader}' AND biz_time BETWEEN '{start_datetime}' AND '{end_datetime}' ORDER BY biz_time ASC"
    cursor.execute(query)
    result = cursor.fetchall()
    return result

def close():
    cursor.close()
    connection.close()