/
util.py
56 lines (37 loc) · 1.53 KB
/
util.py
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
import config
import praw
import re
import os
import urllib.request
import ctypes
reddit_instance = praw.Reddit(client_id=config.REDDIT_CLIENT_ID,
client_secret=config.REDDIT_CLIENT_SECRET,
user_agent=config.USER_AGENT)
def get_subreddit(subreddit_name):
return reddit_instance.subreddit(subreddit_name)
def get_top_posts(subreddit, num_post, time='all'):
return subreddit.top(limit=num_post, time_filter=time)
def get_hot_posts(subreddit, num_post):
return subreddit.hot(limit=num_post)
def get_image_urls(posts):
image_urls = []
for submission in posts:
if not submission.over_18:
url = re.sub(R"\?.*", "", submission.url) # remove trailing arguments
if url.endswith(".jpg") or url.endswith(".png"):
image_urls.append(submission.url)
return image_urls
def download_image_url(url, dir_name):
final_directory = ensure_dir(dir_name)
image_file_path = os.path.join(final_directory, url.split('/')[-1])
urllib.request.urlretrieve(url, image_file_path)
return image_file_path
def ensure_dir(dir_name):
current_directory = os.getcwd()
final_directory = os.path.join(current_directory, config.WALLPAPER_DIR, dir_name)
if not os.path.exists(final_directory):
os.makedirs(final_directory)
return final_directory
def set_windows_desktop_background(image_file_path):
SPI_SETDESKWALLPAPER = 20
ctypes.windll.user32.SystemParametersInfoW(SPI_SETDESKWALLPAPER, 0, image_file_path, 0)