Added progressbar based on iterations of number of files in directory.
This commit is contained in:
34
src/core.py
34
src/core.py
@@ -214,39 +214,17 @@ def save_subtitles(files, single=False, directory=None, encoding=None):
|
|||||||
print('Moved: %s ---> %s' % (old, newname))
|
print('Moved: %s ---> %s' % (old, newname))
|
||||||
os.rename(old, newname)
|
os.rename(old, newname)
|
||||||
|
|
||||||
print()
|
|
||||||
|
|
||||||
|
|
||||||
# def refine(video, episode_refiners=None, movie_refiners=None, **kwargs):
|
|
||||||
# """Refine a video using :ref:`refiners`.
|
|
||||||
# .. note::
|
|
||||||
# Exceptions raised in refiners are silently passed and logged.
|
|
||||||
# :param video: the video to refine.
|
|
||||||
# :type video: :class:`~subliminal.video.Video`
|
|
||||||
# :param tuple episode_refiners: refiners to use for episodes.
|
|
||||||
# :param tuple movie_refiners: refiners to use for movies.
|
|
||||||
# :param \*\*kwargs: additional parameters for the :func:`~subliminal.refiners.refine` functions.
|
|
||||||
# """
|
|
||||||
# refiners = ()
|
|
||||||
# if isinstance(video, Episode):
|
|
||||||
# refiners = episode_refiners or ('metadata')
|
|
||||||
# elif isinstance(video, Movie):
|
|
||||||
# refiners = movie_refiners or ('metadata')
|
|
||||||
# for refiner in refiners:
|
|
||||||
# logger.info('Refining video with %s', refiner)
|
|
||||||
# try:
|
|
||||||
# print(refiner)
|
|
||||||
# exit(0)
|
|
||||||
# refiner_manager[refiner].plugin(video, **kwargs)
|
|
||||||
# except:
|
|
||||||
# logger.exception('Failed to refine video')
|
|
||||||
|
|
||||||
def scan_folder(path):
|
def scan_folder(path):
|
||||||
videos = []
|
videos = []
|
||||||
ignored_videos = []
|
ignored_videos = []
|
||||||
errored_paths = []
|
errored_paths = []
|
||||||
logging.debug('Collecting path %s', path)
|
logging.debug('Collecting path %s', path)
|
||||||
|
|
||||||
|
content_count = 0
|
||||||
|
for _ in os.listdir(path):
|
||||||
|
content_count += 1
|
||||||
|
|
||||||
|
with click.progressbar(length=content_count, label='Collecting videos') as bar:
|
||||||
# non-existing
|
# non-existing
|
||||||
if not os.path.exists(path):
|
if not os.path.exists(path):
|
||||||
try:
|
try:
|
||||||
@@ -259,6 +237,7 @@ def scan_folder(path):
|
|||||||
|
|
||||||
refine(video)
|
refine(video)
|
||||||
videos.append(video)
|
videos.append(video)
|
||||||
|
# Increment bar to full ?
|
||||||
|
|
||||||
# directories
|
# directories
|
||||||
if os.path.isdir(path):
|
if os.path.isdir(path):
|
||||||
@@ -274,6 +253,7 @@ def scan_folder(path):
|
|||||||
directory=path).values())
|
directory=path).values())
|
||||||
refine(video)
|
refine(video)
|
||||||
videos.append(video)
|
videos.append(video)
|
||||||
|
bar.update(1)
|
||||||
|
|
||||||
return videos
|
return videos
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user