StarlioX/main.py

"""
   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

import requests
import ctypes
import os
import urllib
import schedule
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()