mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-07 12:05:35 +00:00
ef893d601d
Added an admin check, otherwise the script will always fail if launched by a normal user when attempting to add ffmpeg to the system path
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
# 10.12.23 -> 31.01.24
|
|
|
|
# From https://raccoon.ninja/en/dev/using-python-to-check-if-the-application-is-running-as-an-administrator/
|
|
import ctypes, os, sys
|
|
|
|
def isAdmin():
|
|
try:
|
|
is_admin = (os.getuid() == 0)
|
|
except AttributeError:
|
|
is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0
|
|
return is_admin
|
|
|
|
if not isAdmin():
|
|
print("You need to be admin to proceed!")
|
|
sys.exit()
|
|
else:
|
|
print("You're admin! Let's start!")
|
|
|
|
# Class import
|
|
import Src.Api.page as Page
|
|
from Src.Api.film import main_dw_film as download_film
|
|
from Src.Api.tv import main_dw_tv as download_tv
|
|
from Src.Util.Helper.message import msg_start
|
|
from Src.Util.Helper.console import console, msg
|
|
from Src.Upload.update import main_update
|
|
from Src.Util.FFmpeg.installer import check_ffmpeg
|
|
from Src.Util.Helper.os import remove_folder
|
|
|
|
# [ main ]
|
|
def initialize():
|
|
|
|
remove_folder("tmp")
|
|
msg_start()
|
|
|
|
try:
|
|
main_update()
|
|
except Exception as e:
|
|
console.print(f"[blue]Req github [white]=> [red]Failed: {e}")
|
|
|
|
check_ffmpeg()
|
|
print("\n")
|
|
|
|
def display_search_results(db_title):
|
|
for i, title in enumerate(db_title):
|
|
console.print(f"[yellow]{i} [white]-> [green]{title['name']} [white]- [cyan]{title['type']}")
|
|
|
|
def main():
|
|
|
|
initialize()
|
|
domain, site_version = Page.domain_version()
|
|
|
|
film_search = msg.ask("\n[blue]Insert word to search in all site: ").strip()
|
|
db_title = Page.search(film_search, domain)
|
|
display_search_results(db_title)
|
|
|
|
if len(db_title) != 0:
|
|
index_select = int(msg.ask("\n[blue]Index to download: "))
|
|
|
|
if 0 <= index_select <= len(db_title) - 1:
|
|
selected_title = db_title[index_select]
|
|
|
|
if selected_title['type'] == "movie":
|
|
console.print(f"[green]\nMovie select: {selected_title['name']}")
|
|
download_film(selected_title['id'], selected_title['slug'], domain)
|
|
else:
|
|
console.print(f"[green]\nTv select: {selected_title['name']}")
|
|
download_tv(selected_title['id'], selected_title['slug'], site_version, domain)
|
|
|
|
else:
|
|
console.print("[red]Wrong index for selection")
|
|
else:
|
|
console.print("[red]Cant find a single element")
|
|
|
|
console.print("\n[red]Done")
|
|
|
|
if __name__ == '__main__':
|
|
main()
|