import os
from sklearn.externals import joblib
from .funcs import disk_2_s3, wget

def dump_clf(clf, s3_path):
	'''
	Pickles a Scikit-Learn classifier to local.
	Loads that local file to s3.
	removes the local file.
	'''
	filename = s3_path.split('/')[-1]
	
	joblib.dump(clf, filename)
	
	r = disk_2_s3(filename, s3_path)
	
	os.remove(filename)
	
	return r

def load_clf(s3_path):
	'''
	Downloads a pickled classifier from s3.
	unpickles it into a Scikit-Learn classifer.
	removes the pickled classifier.
	'''
	filename = s3_path.split('/')[-1]
	
	wget(s3_path, filename)

	clf = joblib.load(filename)

	os.remove(filename)
	
	return clf