OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-29 16:09:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

setup script 2.4.1: fixed Gifski setup issues and enhanced garbage cleaning

Browse Source
This commit is contained in:
K4YT3X 2020-12-13 16:39:13 -05:00
parent df0861ff5c
commit 9e8e95803f
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/video2x_setup.py
View File

@ -4,7 +4,7 @@
Name: Video2X Setup Script Name: Video2X Setup Script
Creator: K4YT3X Creator: K4YT3X
Date Created: November 28, 2018 Date Created: November 28, 2018
Last Modified: November 18, 2020 Last Modified: December 13, 2020
Editor: BrianPetkovsek Editor: BrianPetkovsek
Editor: SAT3LL Editor: SAT3LL
@ -45,7 +45,7 @@ import zipfile
# Therefore, they will be installed during the Python dependency # Therefore, they will be installed during the Python dependency
# installation step and imported later in the script. # installation step and imported later in the script.
SETUP_VERSION = '2.4.0' SETUP_VERSION = '2.4.1'
# global static variables # global static variables
LOCALAPPDATA = pathlib.Path(os.getenv('localappdata')) LOCALAPPDATA = pathlib.Path(os.getenv('localappdata'))
@ -99,8 +99,7 @@ class Video2xSetup:
else: else:
getattr(self, f'_install_{self.driver}')() getattr(self, f'_install_{self.driver}')()
print('\nCleaning up temporary files') # self._cleanup()
self._cleanup()
def _install_python_requirements(self): def _install_python_requirements(self):
""" Read requirements.txt and return its content """ Read requirements.txt and return its content
@ -110,6 +109,8 @@ class Video2xSetup:
def _cleanup(self): def _cleanup(self):
""" Cleanup all the temp files downloaded """ Cleanup all the temp files downloaded
""" """
print('\nCleaning up temporary files')
for file in self.trash: for file in self.trash:
try: try:
if file.is_dir(): if file.is_dir():
@ -153,17 +154,19 @@ class Video2xSetup:
import requests import requests
# Get latest release of Gifski via Github API # Get latest release of Gifski via Github API
latest_release = requests.get('https://api.github.com/repos/ImageOptim/gifski/releases/latest').json() releases = requests.get('https://api.github.com/repos/ImageOptim/gifski/releases').json()
for release in releases:
for a in latest_release['assets']: for asset in release['assets']:
if re.search(r'gifski-.*\.tar\.xz\.7z', a['browser_download_url']): if re.search(r'gifski-.*\.tar\.xz', asset['browser_download_url']):
gifski_tar_gz = download(a['zipball_url'], tempfile.gettempdir()) gifski_tar_xz = download(asset['browser_download_url'], tempfile.gettempdir())
self.trash.append(gifski_tar_gz) self.trash.append(gifski_tar_xz)
# extract and rename # extract and rename
with tarfile.open(gifski_tar_gz) as archive: with tarfile.open(gifski_tar_xz) as archive:
archive.extractall(LOCALAPPDATA / 'video2x' / 'gifski') archive.extractall(LOCALAPPDATA / 'video2x' / 'gifski')
return
def _install_waifu2x_caffe(self): def _install_waifu2x_caffe(self):
""" Install waifu2x_caffe """ Install waifu2x_caffe
""" """
@ -434,18 +437,18 @@ except Exception:
print('An error has occurred') print('An error has occurred')
print('Video2X Automatic Setup has failed') print('Video2X Automatic Setup has failed')
# in case of a failure, try cleaning up temp files EXIT_CODE = 1
# regardless if script finishes successfully or not
# print script execution summary
finally:
# always try cleaning up trash
try: try:
setup._cleanup() setup._cleanup()
except Exception: except Exception:
traceback.print_exc() traceback.print_exc()
print('An error occurred while trying to cleanup files') print('An error occurred while trying to cleanup files')
EXIT_CODE = 1
# regardless if script finishes successfully or not
# print script execution summary
finally:
print('Script finished') print('Script finished')
print(f'Time taken: {timedelta(seconds=round(time.time() - start_time))}') print(f'Time taken: {timedelta(seconds=round(time.time() - start_time))}')