#!/usr/bin/python3
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

def plot_csv(
	file_path,
	line_columns,
	grid_column=None,
	line_colors=None, #ex. ['red','blue']
	line_styles=None, #ex. ['-','--']
	grid_on=True,
	x_label=None,
	y_label=None,
	xticks_rotation=0,
	title="CSV Data Plot"
	):
	df=pd.read_csv(file_path)
	fig,ax=plt.subplots(figsize=(10,6))
	for i,col in enumerate(line_columns):
		color=line_colors[i] if line_colors else None
		linestyle=line_styles[i] if line_styles else '-'
		ax.plot(df.index,df[col], label=col, color=color,
			linestyle=linestyle)
	if x_label:
		ax.set_xlabel(x_label)
	if y_label:
		ax.set_ylabel(y_label)
	ax.set_title(title)
	ax.legend()
	plt.tight_layout()
	plt.savefig('output.png',dpi=300,transparent=True)


def main():
	'''
	plot_csv(file_path="data.csv",
		line_columns=["Temperature","Humidity","Wetness"],
		grid_column="Time",
		line_colors=["red","blue","orange"],
		line_styles=["-","--","dotted"],
		grid_on=True,
		x_label="Time (s)",
		y_label="Values",
		xticks_rotation=45,
		title="Environmentral Data"
	)
	'''
	import argparse
	parser=argparse.ArgumentParser(description="Ploter auto")
	parser.add_argument("--file",type=str, required=True,
		help="data filename")
	parser.add_argument("--lines",nargs="+", required=True,
		help="colums sep by space")
	args=parser.parse_ags()
	plot_csv(
		file_path=args.file
		line_columns=args.lines
		)

if __name__=="__main__":
	main()