Adding support for Login screen wallpaper

cinnamon-dynamic-wallpaper@TobiZog/5.4/src/model/main_view_model.py

@@ -2,7 +2,7 @@
 from gi.repository import Gio, Gdk
 
 # Packages
-import os, time, gi, locale
+import os, time, locale, subprocess, sys
 from PIL import Image
 
 # Local scripts
@@ -183,9 +183,20 @@ class Main_View_Model:
 				self.current_image_uri = self.cinnamon_prefs.source_folder + self.cinnamon_prefs.period_images[i]
 				break
 
-		# Set the background
+		# Update the background
 		self.background_settings['picture-uri'] = "file://" + self.current_image_uri
 
+		# Update the login_image
+		if self.cinnamon_prefs.login_image:
+			# Create the folder in /tmp
+			try:
+				os.mkdir("/tmp/cinnamon_dynamic_wallpaper")
+			except:
+				print("Folder already exists")
+			
+			# Copy the current image to the temp folder for the login screen
+			os.system("cp " + self.current_image_uri + " " + "/tmp/cinnamon_dynamic_wallpaper/login_image.jpg")
+
 		# Set background stretching
 		self.background_settings['picture-options'] = self.cinnamon_prefs.picture_aspect
 
@@ -269,3 +280,52 @@ class Main_View_Model:
 		except:
 			self.background_settings['primary-color'] = "#000000"
 			self.background_settings['secondary-color'] = "#000000"
+
+
+	def set_login_image(self):
+		""" Writes a path to file in /tmp/cinnamon_dynamic_wallpaper to display the wallpaper on the login screen
+		"""
+		# New config file content
+		file_content = ""
+
+		# Location of the config file
+		file_location = self.WORKING_DIR + "/slick-greeter.conf"
+
+		if os.path.isfile("/etc/lightdm/slick-greeter.conf"):
+			# File already exists, make a copy of the config
+			with open("/etc/lightdm/slick-greeter.conf", "r") as conf_file:
+				for line in conf_file.readlines():
+					if line.startswith("background"):
+						# Case 1: Preference is already set as expected -> leave function
+						print(line)
+						if line.find("cinnamon_dynamic_wallpaper/login_image.jpg") != -1 and self.cinnamon_prefs.login_image or \
+							line.find("cinnamon_dynamic_wallpaper/login_image.jpg") == -1 and not self.cinnamon_prefs.login_image:
+							return
+						
+						# Case 2: Function enabled -> Set the path to the login image
+						elif self.cinnamon_prefs.login_image:
+							file_content += "background=/tmp/cinnamon_dynamic_wallpaper/login_image.jpg\n"
+						
+						# Case 3: Function disabled -> Remove the custom login image
+						elif not self.cinnamon_prefs.login_image:
+							break
+
+					# Other config lines will be simply copied
+					else:
+						file_content += line
+
+		else:
+			# File doesn't exists	
+			file_content = "[Greeter]\n"
+			file_content += "background=/tmp/cinnamon_dynamic_wallpaper/login_image.jpg"
+
+		# Create the file
+		with open(file_location, "w") as conf_file:
+			conf_file.write(file_content)
+			conf_file.close()
+
+		# Move it to /etc/lightdm
+		if os.path.isfile("/etc/lightdm/slick-greeter.conf"):
+			subprocess.call(['pkexec', 'rm', '/etc/lightdm/slick-greeter.conf', 'mv', file_location, '/etc/lightdm/'])
+		else:
+			subprocess.call(['pkexec', 'mv', file_location, '/etc/lightdm/'])

cinnamon-dynamic-wallpaper@TobiZog/5.4/src/service/cinnamon_pref_handler.py

@@ -73,6 +73,8 @@ class Cinnamon_Pref_Handler:
       self.extract_json('period_9_custom_start_time')
     ]
 
+    self.login_image = self.extract_json('login_image')
+
 
   def value_to_json(self, parameter: str, value: str):
     """ Storing safely a value to the dictionary
@@ -127,6 +129,7 @@ class Cinnamon_Pref_Handler:
     self.value_to_json('period_7_custom_start_time', self.period_custom_start_time[7])
     self.value_to_json('period_8_custom_start_time', self.period_custom_start_time[8])
     self.value_to_json('period_9_custom_start_time', self.period_custom_start_time[9])
+    self.value_to_json('login_image', self.login_image)
 
 
     # Write to file

cinnamon-dynamic-wallpaper@TobiZog/5.4/src/view/main_window.py

@@ -150,6 +150,7 @@ class Main_Window:
 		# Page 3: Behaviour
 		self.cb_picture_aspect: Gtk.ComboBox = self.builder.get_object("cb_picture_aspect")
 		self.sw_dynamic_background_color: Gtk.Switch = self.builder.get_object("sw_dynamic_background_color")
+		self.sw_login_image: Gtk.Switch = self.builder.get_object("sw_login_image")
 
 
 	def show(self):
@@ -178,6 +179,7 @@ class Main_Window:
 		self.add_items_to_combo_box(self.cb_picture_aspect, self.view_model.picture_aspects)
 		self.set_active_combobox_item(self.cb_picture_aspect, self.view_model.cinnamon_prefs.picture_aspect)
 		self.sw_dynamic_background_color.set_active(self.view_model.cinnamon_prefs.dynamic_background_color)
+		self.sw_login_image.set_active(self.view_model.cinnamon_prefs.login_image)
 
 
 		# Show the main window
@@ -693,6 +695,16 @@ class Main_Window:
 		"""
 		self.view_model.cinnamon_prefs.dynamic_background_color = state
 
+	
+	def on_sw_login_image_state_set(self, _: Gtk.Switch, state: bool):
+		""" User switches login background image on or off
+
+		Args:
+				_ (Gtk.Switch): Used Switch
+				state (bool): Current state
+		"""
+		self.view_model.cinnamon_prefs.login_image = state
+
 
 	# About
 
@@ -747,6 +759,7 @@ class Main_Window:
 		# Use the new settings
 		self.view_model.refresh_image()
 		self.view_model.set_background_gradient()
+		self.view_model.set_login_image()
 
 
 	def on_destroy(self, *args):