mirror of
https://github.com/Benexl/FastAnime.git
synced 2025-12-29 22:20:34 -08:00
218 lines
8.9 KiB
Python
218 lines
8.9 KiB
Python
|
|
from kivy.config import Config,ConfigParser
|
|
# Config.set('kivy', 'window_icon', "logo.ico")
|
|
# Config.write()
|
|
import os
|
|
from kivy.loader import Loader
|
|
Loader.num_workers = 5
|
|
Loader.max_upload_per_frame = 5
|
|
from libs.animdl.animdl_api import AnimdlApi
|
|
os.environ["KIVY_VIDEO"] = "ffpyplayer"
|
|
from kivymd.icon_definitions import md_icons
|
|
import json
|
|
from queue import Queue
|
|
from threading import Thread
|
|
import plyer
|
|
# plyer.facades.StoragePath.get_application_dir
|
|
# plyer.facades.StoragePath.get_downloads_dir
|
|
# plyer.facades.StoragePath.get_videos_dir()
|
|
# plyer.facades.StoragePath.
|
|
# plyer.facades.StoragePath.get_application_dir
|
|
from kivymd.app import MDApp
|
|
|
|
from kivy.uix.settings import SettingsWithSidebar,Settings
|
|
from kivy.uix.screenmanager import ScreenManager,FadeTransition
|
|
from kivy.clock import Clock
|
|
from kivy.storage.jsonstore import JsonStore
|
|
from datetime import date,datetime
|
|
from subprocess import Popen
|
|
from View.screens import screens
|
|
from View import DownloadAnimePopup,AnimdlDialogPopup
|
|
import time
|
|
|
|
import webbrowser
|
|
from Utility import themes_available,show_notification
|
|
|
|
user_data = JsonStore("user_data.json")
|
|
today = date.today()
|
|
now = datetime.now()
|
|
if not(user_data.exists("my_list")):
|
|
user_data.put("my_list",list=[])
|
|
|
|
yt_cache = JsonStore("yt_cache.json")
|
|
links_cache_name= f"{today}{0 if now.hour>=12 else 1}"
|
|
if not(yt_cache.exists("yt_stream_links")):
|
|
yt_cache.put("yt_stream_links",**{f"{links_cache_name}":[]})
|
|
elif not( yt_cache.get("yt_stream_links").get(f"{links_cache_name}")):
|
|
yt_cache.put("yt_stream_links",**{f"{links_cache_name}":[]})
|
|
|
|
|
|
|
|
class AniXStreamApp(MDApp):
|
|
queue = Queue()
|
|
downloads_queue = Queue()
|
|
animdl_streaming_subprocess:Popen|None = None
|
|
def worker(self,queue:Queue):
|
|
while True:
|
|
task = queue.get() # task should be a function
|
|
task()
|
|
self.queue.task_done()
|
|
def downloads_worker(self,queue:Queue):
|
|
while True:
|
|
download_task = queue.get() # task should be a function
|
|
download_task()
|
|
self.downloads_queue.task_done()
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.load_all_kv_files(self.directory)
|
|
self.theme_cls.theme_style = "Dark"
|
|
self.theme_cls.primary_palette = "Lightcoral"
|
|
self.manager_screens = ScreenManager()
|
|
self.manager_screens.transition = FadeTransition()
|
|
|
|
# initialize worker
|
|
self.worker_thread = Thread(target=self.worker,args=(self.queue,))
|
|
self.worker_thread.daemon = True
|
|
self.worker_thread.start()
|
|
|
|
# initialize downloads worker
|
|
self.downloads_worker_thread = Thread(target=self.downloads_worker,args=(self.downloads_queue,))
|
|
self.downloads_worker_thread.daemon = True
|
|
self.downloads_worker_thread.start()
|
|
|
|
def build(self) -> ScreenManager:
|
|
self.settings_cls = SettingsWithSidebar
|
|
self.generate_application_screens()
|
|
if config:=self.config:
|
|
if theme_color:=config.get("Preferences","theme_color"):
|
|
self.theme_cls.primary_palette = theme_color
|
|
if theme_style:=config.get("Preferences","theme_style"):
|
|
self.theme_cls.theme_style = theme_style
|
|
self.anime_screen = self.manager_screens.get_screen("anime screen")
|
|
self.search_screen = self.manager_screens.get_screen("search screen")
|
|
self.download_screen = self.manager_screens.get_screen("downloads screen")
|
|
return self.manager_screens
|
|
|
|
def on_start(self,*args):
|
|
super().on_start(*args)
|
|
|
|
def generate_application_screens(self) -> None:
|
|
for i, name_screen in enumerate(screens.keys()):
|
|
model = screens[name_screen]["model"]()
|
|
controller = screens[name_screen]["controller"](model)
|
|
view = controller.get_view()
|
|
view.manager_screens = self.manager_screens
|
|
view.name = name_screen
|
|
self.manager_screens.add_widget(view)
|
|
|
|
def build_config(self, config):
|
|
config.setdefaults('Preferences', {
|
|
'theme_color': 'Cyan',
|
|
"theme_style": "Dark",
|
|
"downloads_dir": plyer.storagepath.get_videos_dir() if plyer.storagepath.get_videos_dir() else ".",
|
|
"is_startup_anime_enable":False
|
|
})
|
|
print(self.config.get("Preferences","is_startup_anime_enable"))
|
|
|
|
def build_settings(self,settings:Settings):
|
|
settings.add_json_panel("Settings",self.config,"settings.json")
|
|
|
|
|
|
def on_config_change(self, config, section, key, value):
|
|
if section=="Preferences":
|
|
match key:
|
|
case "theme_color":
|
|
if value in themes_available:
|
|
self.theme_cls.primary_palette = value
|
|
else:
|
|
config.set("Preferences","theme_color","Cyan")
|
|
config.write()
|
|
case "theme_style":
|
|
self.theme_cls.theme_style = value
|
|
|
|
def on_stop(self):
|
|
if self.animdl_streaming_subprocess:
|
|
self.animdl_streaming_subprocess.terminate()
|
|
|
|
|
|
# custom methods
|
|
# TODO: move theme to a personalized class
|
|
def search_for_anime(self,search_field,**kwargs):
|
|
if self.manager_screens.current != "search screen":
|
|
self.manager_screens.current = "search screen"
|
|
self.search_screen.handle_search_for_anime(search_field,**kwargs)
|
|
|
|
def show_anime_screen(self,id):
|
|
self.manager_screens.current = "anime screen"
|
|
self.anime_screen.controller.update_anime_view(id)
|
|
|
|
def watch_on_allanime(self,title_):
|
|
"""
|
|
Opens the given anime in your default browser on allanimes site
|
|
Parameters:
|
|
----------
|
|
title_: The anime title requested to be opened
|
|
"""
|
|
if anime:=AnimdlApi.get_anime_url_by_title(title_):
|
|
title,link = anime
|
|
parsed_link = f"https://allmanga.to/bangumi/{link.split('/')[-1]}"
|
|
else:
|
|
show_notification("Failure",f"Failed to open {title} in browser on allanime site")
|
|
if webbrowser.open(parsed_link):
|
|
show_notification("Success",f"Successfully opened {title} in browser allanime site")
|
|
else:
|
|
show_notification("Failure",f"Failed to open {title} in browser on allanime site")
|
|
|
|
|
|
def download_anime_complete(self,successful_downloads:list,failed_downloads:list,anime_title:str):
|
|
show_notification(f"Finished Dowloading {anime_title}",f"There were {len(successful_downloads)} successful downloads and {len(failed_downloads)} failed downloads")
|
|
|
|
def download_anime(self,default_cmds):
|
|
# TODO:Add custom download cmds functionality
|
|
on_progress = lambda *args:self.download_screen.on_episode_download_progress(*args)
|
|
output_path = self.config.get("Preferences","downloads_dir")
|
|
if episodes_range:=default_cmds.get("episodes_range"):
|
|
download_task =lambda: AnimdlApi.download_anime_by_title(default_cmds["title"],on_progress,self.download_anime_complete,output_path,episodes_range) # ,default_cmds["quality"]
|
|
self.downloads_queue.put(download_task)
|
|
else:
|
|
download_task =lambda: AnimdlApi.download_anime_by_title(default_cmds["title"],on_progress,self.download_anime_complete,output_path) # ,default_cmds.get("quality")
|
|
self.downloads_queue.put(download_task)
|
|
|
|
def stream_anime_with_custom_input_cmds(self,*cmds):
|
|
self.animdl_streaming_subprocess = AnimdlApi.run_custom_command("stream",*cmds)
|
|
|
|
def stream_anime_by_title_with_animdl(self,title,episodes_range:str|None=None):
|
|
self.animdl_streaming_subprocess = AnimdlApi.stream_anime_by_title(title,episodes_range)
|
|
# self.stop_streaming = False
|
|
|
|
def watch_on_animdl(self,title_dict:dict|None=None,episodes_range:str|None=None,custom_options:tuple[str]|None=None):
|
|
"""
|
|
Enables you to stream an anime using animdl either by parsing a title or custom animdl options
|
|
|
|
parameters:
|
|
-----------
|
|
title_dict:dict["japanese","kanji"]
|
|
a dictionary containing the titles of the anime
|
|
custom_options:tuple[str]
|
|
a tuple containing valid animdl stream commands
|
|
"""
|
|
if self.animdl_streaming_subprocess:
|
|
self.animdl_streaming_subprocess.terminate()
|
|
|
|
if title_dict:
|
|
if title:=title_dict.get("japanese"):
|
|
stream_func = lambda: self.stream_anime_by_title_with_animdl(title,episodes_range)
|
|
self.queue.put(stream_func)
|
|
elif title:=title_dict.get("english"):
|
|
stream_func = lambda:self.stream_anime_by_title_with_animdl(title,episodes_range)
|
|
self.queue.put(stream_func)
|
|
else:
|
|
stream_func = lambda:self.stream_anime_with_custom_input_cmds(*custom_options)
|
|
self.queue.put(stream_func)
|
|
|
|
if __name__ == "__main__":
|
|
# try:
|
|
AniXStreamApp().run()
|
|
# except:
|
|
# print(plyer.storagepath.get_videos_dir()) |