Titlecase the homepath and try except creating hash for filename.
This commit is contained in:
29
src/core.py
29
src/core.py
@@ -13,6 +13,7 @@ import logging
|
||||
import tvdb_api
|
||||
import click
|
||||
from pprint import pprint
|
||||
from titlecase import titlecase
|
||||
|
||||
import env_variables as env
|
||||
|
||||
@@ -72,12 +73,15 @@ def scan_video(path):
|
||||
video.size = os.path.getsize(path)
|
||||
|
||||
# hash of name
|
||||
if isinstance(video, Movie):
|
||||
hash_str = ''.join([video.title, str(video.year) or ''])
|
||||
elif isinstance(video, Episode):
|
||||
hash_str = ''.join([video.series, str(video.season), str(video.episode)])
|
||||
videoHash = hashlib.md5(hash_str.encode()).hexdigest()
|
||||
video.hash = videoHash
|
||||
try:
|
||||
if isinstance(video, Movie):
|
||||
hash_str = ''.join([video.title, str(video.year) or ''])
|
||||
elif isinstance(video, Episode):
|
||||
hash_str = ''.join([video.series, str(video.season), str(video.episode)])
|
||||
videoHash = hashlib.md5(hash_str.encode()).hexdigest()
|
||||
video.hash = videoHash
|
||||
except:
|
||||
print(video)
|
||||
|
||||
return video
|
||||
|
||||
@@ -96,7 +100,6 @@ def scan_subtitle(path):
|
||||
|
||||
return subtitle
|
||||
|
||||
|
||||
def scan_videos(path):
|
||||
"""Scan `path` for videos and their subtitles.
|
||||
|
||||
@@ -118,7 +121,7 @@ def scan_videos(path):
|
||||
# setup progress bar
|
||||
path_children = 0
|
||||
for _ in os.walk(path): path_children += 1
|
||||
with click.progressbar(length=path_children, show_pos=True, label='Searching folders for videos') as bar:
|
||||
with click.progressbar(length=path_children, show_pos=True, label='Collecting videos') as bar:
|
||||
|
||||
# walk the path
|
||||
videos = []
|
||||
@@ -270,7 +273,7 @@ def scan_folder(path):
|
||||
return videos
|
||||
|
||||
def movingToCollege(video, home_path):
|
||||
video.home = home_path
|
||||
video.home = titlecase(home_path)
|
||||
|
||||
def pickforgirlscouts(video):
|
||||
if isinstance(video, Movie):
|
||||
@@ -281,14 +284,14 @@ def pickforgirlscouts(video):
|
||||
|
||||
elif isinstance(video, Episode):
|
||||
if video.series != None and video.season != None and video.episode != None and type(video.episode) != list:
|
||||
home_path = '{} S{:02d}E{:02d}'.format(video.series, video.season, video.episode)
|
||||
home_path = '{} S{}E{}'.format(str(video.series), str(video.season), str(video.episode))
|
||||
movingToCollege(video, home_path)
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def main():
|
||||
path = '/mnt/rescue/'
|
||||
path = '/mnt/mainframe/'
|
||||
|
||||
videos = scan_folder(path)
|
||||
|
||||
@@ -310,8 +313,8 @@ def main():
|
||||
's' if len(videos) > 1 else ''
|
||||
))
|
||||
|
||||
for video in civilian:
|
||||
print(video)
|
||||
for video in scout:
|
||||
print('{} lives: {}'.format(video, video.home))
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user