theme_dialog.py

import gi
import os
gi.require_version('Gtk','3.0')
from gi.repository import Gtk
from yeelight_lib import Bulb
import JSONutils as themeFile

class ThemeDialog(Gtk.Dialog):

	def __init__(self,parent,bulb):
		Gtk.Dialog.__init__(self,"Themes",
								 parent,
								 0)

		#Get the bulb control
		self.bulb = bulb

		#Load themes
		self.themes = themeFile.loadTheme()

		#Setup GUI
		self.createWidget()
		self.createButtons()

		self.show_all()

	def createWidget(self):

		#Main grid	
		self.contentArea = self.get_content_area()
		
		self.mainGrid = Gtk.Grid() 
		self.mainGrid.set_property("row-spacing",4)
		self.mainGrid.set_property("column-spacing",10)

		self.contentArea.add(self.mainGrid)

		#Setup grid 3xN
		i = 0
		j = 0
		id = 0
		for currentTheme in self.themes.items():
			
			#Create template
			tmpBox = Gtk.Box(orientation=Gtk.STYLE_CLASS_VERTICAL,spacing = 2)

			#Load the image
			tmpImage = Gtk.Image()
			tmpImage.set_from_file("img/themes/"+currentTheme[1]["pic"])

			tmpBox.add(tmpImage)
			
			#Load the label
			tmpButton = Gtk.Button(label=currentTheme[1]["name"])
			tmpButton.connect("clicked",self.buttonPressed,currentTheme[0])

			tmpBox.add(tmpButton)		

			#Add to the grid
			self.mainGrid.attach(tmpBox,j,i,1,1)

			#Continue to the next cell
			j += 1

			if(j == 2):
				i += 1
				j = 0

			id += 1

		# Seperator
		self.seperator = Gtk.Separator(orientation=Gtk.STYLE_CLASS_HORIZONTAL)
		self.contentArea.pack_start(self.seperator,False,True,4)


	def createButtons(self):

		#Cancel button
		self.add_buttons(Gtk.STOCK_CANCEL,Gtk.ResponseType.CANCEL)

	def buttonPressed(self,widget,themeName):

		# Set the bulb property
		self.bulb.setColorInt(self.themes[themeName]["rgb"])
		self.bulb.adjustBrightness(self.themes[themeName]["brightness"])
		self.bulb.adjustTemperature(self.themes[themeName]["temperature"])