mirror of
https://github.com/Starlio-app/StarlioX.git
synced 2025-04-07 03:35:27 +01:00
94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
"""
|
||
Copyright 2022 Redume
|
||
|
||
Licensed under the Apache License, Version 2.0 (the "License");
|
||
you may not use this file except in compliance with the License.
|
||
You may obtain a copy of the License at
|
||
|
||
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
||
Unless required by applicable law or agreed to in writing, software
|
||
distributed under the License is distributed on an "AS IS" BASIS,
|
||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
See the License for the specific language governing permissions and
|
||
limitations under the License.
|
||
"""
|
||
|
||
|
||
from bs4 import BeautifulSoup
|
||
from win10toast import ToastNotifier
|
||
from pystray import MenuItem, Menu, Icon
|
||
from PIL import Image
|
||
from elevate import elevate
|
||
|
||
import json
|
||
import requests
|
||
import ctypes
|
||
import os
|
||
import urllib
|
||
import schedule
|
||
import winreg as reg
|
||
import getpass
|
||
import sys
|
||
|
||
|
||
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.toaster = ToastNotifier()
|
||
|
||
def resource_path(self, relative_path):
|
||
base_path = getattr(sys, '_MEIPASS', os.path.dirname(os.path.abspath(__file__)))
|
||
return os.path.join(base_path, relative_path)
|
||
|
||
def tray(self):
|
||
tray = Icon("EveryDayPhotoNasa", title="EveryDayPhotoNasa",
|
||
icon=Image.open(self.resource_path("nasa.ico")),
|
||
menu=Menu(
|
||
MenuItem("Выход",
|
||
self.tray_close),
|
||
))
|
||
tray.run()
|
||
|
||
def tray_close(self, tray):
|
||
tray.visible = False
|
||
sys.exit(0)
|
||
|
||
def download_photo(self):
|
||
try:
|
||
full_page = requests.get(self.url, headers=self.headers)
|
||
soup = BeautifulSoup(full_page.content, 'html.parser')
|
||
lnk = str
|
||
for link in soup.select("img"):
|
||
lnk = link["src"]
|
||
|
||
img = urllib.request.urlopen(self.url + lnk).read()
|
||
out = open(self.photoName, "wb")
|
||
out.write(img)
|
||
out.close()
|
||
self.set_wallpaper()
|
||
except requests.exceptions.ConnectionError:
|
||
return self.toaster.show_toast("EveryDayPhotoNasa",
|
||
"Не получилось подключится к сайту, проверьте подключение к интернету.",
|
||
duration=3,
|
||
icon_path=self.resource_path("nasa.ico"))
|
||
|
||
def set_wallpaper(self):
|
||
path = os.path.abspath(self.photoName)
|
||
ctypes.windll.user32.SystemParametersInfoW(20, 0, path, 0)
|
||
self.toaster.show_toast("EveryDayPhotoNasa",
|
||
"Обои установлены.",
|
||
duration=3,
|
||
icon_path=self.resource_path("nasa.ico"))
|
||
|
||
|
||
nasa = Nasa()
|
||
nasa.download_photo()
|
||
|
||
schedule.every().day.at("00:30").do(nasa.download_photo)
|
||
while True:
|
||
schedule.run_pending()
|
||
nasa.tray()
|