1 files changed, 57 insertions, 28 deletions
diff --git a/.config/sway/bin/wallpaper.py b/.config/sway/bin/wallpaper.py
index 906b9fe..8838053 100755
--- a/.config/sway/bin/wallpaper.py
+++ b/.config/sway/bin/wallpaper.py
@@ -3,7 +3,7 @@
 
 import sys, os
 import argparse, json
-import random, re, subprocess, time
+import random, re, subprocess, shutil, time
 import datetime
 
 
@@ -32,11 +32,30 @@ parser.add_argument(
     default=False
 )
 
+parser.add_argument(
+    '--consider_daylight',
+    default=False
+)
+
+parser.add_argument(
+    '--brightness_threshold',
+    default=0.35
+)
+
 args = parser.parse_args()
 
 config = json.loads(args.config)
 
 
+if shutil.which('swaybg') is None:
+    sys.exit('swaybg not found')
+
+if args.consider_daylight and shutil.which('sunwait') is None:
+    sys.exit('sunwait not found')
+if args.consider_daylight and shutil.which('identify') is None:
+    sys.exit('identify not found')
+
+
 """
 " Program
 """
@@ -48,6 +67,14 @@ if args.unique:
         config[weight]['id'] = hash(json.dumps(config_item, sort_keys=True))
         already_selected_file_paths[config_item['id']] = []
 
+# set variables for --consider_daylight flag
+location = {}
+if args.consider_daylight:
+    location = {
+        'latitude': float(subprocess.run(['pass', 'latitude'], capture_output=True, text=True).stdout),
+        'longitude': float(subprocess.run(['pass', 'longitude'], capture_output=True, text=True).stdout),
+    }
+
 # run the loop
 running_bg_process = None
 while True:
@@ -94,33 +121,35 @@ while True:
         found_file_paths = filtered_file_paths
 
     # select file
-    selected_file_path = ''
-    tries = 0
-    while True:
-        selected_file_path = random.choice(found_file_paths)
-
-        now_hour = datetime.datetime.now().hour
-        if now_hour > 6 and now_hour < 17: # daylight
-            break
-
-        identify_process = subprocess.run(['identify', '-format', '%[fx:mean]', selected_file_path], capture_output=True, text=True)
-        brightness = float(identify_process.stdout)
-        if brightness < 0.35: # non-bright image
-            break
-        else:
-            # remove identified bright image from possible selections
-            found_file_paths.remove(selected_file_path)
-
-            if args.unique:
-                already_selected_file_paths[selected_config['id']].append(selected_file_path)
-
-        # can't find suitable image
-        tries += 1
-        if tries >= len(found_file_paths) - 1:
-            if args.unique:
-                already_selected_file_paths[selected_config['id']] = []
-
-            break
+    selected_file_path = random.choice(found_file_paths)
+
+    if args.consider_daylight:
+        daytime_process = subprocess.run(['sunwait', 'poll', '{0}N'.format(location['latitude']), '{0}E'.format(location['longitude'])], capture_output=True, text=True)
+        daytime = daytime_process.stdout.strip()
+
+        if daytime == 'NIGHT':
+            tries = 0
+            while True:
+                selected_file_path = random.choice(found_file_paths)
+
+                identify_process = subprocess.run(['identify', '-format', '%[fx:mean]', selected_file_path], capture_output=True, text=True)
+                brightness = float(identify_process.stdout)
+                if brightness < float(args.brightness_threshold): # non-bright image
+                    break
+                else:
+                    # remove identified bright image from possible selections
+                    found_file_paths.remove(selected_file_path)
+
+                    if args.unique:
+                        already_selected_file_paths[selected_config['id']].append(selected_file_path)
+
+                # can't find suitable image
+                tries += 1
+                if tries >= len(found_file_paths) - 1:
+                    if args.unique:
+                        already_selected_file_paths[selected_config['id']] = []
+
+                    break
 
     if args.unique:
         already_selected_file_paths[selected_config['id']].append(selected_file_path)