dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update launcher.py

Browse Source
This commit is contained in:
McCloudS
2024-03-19 14:36:37 -06:00
committed by GitHub
parent 35ad3df87f
commit e81cbd0383
1 changed files with 47 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
launcher.py
View File

@@ -1,12 +1,23 @@
import os
import requests
import subprocess
import argparse
def convert_to_bool(in_bool):
if isinstance(in_bool, bool):
return in_bool
else:
value = str(in_bool).lower()
return value not in ('false', 'off', '0', 0)
# Convert the input to string and lower case, then check against true values
return str(in_bool).lower() in ('true', 'on', '1')
def install_packages_from_requirements(requirements_file, force_update):
try:
# If force_update is True, add the '--upgrade' flag to force an update
if force_update:
subprocess.run(['pip3', 'install', '-r', requirements_file, '--upgrade'], check=True)
else:
subprocess.run(['pip3', 'install', '-r', requirements_file], check=True)
print(f"Requirements from {requirements_file} have been successfully installed.")
except subprocess.CalledProcessError as e:
print(f"Failed to install requirements: {e}")
def download_from_github(url, output_file):
response = requests.get(url)
@@ -18,20 +29,41 @@ def download_from_github(url, output_file):
print(f"Failed to download file from {url}")
def main():
github_url = "https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py"
output_file = "./subgen.py"
# Construct the argument parser
parser = argparse.ArgumentParser()
parser.add_argument('--debug', default=False, action='store_true', help="Enable console debugging (default: False)")
parser.add_argument('--install', default=False, action='store_true', help="Install packages (default: False)")
parser.add_argument('--packageupdate', default=False, action='store_true', help="Force update PIP packages (default: False)")
parser.add_argument('--append', default=False, action='store_true', help="Append 'Transcribed by whisper' to generated subtitle (default: False)")
parser.add_argument('--update', default=False, action='store_true', help="Update Subgen (default: False)")
args = parser.parse_args()
# Set environment variables based on the parsed arguments
os.environ['DEBUG'] = str(args.debug)
os.environ['APPEND'] = str(args.append)
os.environ['UPDATE'] = str(args.update)
# URL to the requirements.txt file on GitHub
requirements_url = "https://raw.githubusercontent.com/McCloudS/subgen/main/requirements.txt"
requirements_file = "requirements.txt"
# Install packages from requirements.txt if the install or packageupdate argument is True
if args.install or args.packageupdate:
install_packages_from_requirements(requirements_file, args.packageupdate)
# Check if the environment variable is set
github_download_enabled = convert_to_bool(os.getenv("UPDATE", True))
subgen_script_name = "./subgen.py"
if not os.path.exists(output_file):
print(f"File {output_file} does not exist. Downloading from GitHub...")
download_from_github(github_url, output_file)
elif github_download_enabled:
print(f"File exists, but UPDATE is set to True. Downloading {output_file} from GitHub...")
download_from_github(github_url, output_file)
if not os.path.exists(subgen_script_name):
print(f"File {subgen_script_name} does not exist. Downloading from GitHub...")
download_from_github("https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py", subgen_script_name)
elif convert_to_bool(os.getenv("UPDATE", "False")) or args.update:
print(f"File exists, but UPDATE is set to True. Downloading {subgen_script_name} from GitHub...")
download_from_github("https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py", subgen_script_name)
else:
print("Environment variable UPDATE is not set or set to False, skipping download.")
subprocess.run(['python3', '-u', 'subgen.py'], check=True)
if __name__ == "__main__":
main()