Added downloads folder

This commit is contained in:
Thord Johansson 2024-02-28 10:59:34 +01:00
parent bb507a2778
commit c281a05a39
3 changed files with 98 additions and 0 deletions

3
downloads/scan-downloads.sh Executable file
View file

@ -0,0 +1,3 @@
#!/bin/bash
mkdir -p $HOME/.clamscan/
clamscan -r --remove --log=$HOME/.clamscan/$(date +"%Y%m%d-%H%M%S").log /mnt/data/downloads

95
downloads/sort-downloads.py Executable file
View file

@ -0,0 +1,95 @@
#!/usr/bin/env python
# - generated, gpt3.5
import os
import shutil
import sys
import datetime
from pathlib import Path
# Configurable variables
MOVIES_EXTENSION = {".mkv", ".mp4", ".avi"}
MUSIC_EXTENSION = {".mp3", ".ogg", ".flac", ".wma", ".wav"}
EPISODIC_THRESHOLD = 1 # Maximum number of video files to consider as episodic (inclusive)
MOVIES_PATH = "Movies"
MUSIC_PATH = "Music"
SHOWS_PATH = "Shows"
def organize_folders(directory, root_path):
log_file = create_log_file(directory, root_path)
write_log(log_file, f"Organizing folders in {directory}")
actions = preview_actions(directory, root_path)
if confirm_actions(actions):
execute_actions(actions, log_file)
else:
log_file.close()
os.remove(log_file.name)
print("Log file deleted.")
log_file_path = os.path.abspath(log_file.name)
print(f"Log file saved at: {log_file_path}")
def preview_actions(directory, root_path):
actions = []
for folder_name in os.listdir(directory):
folder_path = os.path.join(directory, folder_name)
if os.path.isdir(folder_path):
file_types = set()
for file in os.listdir(folder_path):
file_extension = Path(file).suffix.lower()
file_types.add(file_extension)
if MOVIES_EXTENSION.intersection(file_types) and len(file_types) <= EPISODIC_THRESHOLD:
actions.append((folder_path, os.path.join(root_path, MOVIES_PATH, folder_name)))
elif len(file_types.intersection(MUSIC_EXTENSION)) == len(file_types):
actions.append((folder_path, os.path.join(root_path, MUSIC_PATH, folder_name)))
elif len(file_types) > EPISODIC_THRESHOLD:
actions.append((folder_path, os.path.join(root_path, SHOWS_PATH, folder_name)))
return actions
def confirm_actions(actions):
print("Preview of actions:")
for source, destination in actions:
print(f"Move: {source} -> {destination}")
user_input = input("Do you want to proceed with the above actions? (y/n): ").lower()
return user_input == "y"
def execute_actions(actions, log_file):
for source, destination in actions:
message = f"Moving {os.path.basename(source)} to {destination}"
write_log(log_file, message)
print(message)
shutil.move(source, destination)
def move_folder(folder_path, destination):
if not os.path.exists(destination):
os.makedirs(destination)
shutil.move(folder_path, os.path.join(destination, os.path.basename(folder_path)))
def create_log_file(directory, root_path):
current_time = datetime.datetime.now().strftime("%y%m%d_%H%M%S")
log_file_name = f"{current_time}.log"
log_file_path = os.path.join(root_path, directory, log_file_name)
return open(log_file_path, "w")
def write_log(log_file, message):
timestamp = datetime.datetime.now().strftime("[%Y-%m-%d %H:%M:%S]")
log_message = f"{timestamp} {message}\n"
log_file.write(log_message)
print(log_message, end="")
if __name__ == "__main__":
if len(sys.argv) != 3:
print("Usage: ./organize_folders.py <directory_path> <root_path>")
sys.exit(1)
directory = sys.argv[1]
root_path = sys.argv[2]
organize_folders(directory, root_path)