mirror of
https://github.com/Starlio-app/Starlio-web.git
synced 2024-11-05 17:03:58 +03:00
Автоматические космические обои
This commit is contained in:
commit
a8696aa97b
1 changed files with 75 additions and 0 deletions
75
main.py
Normal file
75
main.py
Normal file
|
@ -0,0 +1,75 @@
|
|||
import math
|
||||
from bs4 import BeautifulSoup
|
||||
import requests
|
||||
|
||||
import ctypes
|
||||
import os
|
||||
import urllib
|
||||
|
||||
from PIL import Image
|
||||
from screeninfo import get_monitors
|
||||
|
||||
|
||||
class Nasa:
|
||||
|
||||
def __init__(self):
|
||||
self.url = "https://apod.nasa.gov/apod/"
|
||||
self.headers = {'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_3) '
|
||||
'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.149 Safari/537.36'}
|
||||
self.photoName = "everydayphotonasa.jpg"
|
||||
self.screen_size = (self.get_size_screen())
|
||||
|
||||
def download_photo(self):
|
||||
full_page = requests.get(self.url, headers=self.headers)
|
||||
soup = BeautifulSoup(full_page.content, 'html.parser')
|
||||
for link in soup.select("img"):
|
||||
lnk = link["src"]
|
||||
print(f"Сохарняю картинку — {self.url + lnk}")
|
||||
|
||||
img = urllib.request.urlopen(self.url + lnk).read()
|
||||
out = open(self.photoName, "wb")
|
||||
out.write(img)
|
||||
out.close()
|
||||
|
||||
def get_size_photo(self):
|
||||
image = Image.open(self.photoName)
|
||||
width, height = image.size
|
||||
return width, height
|
||||
|
||||
def get_size_screen(self):
|
||||
for m in get_monitors():
|
||||
width = m.width
|
||||
height = m.height
|
||||
return width, height
|
||||
|
||||
def set_wallpaper(self):
|
||||
path = os.path.abspath(self.photoName)
|
||||
ctypes.windll.user32.SystemParametersInfoW(20, 0, path, 0)
|
||||
|
||||
def aspect_ratio(self, width, height):
|
||||
k = math.gcd(width, height)
|
||||
return width // k, height // k
|
||||
|
||||
def valid_image(self):
|
||||
w_image, h_image = self.get_size_photo()
|
||||
w_image, h_image = self.aspect_ratio(w_image, h_image)
|
||||
w_screen, h_screen = self.aspect_ratio(self.screen_size[0], self.screen_size[1])
|
||||
return w_image == w_screen and h_image == h_screen
|
||||
|
||||
def crop_image(self):
|
||||
size_screen = self.get_size_screen()
|
||||
img = Image.open(self.photoName)
|
||||
img_crop = img.crop((0, 0, size_screen[0], size_screen[1]))
|
||||
img_crop.save(self.photoName)
|
||||
|
||||
def start(self):
|
||||
self.download_photo()
|
||||
if not self.valid_image():
|
||||
self.crop_image()
|
||||
|
||||
self.set_wallpaper()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
nasa = Nasa()
|
||||
nasa.start()
|
Loading…
Reference in a new issue