from PyQt6 import QtCore, QtWidgets, uic

import os 
import backtrader as bt
from logger_module import logger
class StrategyManualUI(QtWidgets.QWidget):

    def __init__(self, controller, parent = None):
        super(StrategyManualUI, self).__init__()

        self.controller = controller
        self.cerebro = controller.cerebro
        self.manualStrategy = None

        self.parent = parent

        # It does not finish by a "/"
        self.current_dir_path = os.path.dirname(os.path.realpath(__file__))

        uic.loadUi( self.current_dir_path + "/ui/strategyManual.ui", self)

        self.buyPB = self.findChild(QtWidgets.QPushButton, "buyPB")
        self.buyPB.clicked.connect(self.buy)

        self.sellPB = self.findChild(QtWidgets.QPushButton, "sellPB")
        self.sellPB.clicked.connect(self.sell)
        self.priceLB = self.findChild(QtWidgets.QLabel, "label_4")

        # Run button
        self.runBacktestBtn = self.findChild(QtWidgets.QPushButton, "runBacktestBtn")
        self.runBacktestBtn.clicked.connect(controller.updateCsvData)

    def buy(self):
        s = self.priceLB.text()
        lines = s.splitlines()
        first_line = lines[0]
        price = float(first_line)
        logger.info(price)
        self.manualStrategy.buy(exectype=bt.Order.Limit, price = price*1.1)
        pass

    def sell(self):
        s = self.priceLB.text()
        lines = s.splitlines()
        first_line = lines[0]
        price = float(first_line)
        self.manualStrategy.sell(exectype=bt.Order.Limit, price= price/1.1)
        pass