„FarbenVorigerNaechsertTag.py“ hinzufügen
This commit is contained in:
parent
fb536e7183
commit
a82dd4a8ca
|
@ -0,0 +1,162 @@
|
|||
# main.py
|
||||
|
||||
# Import statements
|
||||
import kivy
|
||||
import numpy as np
|
||||
import matplotlib as mpl
|
||||
import matplotlib.pyplot as plt
|
||||
from datetime import datetime, date, timedelta
|
||||
from kivy.app import App
|
||||
from kivy.lang import Builder
|
||||
from kivy.uix.boxlayout import BoxLayout
|
||||
from kivy.uix.floatlayout import FloatLayout
|
||||
from kivy.uix.label import Label
|
||||
from kivy.uix.screenmanager import ScreenManager, Screen
|
||||
from kivy.uix.togglebutton import ToggleButton
|
||||
from kivy.uix.scrollview import ScrollView
|
||||
from kivy.garden.matplotlib import FigureCanvasKivyAgg
|
||||
from awattar import AwattarClient
|
||||
|
||||
# Create AwattarClient instance
|
||||
client = AwattarClient('AT')
|
||||
|
||||
# Set initial date and get energy prices
|
||||
date = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
|
||||
data = client.request(date)
|
||||
hours = np.arange(24)
|
||||
prices = np.array([item.marketprice/10 for item in data])
|
||||
|
||||
# Define DiagramWidget class
|
||||
class DiagramWidget(FigureCanvasKivyAgg):
|
||||
def __init__(self, figure, **kwargs):
|
||||
super().__init__(figure, **kwargs)
|
||||
self.ax = self.figure.axes[0]
|
||||
figure.patch.set_facecolor('black')
|
||||
self.minimum_height = 100 # Set a minimum height for the widget
|
||||
self.ax.set_facecolor('black') # set background color to black
|
||||
self.ax.tick_params(colors='white') # set tick color to white
|
||||
for spine in self.ax.spines.values():
|
||||
spine.set_edgecolor('white') # set spine edge color to white
|
||||
spine.set_facecolor('black') # set spine face color to none
|
||||
|
||||
# Define MyBoxLayout widget
|
||||
class MyBoxLayout(BoxLayout):
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
|
||||
# Set widget properties
|
||||
self.orientation = 'vertical'
|
||||
self.background_color = (0, 0, 0, 1) # set background color to black
|
||||
self.date_label = Label(text=date.strftime('%d.%m.%Y'), font_size='20sp', halign='left', valign='middle', size_hint=(1, None), height='50dp', pos_hint={'top': 1.0, 'x': 0.02})
|
||||
self.add_widget(self.date_label)
|
||||
|
||||
# Define color map based on energy prices
|
||||
cmap = mpl.colors.ListedColormap(['blue', 'green', 'yellow', 'red'])
|
||||
bounds = [-float('inf'), 0, 5, 20, float('inf')] # Include 0 in color map bounds
|
||||
norm = mpl.colors.BoundaryNorm(bounds, cmap.N)
|
||||
color_map = cmap(norm(prices))
|
||||
|
||||
# Create plot figure
|
||||
self.fig, self.ax = plt.subplots(figsize=(10, 5))
|
||||
self.fig.subplots_adjust(left=0.075, bottom=0.175) # Adjust bottom margin to make space for buttons
|
||||
|
||||
# Add bars to plot using color map
|
||||
for i in range(len(hours)):
|
||||
self.ax.bar(hours[i], prices[i], width=0.5, edgecolor='white', linewidth=0.5, color=color_map[i])
|
||||
|
||||
# Set plot limits and ticks
|
||||
self.ax.set(xlim=(-1, 24), xticks=hours)
|
||||
|
||||
# Create DiagramWidget and add it to the ScrollView
|
||||
self.diagram_widget = DiagramWidget(figure=self.fig)
|
||||
scroll_view = ScrollView(size_hint=(1, 0.6)) # Make ScrollView take up 60% of screen height
|
||||
scroll_view.add_widget(self.diagram_widget)
|
||||
self.add_widget(scroll_view)
|
||||
|
||||
# Add toggle buttons
|
||||
toggle_box = BoxLayout(orientation='horizontal', size_hint=(1, 0.1)) # Make toggle_box take up 10% of screen height
|
||||
toggle_box.spacing = '5dp'
|
||||
toggle_box.padding = '5dp'
|
||||
toggle_button1 = ToggleButton(text='Netto/Brutto')
|
||||
toggle_button2 = ToggleButton(text='Stromherkunft einblenden')
|
||||
toggle_box.add_widget(toggle_button1)
|
||||
toggle_box.add_widget(toggle_button2)
|
||||
self.add_widget(toggle_box)
|
||||
|
||||
# Add navigation buttons
|
||||
button_box = BoxLayout(orientation='horizontal', size_hint=(1, 0.1)) # Make button_box take up 10% of screen height
|
||||
button_box.spacing = '5dp'
|
||||
button_box.padding = '5dp'
|
||||
prev_button = ToggleButton(text='Prev Day')
|
||||
prev_button.bind(on_press=self.previous_day)
|
||||
next_button = ToggleButton(text='Next Day')
|
||||
next_button.bind(on_press=self.next_day)
|
||||
button_box.add_widget(prev_button)
|
||||
button_box.add_widget(next_button)
|
||||
self.add_widget(button_box)
|
||||
|
||||
# Update energy prices and plot
|
||||
self.update_prices()
|
||||
|
||||
# Update energy prices and plot
|
||||
def update_prices(self):
|
||||
global prices, date
|
||||
date_prices = client.request(date)
|
||||
prices = np.array([item.marketprice/10 for item in date_prices])
|
||||
|
||||
# Define color map based on energy prices
|
||||
cmap = mpl.colors.ListedColormap(['blue', 'green', 'yellow', 'red'])
|
||||
bounds = [-float('inf'), 0, 5, 20, float('inf')] # Include 0 in color map bounds
|
||||
norm = mpl.colors.BoundaryNorm(bounds, cmap.N)
|
||||
color_map = cmap(norm(prices))
|
||||
|
||||
# Clear existing plot and add bars using color map
|
||||
self.ax.clear()
|
||||
for i in range(len(hours)):
|
||||
self.ax.bar(hours[i], prices[i], width=0.5, edgecolor='white', linewidth=0.5, color=color_map[i])
|
||||
|
||||
# Set plot limits and ticks
|
||||
self.ax.set(xlim=(-1, 24), xticks=hours)
|
||||
|
||||
self.date_label.text = date.strftime('%d.%m.%Y')
|
||||
self.diagram_widget.draw()
|
||||
|
||||
# Navigate to previous day and update energy prices and plot
|
||||
def previous_day(self, instance):
|
||||
global date
|
||||
date = date - timedelta(days=1)
|
||||
self.update_prices()
|
||||
|
||||
# Navigate to next day and update energy prices and plot
|
||||
def next_day(self, instance):
|
||||
global date
|
||||
date = date + timedelta(days=1)
|
||||
self.update_prices()
|
||||
|
||||
# Dynamically adjust height of ScrollView and DiagramWidget
|
||||
def on_size(self, *args):
|
||||
self.diagram_widget.height = max(self.height * 0.6 - 100, self.diagram_widget.minimum_height) # Adjust height to make space for toggle_box and button_box
|
||||
self.diagram_widget.width = self.width
|
||||
|
||||
# Define WattWatchMain widget
|
||||
class WattWatchMain(Screen):
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
layout = MyBoxLayout()
|
||||
self.add_widget(layout)
|
||||
|
||||
# Define WattWatchApp class
|
||||
class WattWatchApp(App):
|
||||
title = 'WattWatch'
|
||||
icon = 'wwicon.png'
|
||||
|
||||
def build(self):
|
||||
sm = ScreenManager()
|
||||
my_screen = WattWatchMain(name='WattWatchMain')
|
||||
sm.add_widget(my_screen)
|
||||
sm.current = 'WattWatchMain'
|
||||
return sm
|
||||
|
||||
# Run the app
|
||||
if __name__ == '__main__':
|
||||
WattWatchApp().run()
|
Loading…
Reference in New Issue