import sys

from PyQt5.QtWidgets import QApplication, QBoxLayout, QMainWindow, QVBoxLayout, QWidget
from PyQt5 import uic 
from PyQt5.QtWidgets import QApplication, QMainWindow

from chartilo import Chartilo

from chartilo.drawers import GridDrawer, CandleChartDrawer, LineChartDrawer, LineDrawer, MaxMinValuesDrawer

from ast import literal_eval

from chartilo.models import Line, Candle
from chartilo.themes import Light, Dark


class MyWidget(QMainWindow):
	def __init__(self):
		super().__init__()
		uic.loadUi('untitled.ui', self)

		# adding canvas to frame
		self._chartilo = Chartilo()
		layout = QVBoxLayout()
		layout.addWidget(self._chartilo)
		self.frame.setLayout(layout)

		# setting states
		states = {
			"type" : Candle, 
			"theme": Dark,
			"drawers" : {
				"grid" : GridDrawer,
				"lineChart" : CandleChartDrawer,
				"line" : LineDrawer,
				"maxMinDrawer" : MaxMinValuesDrawer
			},
		}

		# getting data
		file = open("data.txt", "r")
		data = literal_eval(file.read())

		# setting data, states and running library
		self._chartilo.setData(data)
		self._chartilo.setStates(states)
		self._chartilo.updateCanvas()


if __name__ == '__main__':
	app = QApplication(sys.argv)
	ex = MyWidget()
	ex.show()
	sys.exit(app.exec_())

class MyWidget(QMainWindow):
	def __init__(self):
		super().__init__()
		uic.loadUi('uifile.ui', self)

		# adding painter to frame
		self._chartilo = Chartilo()
		layout = QVBoxLayout()
		layout.addWidget(self._chartilo)
		self.frame.setLayout(layout)

		# setting states
		states = {
			"type" : Candle, # the type of chart(liner or candler)
			"theme": Dark, # theme of chart (light or dark)
			"drawers" : {
				"grid" : GridDrawer, # class is used to draw grid for chart
				"lineChart" : CandleChartDrawer, # class is used to draw candles
				"line" : LineDrawer, # showing curren price line
				"maxMinDrawer" : MaxMinValuesDrawer # showing maximum and minimum of chart
			},
		}

# this is how you can import drawers from chartilo library
from chartilo.drawers import GridDrawer, CandleChartDrawer, LineChartDrawer, LineDrawer, MaxMinValuesDrawer, Drawer

# this is how you can import models from chartilo library:
from chartilo.models import Line, Candle

class Line:
	width = 8 

	def __init__(self, data):
		self.closeTime = data[0] # NOT REQUIRED close time
		self.openPrice = float(data[1]) # open price
		self.closePrice = float(data[4]) # close price

from . import Line


class Candle(Line):
	width = 8 

	def __init__(self, data):
		super().__init__(data); # open and close price look above
		self.maximalPrice = float(data[2]) # maximal price
		self.minimalPrice = float(data[3]) # minimal price

# this is how you can import themes, the default is Dark
from chartilo.themes import Light, Dark, Theme

		# setting data, states and running library
		self._chartilo.setData(data)
		self._chartilo.setStates(states)
		self._chartilo.updateCanvas()