StockAverageGraph/application.pyw at master · kertmannik/StockAverageGraph · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import sys
import os
from PyQt5.QtWidgets import QDialog, QApplication, QMessageBox
from PyQt5.QtGui import QPixmap
from Yahoo import api as yahoo
from GUI.design import Dialog
from GUI.plotGenerator import PlotGenerator
from Calculations.nDayAverage import MovingAverage
from Calculations.growth import Growth
from Variables.variables import *


def is_valid_date(date):
    return date is not "" and len(date) == 10


class AppWindow(QDialog):
    def __init__(self):
        super().__init__()
        self.ui = Dialog()
        self.ui.setupUi(self)

        self.closing_prices = None
        self.dates = None

        self.stock = None
        self.stock_index = None

        self.moving_average = MovingAverage()
        self.growth = Growth()
        self.plot_generator = PlotGenerator(self.moving_average, self.growth)

        self.ui.uuesti.clicked.connect(lambda: self.display_data(True))
        self.ui.display.clicked.connect(lambda: self.display_data(False))

    def display_data(self, recalculate):
        try:
            if self.ui.comboBox.currentData() is not None:
                temp_begin = self.ui.daysbegin.date().toString('yyyy-MM-dd')
                if is_valid_date(temp_begin):
                    temp_end = self.ui.daysend.date().toString('yyyy-MM-dd')
                    if is_valid_date(temp_end):
                        if os.path.exists(DEFAULT_PICTURE_NAME):
                            os.remove(DEFAULT_PICTURE_NAME)
                        if recalculate:
                            if self.stock_index is not self.ui.comboBox.currentIndex():
                                self.ui.comboBox.setCurrentIndex(self.stock_index)
                        else:
                            stock_name = self.ui.comboBox.currentData()
                            self.closing_prices, self.dates = yahoo.load_yahoo_quote(stock_name, temp_begin, temp_end)
                            self.stock = stock_name
                            self.stock_index = self.ui.comboBox.currentIndex()
                        days = self.is_valid_number(self.ui.daysaverage.text())
                        self.plot_generator.create_image(self.stock, days, self.closing_prices, self.dates, recalculate)
                        self.ui.update_start_time(self.dates[0])
                        self.ui.pilt.setScaledContents(True)
                        self.ui.pilt.setPixmap(QPixmap(DEFAULT_PICTURE_NAME))
        except Exception as exception:
            QMessageBox.warning(None, "Veateade", str(exception))

    def is_valid_number(self, days_from_ui):
        days_from_ui = int(days_from_ui)
        if self.closing_prices is None:
            raise Exception("Puuduvad andmed arvutamiseks!")
        if days_from_ui < 2:
            raise Exception("Jooksva keskmise arvutamise päevi peab olema vähemalt 2!")
        if (len(self.closing_prices) - 2) < days_from_ui:
            raise Exception("Vaja pikemat ajaperioodi arvutamiseks!")
        return days_from_ui


app = QApplication(sys.argv)
w = AppWindow()
w.show()
sys.exit(app.exec_())