charging_data.py

import matplotlib
import pandas as pd
from iss4e.db import influxdb, mysql
from iss4e.util.config import load_config
from iss4e.webike.trips.auxiliary import DateTime

matplotlib.use('Agg')
from pytz import timezone

config = load_config()

participants = pd.read_excel("participant map.xlsx", header=[1])
female_staff = participants.iloc[1:8][["Email", "IMEI"]]
male_staff = participants.iloc[10:16][["Email", "IMEI"]]
female_students = participants.iloc[18:23][["Email", "IMEI"]]
male_students = participants.iloc[25:32][["Email", "IMEI"]]

eastern = timezone('Canada/Eastern')


def analyze_charge(list):
    last_time = None
    start = None
    entires_count = 0
    for entry in list[config["webike.measurement"]]:
        time = DateTime(entry["time"])
        if start is None:
            start = time
            start_entry = entry
            entries_count = 1
        elif (time - last_time).total_seconds() > 60*5:
            if entries_count >= 5 and (last_time - start).total_seconds() >= 5 * 60:
                if(start._datetime.hour > 8 and start._datetime.hour < 10):
                    print(start_entry)
                yield {"time": str(start), "voltage":start_entry["voltage"], "temp":start_entry["battery_temperature"]}

            start = time
            start_entry = entry
            entries_count = 1
        else:
            entries_count += 1

        last_time = time


def get_charging(l):
    charges = []
    for imei in l["IMEI"]:
        query = "select charging_current, discharge_current,voltage, battery_temperature from {measurement} where imei='{imei}' and \
            (charging_current>70 or (discharge_current < 450 and discharge_current >50))".format(measurement=config["webike.measurement"], imei=int(imei))
        result = influx_client.query(query)
        charges += list(analyze_charge(result))
    return charges


with mysql.connect(**config["webike.mysql"]) as mysql_client, influxdb.connect(
        **config["webike.influx"]) as influx_client:
    fstaff = get_charging(female_staff)
    with open("data/fstaff_charging",mode='w') as file:
        file.writelines("{line}\n".format(line= line) for line in fstaff)
    mstaff = get_charging(male_staff)
    with open("data/mstaff_charging",mode='w') as file:
        file.writelines("{line}\n".format(line=line) for line in mstaff)
    fstuds = get_charging(female_students)
    with open("data/fstuds_charging",mode='w') as file:
        file.writelines("{line}\n".format(line=line) for line in fstuds)
    mstuds = get_charging(male_students)
    with open("data/mstuds_charging",mode='w') as file:
        file.writelines("{line}\n".format(line=line) for line in mstuds)