redditwpscript/wallpaper.py at main · BitWire/redditwpscript · GitHub

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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#!/usr/bin/env python3

import requests, random, shutil, subprocess, time, sys
#from difPy import dif
from screeninfo import get_monitors
from configparser import ConfigParser
from os.path import expanduser, dirname, realpath, getctime, splitext, exists
from os import rename, listdir
from pathlib import Path

# Get configfile
directory=dirname(realpath(__file__))
config_object = ConfigParser()
config_object.read(directory +"/config.ini")
config = config_object["CONFIG"]
data = 'no'
arguments = sys.argv
monitor_count = len(get_monitors())

# Only get the monitor count from the config if its set, else use the automatically found
if 'monitors' in config:
    monitor_count = int(config['monitors'])

# Get additional arguments if available
if len(arguments) == 2:
    data = sys.argv[1]

# Set Images by path
def setImages(paths):
    # Call superpaper with the pictures
    path=expanduser("~") + "/Programs/"
    call = [path + 'Superpaper.AppImage','-s'] + paths
    subprocess.check_call(call)

# Get the data link of the subreddit
link='https://www.reddit.com/r/'+ config['subreddit'] +'/.rss'

# Get the Path to save the pictures we will download, if it's not
# there, it will be made
path= expanduser("~") + '/Pictures/' + config['foldername'] + '/'
archivepath= expanduser("~") + '/Pictures/' + config['foldername'] + '/archive/'
Path(path).mkdir(parents=True, exist_ok=True)
Path(archivepath).mkdir(parents=True, exist_ok=True)
paths = []

#cleanup archive
if (data == 'cleanup'):
    search = dif(archivepath, show_output=True, delete=True)
    quit()

#use archive images
if (data == 'local'):
    for i in range(0,monitor_count):
        paths.append(archivepath + random.choice(listdir(archivepath)))
    setImages(paths)
    quit()

# Save wallpapers
for i in range(0,monitor_count):
    # Getting the path of the file
    f_path = path + 'pic_' + str(i)

    if not exists(f_path):
        continue
    # Obtaining the creation time (in seconds)
    # of the file/folder (datatype=int)
    timestamp = getctime(f_path)

    # Converting the time to an epoch string
    # (the output timestamp string would
    # be recognizable by strptime() without
    # format quantifers)
    t_str = time.ctime(timestamp)

    # Converting the string to a time object
    t_obj = time.strptime(t_str)

    # Transforming the time object to a timestamp
    # of ISO 8601 format
    form_t = time.strftime("%Y-%m-%d %H:%M:%S", t_obj)

    # Since colon is an invalid character for a
    # Windows file name Replacing colon with a
    # similar looking symbol found in unicode
    # Modified Letter Colon " " (U+A789)
    form_t = form_t.replace(":", "꞉")
    # Renaming the filename to its timestamp
    rename(f_path, archivepath + '/' + 'pic_' + str(i) + '_' + form_t + splitext(f_path)[1])

# Make the actual request to get the data via RSS
r = requests.get(link, headers={'User-Agent': 'wallpaperscript 0.4'})

if r.status_code != 200:
    print('Reddit gives this error: ' + str(r.status_code))
    quit()

# Parse RSS feed and extract image URLs
root = ET.fromstring(r.text)
ns = {'atom': 'http://www.w3.org/2005/Atom', 'media': 'http://search.yahoo.com/mrss/'}
entries = root.findall('atom:entry', ns)

# Collect all entries that have full-res images
image_urls = []
for entry in entries:
    content = entry.find('atom:content', ns)
    if content is not None and content.text:
        # Find full-resolution i.redd.it links
        imgs = re.findall(r'https://i\.redd\.it/[^"&]+\.(?:jpg|png|jpeg|webp)', content.text)
        if imgs:
            image_urls.append(imgs[0])

if len(image_urls) == 0:
    print('No images found in RSS feed')
    quit()

# Get a picture for every monitor
# Get a picture for every monitor
for i in range(0, monitor_count):
    url = random.choice(image_urls)
    image = requests.get(url, stream=True)
    if image.status_code == 200:
        with open(path + 'pic_' + str(i), 'wb') as f:
            image.raw.decode_content = True
            shutil.copyfileobj(image.raw, f)
    paths.append(path + 'pic_' + str(i))

setImages(paths)
quit()