from utils import *
import datetime
import pickle

if __name__ == '__main__':
    data = pd.read_csv("../data/test-data/raw-exposure.csv")[["cid", "device_id"]]
    data["y"] = 1
    test_data = data.tail(1)

    ffm = FFMFormatPandas()
    data = ffm.fit_transform(data, y='y')
    data.to_csv("../data/ffm_data.csv", index=False)

    with open("../data/ffm.object", "wb") as f:
        pickle.dump(ffm, f)
    with open("../data/ffm.object", "rb") as f:
        ffm = pickle.load(f)
    result = ffm.transform(test_data)
    print(result)
    data_1 = pd.read_csv("../data/ffm_data.csv", header=None).tail(5)
    print(data_1)