def load_config(loaded_config):
settings.app_name = loaded_config['Info']['application_name']
settings.delay_time = int(loaded_config['Info']['delay_time'])
settings.debug = True
settings.incoming_dir = loaded_config['Directories']['incoming_directory']
settings.movie_dir = loaded_config['Directories']['movie_directory']
settings.min_movie_size = loaded_config['File Information'][
'minimum_movie_size']
settings.min_episode_size = loaded_config['File Information'][
'minimum_episode_size']
settings.exts_to_delete = [
e.strip() for e in loaded_config['File Information']
['extensions_to_delete'].split(',')
]
settings.video_formats = [
e.strip()
for e in loaded_config['File Information']['video_formats'].split(',')
]
settings.audio_dir = loaded_config['Directories']['audio_directory']
settings.audio_formats = [
e.strip()
for e in loaded_config['File Information']['video_formats'].split(',')
]
# check validity of config entries
for dir_checker in (settings.incoming_dir, settings.movie_dir,
settings.audio_dir):
if not os.path.isdir(dir_checker):
console.print(
f"[red]Directory \"{dir_checker}\" is invalid! Please check the config!",
justify="center")
console.print("Exiting application.", justify="center")
sys.exit()
def main_loop():
# search directories and start figuring out what's what
# returns first level folder names under dir
dirs = get_root_directories()
settings.debug_message("Found directories: {}".format(dirs))
process_folders(dirs)
# check for files that aren't under their own folders for some
# godforsaken reason
files = get_root_files()
if files is not []:
root_level_files(files)
settings.debug_message(
"Sleeping for {} seconds!".format(settings.delay_time)
)
console.print(f"Sleeping for {settings.delay_time} seconds!\r")
time.sleep(int(settings.delay_time))
console.print("Working... \r")
File Name: main.py Author: Joe Kaufeld Email: [email protected] Purpose: Continuously scans a directory where new files are added automatically, make sure they're not currently in use, then try to identify them based on file type, number, and size. It uses that information to attempt placing them in the right final directory. Credits: Special thanks to https://github.com/sposterkil for the regex and code review! ********************************************* ''' import sys from filewatcher.core.initialize import initialize from filewatcher.core.filewatcher import main_loop from filewatcher.core.console import console console.rule("[white]FileWatcher") if __name__ == "__main__": initialize() while True: try: main_loop() except KeyboardInterrupt: console.print() console.rule("[bold red]Exiting!") sys.exit(0)
def generate_config(updated_config=False):
config = ConfigObj()
config.filename = 'config.ini'
config['Info'] = {}
config['Info'].comments.update({
'Info': ['# For themeing purposes!'],
'key': ['application_name']
})
config['Info']['application_name'] = settings.app_name
config['Info'].comments.update({
'Info':
['# Delay between filesystem checks - default is 180 '
'seconds.'],
'key': ['delay_time']
})
config['Info']['delay_time'] = settings.delay_time
config['Directories'] = {}
config['Directories'].comments.update({
'Directories': ['# Example: "C:\Things" or "/Users/[you]/Things"'],
'key': []
})
config['Directories']['incoming_directory'] = settings.incoming_dir
config['Directories']['movie_directory'] = settings.movie_dir
config['Directories']['audio_directory'] = settings.audio_dir
config['File Information'] = {}
config['File Information'].comments.update({
'File Information': [
'# All file sizes are in MB!',
'# The program assumes that all files under a'
'# size threshold are actually TV shows.'
],
'key': []
})
config['File Information']['minimum_movie_size'] = settings.min_movie_size
config['File Information'].comments.update({
'File Information': ['# For auto-deleting crap promo videos'],
'key': ['minimum_episode_size']
})
config['File Information']['minimum_episode_size'] = \
settings.min_episode_size
config['File Information'].comments.update({
'File Information': [
'# List of file extensions worthy of being',
'# immediately deleted.'
],
'key': ['extensions_to_delete']
})
config['File Information'][
'extensions_to_delete'] = settings.exts_to_delete
config['File Information']['video_formats'] = settings.video_formats
config['File Information']['audio_formats'] = settings.audio_formats
config.write()
config_nonexist_error = textwrap.fill(
'Configuration not found, a default config.ini was created.'
' You must edit it before using this script.'
' Application exiting.',
width=70)
config_out_of_date_error = textwrap.fill(
'Configuration found is for an older version of the software. It has'
' been backed up and a new one has been generated; please fill in the'
' required information in the new config.ini and run the program again.',
width=70)
# The only reason we actually perform a check here is if you have an older
# version of the software and we need to update the config file - that way
# we can just update in place and keep rolling.
if updated_config:
console.print(config_out_of_date_error)
else:
console.print(config_nonexist_error)
sys.exit()
def initialize(settings=settings):
set_init_phrase = random.randrange(len(init_phrases))
# start initialization
console.log("{}...".format(init_phrases[set_init_phrase]))
if os.path.isfile("./config.ini"):
loaded_config = ConfigObj('config.ini')
else:
# if this fires, it will exit after building the config
generate_config()
# noinspection PyUnboundLocalVariable
load_config(loaded_config)
console.log(init_endings[set_init_phrase])
console.print()
__version__ = pkg_resources.get_distribution('filewatcher').version
console.print("Ready to go - starting main loop!")
console.print(f"Running {settings.app_name}, version {__version__}")
console.print(f"\nIncoming file directory: {settings.incoming_dir}")
console.print(f"Movie directory: {settings.movie_dir}")
console.print(f"Audio directory: {settings.audio_dir}")
intro_text = (
"Folders identified as containing movies (along with root level "
"files) will be moved to the movie directory. Folders identified "
"as containing TV shows will have their parent folder renamed to "
"have [TV] in front of it and otherwise left alone.")
console.print() # blank line
console.print(textwrap.fill(intro_text, width=70))
# successful init!
return True
def delay_time(self, number):
if number > 0:
self._delay_time = number
else:
console.print(
"Warning! You can't set the delay time to less than 0!")