Merge pull request #6 from caelestia-dots/python-rework

feat: rewrite in python

1 files changed, 21 insertions, 0 deletions

diff --git a/src/caelestia/subcommands/wallpaper.py b/src/caelestia/subcommands/wallpaper.py
new file mode 100644
index 0000000..940dcb5
--- /dev/null
+++ b/src/caelestia/subcommands/wallpaper.py
@@ -0,0 +1,21 @@
+import json
+from argparse import Namespace
+
+from caelestia.utils.wallpaper import get_colours_for_wall, get_wallpaper, set_random, set_wallpaper
+
+
+class Command:
+    args: Namespace
+
+    def __init__(self, args: Namespace) -> None:
+        self.args = args
+
+    def run(self) -> None:
+        if self.args.print:
+            print(json.dumps(get_colours_for_wall(self.args.print, self.args.no_smart)))
+        elif self.args.file:
+            set_wallpaper(self.args.file, self.args.no_smart)
+        elif self.args.random:
+            set_random(self.args)
+        else:
+            print(get_wallpaper() or "No wallpaper set")