You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
from watchdog.events import FileSystemEventHandler
|
|
from watchdog.observers import Observer
|
|
import shutil
|
|
import os
|
|
from funcs import get_media_dimensions
|
|
|
|
media_dir = "media"
|
|
stories_dir = os.path.join(media_dir, "stories")
|
|
posts_dir = os.path.join(media_dir, "posts")
|
|
|
|
os.makedirs(stories_dir, exist_ok=True)
|
|
os.makedirs(posts_dir, exist_ok=True)
|
|
|
|
|
|
def is_story(width, height, tolerance=0.02):
|
|
if width == 0 or height == 0:
|
|
return False
|
|
ratio = min(width, height) / max(width, height)
|
|
return abs(ratio - (9 / 16)) <= (9 / 16 * tolerance)
|
|
|
|
|
|
def determine_post_type(filepath):
|
|
lower = filepath.lower()
|
|
if "posts" in lower:
|
|
return "posts"
|
|
try:
|
|
width, height = get_media_dimensions(filepath)
|
|
except Exception as e:
|
|
print(f"Error getting dimensions for {filepath}: {e}")
|
|
return None
|
|
return "stories" if is_story(width, height) else "posts"
|
|
|
|
|
|
class DownloadHandler(FileSystemEventHandler):
|
|
def process_file(self, file_path):
|
|
file = os.path.basename(file_path)
|
|
|
|
# Ignore incomplete or weird temp names
|
|
if "crdownload" in file or file.count("~") != 3:
|
|
return
|
|
|
|
if not os.path.exists(file_path):
|
|
return
|
|
|
|
post_type = determine_post_type(file_path)
|
|
if post_type == "posts":
|
|
dest_dir = posts_dir
|
|
elif post_type == "stories":
|
|
dest_dir = stories_dir
|
|
else:
|
|
print(f"Could not determine post type for {file}. Skipping...")
|
|
return
|
|
|
|
output_path = os.path.join(dest_dir, file)
|
|
|
|
if os.path.exists(output_path):
|
|
print(f"File already exists {output_path}. Removing...")
|
|
os.remove(file_path)
|
|
return
|
|
|
|
shutil.move(file_path, output_path)
|
|
print(f"Moved {file_path} → {output_path}")
|
|
|
|
def on_created(self, event):
|
|
if not event.is_directory:
|
|
self.process_file(event.src_path)
|
|
|
|
def on_moved(self, event):
|
|
if not event.is_directory:
|
|
self.process_file(event.dest_path)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
download_path = os.path.join(os.path.expanduser("~"), "Downloads")
|
|
event_handler = DownloadHandler()
|
|
|
|
# Initial scan for files already in Downloads
|
|
for f in os.listdir(download_path):
|
|
full_path = os.path.join(download_path, f)
|
|
if os.path.isfile(full_path):
|
|
event_handler.process_file(full_path)
|
|
|
|
observer = Observer()
|
|
observer.schedule(event_handler, download_path, recursive=False)
|
|
observer.start()
|
|
|
|
try:
|
|
observer.join()
|
|
except KeyboardInterrupt:
|
|
observer.stop()
|
|
observer.join() |