import os
from arcproject.waterquality import classes
from arcproject.scripts.load_data_bulk import slurp

# path to location with data
data = r"C:\Users\Andy\Desktop\ArcData" # or location on x drive


def jan():
	print("January 2016")

	path = os.path.join(data, "Jan_2016")
	s = slurp.Slurper()
	s.add_new_sites = True
	s.dst = True
	s.gain_pattern = "*WQ_*"
	s.skipext = [".csv", ".xlsx", ".xls", ".dbf", ".prj", ".shp", ".shx", ".lyr"]
	s.site_function_params = {"site_part": 2, "gain_part": 4}
	s.exclude = ['StatePlaneCAII', 'SummaryFiles', "Arc_022016"]
	print("Adding gain files to database")
	s.slurp_gains(path)
	print("Adding water quality transects to database")
	s.slurp_trans(path)


def feb():
	print("Feburary 2016")
	path = os.path.join(data, "Feb_2016")
	s = slurp.Slurper()
	s.add_new_sites = True
	s.dst = True
	s.gain_pattern = "*WQ_*"
	s.skipext = [".csv", ".xlsx", ".xls", ".dbf", ".prj", ".shp", ".shx", ".lyr"]
	s.site_function_params = {"site_part": 2, "gain_part": 4}
	s.exclude = ['StatePlaneCAII', 'SummaryFiles', "Arc_022016"]
	print("Adding gain files to database")
	s.slurp_gains(path)
	print("Adding water quality transects to database")
	s.slurp_trans(path)


def mar():
	print("March 2016")
	path = os.path.join(data, "Mar_2016")
	s = slurp.Slurper()
	s.add_new_sites = True