from proxAddDataWindPoI import *
import pandas as pd

def nearestBiomass(startyr, endyr):
    """
    This loop goes through csv's by year + adds data from nearest site
    NEED TO EDIT CSV NAME
    """
    for i in range(startyr, endyr, 1):
        year = i
        print("Year: "+str(year))
        csv1 = "RSRaw/RSPytAnyP2"+str(year)+".csv"
        csv2 = "Wind/RS_PoI/RSPoIP2"+str(year)+".csv"
        print("Input CSV: "+str(csv1))
        addWindData(csv1, csv2)

def withinXkm(csv1, csv2, buffer):
    """
    Function which only keeps rows where the distance between site and
    property is less than a specified buffer
    """
    df = pd.read_csv(csv1)
    df = df[df.NWindFarm_DistKm <= buffer]
    print(df.NWindFarm_DistKm)
    df.to_csv(csv2,index=None,encoding='utf-8')

def treatmentGroup(year, buffer):
    """
    The full process of determining treatment group for a certain year of
    Repeat Sales. Need to specify buffer
    """
    print("Starting Treatment Group Identification for Year: "+str(year))
    nearestBiomass(year, year+1)
    csv1 = "Wind/RS_PoI/RSPoIP2"+str(year)+".csv"
    csv2 = "Wind/TG_PoI/RSWindPoI_TGP2"+str(year)+".csv"
    print("Extracting Repeat Sales within a buffer of "+str(buffer)+" from "+str(csv1) + " to "+str(csv2))
    withinXkm(csv1, csv2, buffer)

treatmentGroup(2004, 2)