From 66c0be6569e42f759ed8bc99a95296e864cbb12b Mon Sep 17 00:00:00 2001
From: elwld <elwld@libremain.com>
Date: Fri, 12 Aug 2022 18:54:58 +0500
Subject: [PATCH] add func fot rofi

---
 src/wallpyper.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/src/wallpyper.py b/src/wallpyper.py
index d7621c1..6dc9cd3 100755
--- a/src/wallpyper.py
+++ b/src/wallpyper.py
@@ -3,26 +3,35 @@
 from wallpyperFuncs import *
 
 def main():
+    theme = ""
+
     if len(sys.argv) == 2 and sys.argv[1] in ("help","--help","-h"):
         printHelpInfo()
     else:
-        if len(sys.argv) == 1: setTheme()
+        if len(sys.argv) == 1: theme = None
         elif len(sys.argv) == 2 and sys.argv[1] in ("quit", "stop"): killLastSession(); sys.exit(0)
         elif len(sys.argv) >= 2 and sys.argv[1] == "set": 
             if len(sys.argv) == 2:
+                themes = getThemes()
                 try:
                     import dmenu
 
-                    themes = getThemes()
-                    print(themes)
                     theme = dmenu.show(themes, prompt='Themes:')
-                    setTheme(theme)
                 except ModuleNotFoundError:
-                    printHelpInfo()
+                    try:
+                        from rofi import Rofi
+
+                        r = Rofi()
+                        theme = themes[r.select('Themes:', themes)[0]]
+                    except ModuleNotFoundError:
+                        printHelpInfo()
+                        exit(0)
             elif len(sys.argv) == 3:
-                setTheme(sys.argv[2])
+                theme = sys.argv[2]
             else: printHelpInfo()
         else: printHelpInfo()
+    
+    setTheme(theme)
 
 if __name__ == '__main__':
     main()