1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106 | diff --git a/README.md b/README.md
new file mode 100644
index 0000000..ecd8aaa
--- /dev/null
+++ b/README.md
@@ -0,0 +1,25 @@
+### walrs-fetch
+## Walrs Plugin
+just give it a keywords and its will automatically fetch a wallpaper and generate the palette using walrs
+
+# installation
+```bash
+git clone https://github.com/pixel2175/walrs-fetch
+cd walrs-fetch
+bash
+python -m venv env
+source env/bin/activate
+pip install requests
+```
+
+## how to use
+```bash
+bash
+source env/bin/activate
+python main.py --keywords ergo proxy
+```
+
+!(ergo)[https://pixelll.is-a.dev/static/screenshots/ergo_000.webp]
+!(ergo)[https://pixelll.is-a.dev/static/screenshots/ergo_00.webp]
+!(ergo)[https://pixelll.is-a.dev/static/screenshots/ergo_0.webp]
+
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..7ebe0b4
--- /dev/null
+++ b/main.py
@@ -0,0 +1,69 @@
+from os import path
+import requests
+import random
+import argparse
+
+def fetch_wallpapers(keywords):
+ """
+ Fetch wallpaper links from Wallhaven API based on keywords
+ """
+ base_url = "https://wallhaven.cc/api/v1/search"
+
+ query = '+'.join(keywords)
+
+ params = {
+ 'q': query,
+ 'page': '1',
+ 'sorting': 'random',
+ }
+
+ try:
+ response = requests.get(base_url, params=params)
+ response.raise_for_status()
+ data = response.json()
+
+ wallpapers = data.get('data', [])
+ if len(wallpapers) >1:
+ wallpaper = random.choice(wallpapers)["path"]
+ else:
+ print(wallpapers)
+ exit()
+
+ return str(wallpaper)
+
+ except requests.exceptions.RequestException as e:
+ print(f"Error fetching wallpapers: {e}")
+ return ""
+ except ValueError:
+ print("Error parsing API response")
+ return ""
+
+def main():
+ parser = argparse.ArgumentParser(description="Fetch wallpapers from Wallhaven based on keywords")
+ parser.add_argument("--keywords", nargs='+', required=True, help="Keywords for search")
+ args = parser.parse_args()
+
+ wallpaper_links = fetch_wallpapers(args.keywords)
+
+ if wallpaper_links:
+ from os import system
+ import os
+ from urllib.parse import urlparse
+ link = os.path.basename(urlparse(wallpaper_links).path)
+ r = requests.get(wallpaper_links,stream=True)
+ cache_dir = os.path.expanduser("~/.cache/wallpapers/")
+ link = os.path.basename(urlparse(wallpaper_links).path)
+ cache = os.path.join(cache_dir, link)
+
+ with open(cache,"wb") as f:
+ for chunk in r.iter_content(1024):
+ f.write(chunk)
+ if path.exists("~/.cache/wallpapers/"):
+ system("mkdir ~/.cache/wallpapers")
+ system(f"walrs -i ~/.cache/wallpapers/{link}")
+
+ else:
+ print("No wallpapers found matching your criteria.")
+
+if __name__ == "__main__":
+ main()
|
