Exemplo n.º 1
0
    def __init__(self):

        # initialize inherited class, resize, center, set title and icon
        super().__init__()
        self.resize(900, 500)
        self.setWindowTitle('Stock Gui v0.00')
        self.setWindowIcon(QtGui.QIcon('pngs/wolf.png'))
        center(self)

        # set default format for date strings and plots
        self.format = '%d %b %Y'
        self.linesty = 'o-'
        self.lw = 2
        self.ms = 4
        plt.style.use('bmh')

        # set up a HBoxLayout, set up an AddStrToListWidget, add it to the layout
        layout = QtGui.QHBoxLayout(self)
        self.ticker_widget = AddStrToListWidget(parent_method=self.updateAxes,
            title='Ticker Symbols', text0='Enter Stock Ticker', unique=True,
            Nlimit=5, upper=True)
        layout.addWidget(self.ticker_widget)

        # define a figure, canvas on the figure, and nav-bar on the canvas
        self.figure = plt.figure()
        canvas = FigureCanvas(self.figure)
        navbar = NavigationToolbar(canvas, self)

        # set up a VBoxLayout on the HBoxLayout for the canvas and nav-bar
        fig_layout = QtGui.QVBoxLayout()
        layout.addLayout(fig_layout)

        # add the nav-bar then the canvas to fig_layout
        fig_layout.addWidget(navbar)
        fig_layout.addWidget(canvas)

        # create an HBoxLayout, add to fig_layout
        dates_layout = QtGui.QHBoxLayout()
        fig_layout.addLayout(dates_layout)

        # get today and today-30days in self.format, initialize QLineEdits
        # for text entry, create a QCheckBox for scaling data
        today = datetime.date.today()
        today_minus30 = today - datetime.timedelta(days=365)
        self.start_edit = QtGui.QLineEdit(today_minus30.strftime(self.format))
        self.end_edit = QtGui.QLineEdit(today.strftime(self.format))
        self.scale_box = QtGui.QCheckBox()

        # add QLineEdits to dates_layout, with QLabels, wire the date boxes
        # and the scale check-box to the dateChanged method
        dates_layout.addWidget(QtGui.QLabel('Start Date:'))
        dates_layout.addWidget(self.start_edit)
        dates_layout.addWidget(QtGui.QLabel('End Date:'))
        dates_layout.addWidget(self.end_edit)
        dates_layout.addWidget(QtGui.QLabel('Scale by %'))
        dates_layout.addWidget(self.scale_box)
        self.start_edit.editingFinished.connect(self.dateChanged)
        self.end_edit.editingFinished.connect(self.dateChanged)
        self.scale_box.stateChanged.connect(self.dateChanged)
Exemplo n.º 2
0
    def __init__(self):
        """ initialize method """

        # define the inheritance, initialize class data
        super().__init__()
        self.dir0 = os.path.expanduser('~')     # default data directory
        self.sname = ''                         # default data filename
        self.format = '%d %B %Y'                # default date format
        self.origin = None                      # current flight origin
        plt.style.use('bmh')

        # set the list of supported flight metrics
        self.supported_metrics = ['Sale Total, USD', 'Number of Layovers',
            'Total Flight Duration, Hours', 'Hours In Air', 'Hours In Layover']

        # set the number of flights to search for each flight combination
        # cannot exceed 500 by rules of the QPX API
        self.n_flights = 100

        # resize, center, set window title and icon
        self.resize(900, 500)
        self.setWindowTitle('Basic Flight Gui v0.00')
        self.setWindowIcon(QtGui.QIcon('wolf.png'))
        center(self)

        # add tab panels
        self.sPanel = QtGui.QWidget()
        self.addTab(self.sPanel, 'Flight &Selection')
        self.rPanel = QtGui.QWidget()
        self.addTab(self.rPanel, 'Flight &Results')
        self.currentChanged.connect(self.misc_action)

        # set up a VBoxLayout on the selection panel, and an HBoxLayout on it
        main_layout = QtGui.QVBoxLayout(self.sPanel)
        items_layout = QtGui.QHBoxLayout()
        main_layout.addLayout(items_layout)

        # set up 3 AddStrToListWidget's on items_layout
        self.origin_widget = AddStrToListWidget(parent_method=self.misc_action,
            title='Origin Airport Codes', text0='Enter 3-Letter Code', unique=True,
            isDate=False, Nexact=3, upper=True)
        items_layout.addWidget(self.origin_widget)
        self.dest_widget = AddStrToListWidget(parent_method=self.misc_action,
            title='Destination Airport Codes', text0='Enter 3-Letter Code', unique=True,
            isDate=False, Nexact=3, upper=True)
        items_layout.addWidget(self.dest_widget)
        self.dates_widget = AddStrToListWidget(parent_method=self.misc_action,
            title='Flight Dates', text0='Enter Flight Date',
            isDate=True, unique=True, format=self.format)
        items_layout.addWidget(self.dates_widget)

        # store names for the three AddStrToListWidgets in class data
        self.widget_list = ['origin_widget', 'dest_widget', 'dates_widget']

        # create another HBoxLayout for misc. gui elements, add to main_layout
        misc_layout = QtGui.QHBoxLayout()
        main_layout.addLayout(misc_layout)

        # create widgets for displaying nFlights, load/save search data
        self.n_display = QtGui.QLabel()
        self.clear_button = QtGui.QPushButton('&Clear Search')
        self.load_button = QtGui.QPushButton('&Load Search')
        self.save_button = QtGui.QPushButton('Sa&ve Search')
        self.flight_button = QtGui.QPushButton('Search Flights')

        # add all pushbuttons to the gui and set actions
        misc_layout.addWidget(self.n_display)
        misc_layout.addWidget(self.clear_button)
        misc_layout.addWidget(self.load_button)
        misc_layout.addWidget(self.save_button)
        misc_layout.addWidget(self.flight_button)
        misc_layout.setSpacing(10)
        self.clear_button.clicked.connect(self.clear_action)
        self.load_button.clicked.connect(self.load_action)
        self.save_button.clicked.connect(self.save_action)
        self.flight_button.clicked.connect(self.flight_action)

        # set initial gui visibilities, set initial flight counter text
        self.quik_vis(results=False, clear=False, save=False, flights=False)
        self.count_flights()

        # set up flight results panel with a HBoxLayout
        results_layout = QtGui.QHBoxLayout(self.rPanel)

        # create a VBoxLayout with 350 bottom margin, add to results_layout
        setup_layout = QtGui.QVBoxLayout()
        #setup_layout.setContentsMargins(0, 0, 0, 350)
        results_layout.addLayout(setup_layout)

        # add the matplotlib canvas to results_layout
        self.figure = plt.figure()
        self.canvas = FigureCanvas(self.figure)
        results_layout.addWidget(self.canvas)

        # add a drop-down menu for selecting flight origin, wire to init_results
        setup_layout.addWidget(QtGui.QLabel('Flight Origin'))
        self.origin_combo = QtGui.QComboBox()
        setup_layout.addWidget(self.origin_combo)
        self.origin_combo.currentIndexChanged.connect(self.init_results)

        # add a drop-down menu for selecting flight metric, wire to init_results
        setup_layout.addWidget(QtGui.QLabel('Flight Metric'))
        self.metric_combo = QtGui.QComboBox()
        [self.metric_combo.addItem(x) for x in self.supported_metrics]
        setup_layout.addWidget(self.metric_combo)
        self.metric_combo.currentIndexChanged.connect(self.init_results)

        # add radio buttons for switching between 1 and n flights
        radio_group = QtGui.QGroupBox()
        radio_layout = QtGui.QVBoxLayout()
        radio_group.setLayout(radio_layout)
        self.radio_button_1 = QtGui.QRadioButton(
            'Show Least Expensive Flight\nper Destination per Day')
        self.radio_button_1.setChecked(True)
        self.radio_button_n = QtGui.QRadioButton('...')
        radio_layout.addWidget(self.radio_button_1)
        radio_layout.addWidget(self.radio_button_n)
        self.flight_slider = QtGui.QSlider(QtCore.Qt.Horizontal)
        self.flight_slider.setEnabled(False)
        radio_layout.addWidget(self.flight_slider)
        radio_layout.setSpacing(10)
        setup_layout.addWidget(radio_group)

        # add radio buttons for additional filtering
        filter_group = QtGui.QGroupBox('Filter By Additional Flight Metrics')
        filter_group.setCheckable(True)
        filter_group.setChecked(False)
        filter_layout = QtGui.QVBoxLayout()
        filter_group.setLayout(filter_layout)
        filter_layout.addWidget(QtGui.QLabel('Layovers, 0 to ...'))
        filter_layout.addWidget(QtGui.QSlider(QtCore.Qt.Horizontal))
        filter_layout.addWidget(QtGui.QLabel('Total Flight Hrs, 0 to ...'))
        filter_layout.addWidget(QtGui.QSlider(QtCore.Qt.Horizontal))
        filter_layout.addWidget(QtGui.QLabel('Total Hrs in Layover, 0 to ...'))
        filter_layout.addWidget(QtGui.QSlider(QtCore.Qt.Horizontal))
        filter_layout.setSpacing(10)
        setup_layout.addWidget(filter_group)