diff options
author | dec05eba <dec05eba@protonmail.com> | 2019-11-18 23:13:21 +0100 |
---|---|---|
committer | dec05eba <dec05eba@protonmail.com> | 2020-07-06 07:12:34 +0200 |
commit | 718f9feb8de96be91748521ff52d802b34d5dc73 (patch) | |
tree | 5cb4906a60ea01a7540e5ad9322bb01ed849a9b0 /open_media.py | |
parent | 606c530f50cc410113aa2807d708b6ce6c84cc83 (diff) |
Add script to open media, and track already seen media
Diffstat (limited to 'open_media.py')
-rwxr-xr-x | open_media.py | 89 |
1 files changed, 89 insertions, 0 deletions
diff --git a/open_media.py b/open_media.py new file mode 100755 index 0000000..e373007 --- /dev/null +++ b/open_media.py @@ -0,0 +1,89 @@ +#!/usr/bin/env python + +import os +import sys +import subprocess + +script_dir = os.path.dirname(os.path.realpath(sys.argv[0])) + +def run_dmenu(input): + process = subprocess.Popen(["rofi", "-dmenu", "-i", "-p", "Select media"], stdin=subprocess.PIPE, stdout=subprocess.PIPE) + stdout, stderr = process.communicate(input.encode()) + if process.returncode == 0: + return stdout + else: + print("Failed to launch rofi, error: {}".format(stderr)) + return None + +def add_seen(seen_filepath, media_name, seen_list): + if media_name in seen_list: + return + with open(seen_filepath, "a") as seen_file: + seen_file.write(media_name + "\n") + +def sort_images(filename): + idx = filename.find(".") + if idx != -1: + return int(filename[0:idx]) + return 0 + +def get_downloaded_list(): + process = subprocess.Popen([os.path.join(script_dir, "automedia.py"), "downloaded"], stdout=subprocess.PIPE) + stdout, stderr = process.communicate() + if process.returncode == 0: + return stdout.decode().splitlines() + else: + print("Failed to list downloaded items, error: {}".format(stderr)) + return [] + +def main(): + if len(sys.argv) < 2: + print("usage: open_media.py <download_dir>") + print("example: open_media.sh /home/user/Downloads/automedia") + exit(1) + + download_dir = sys.argv[1] + if not os.path.isdir(download_dir): + print("No such directory: " % (download_dir)) + exit(2) + + downloaded_list = get_downloaded_list() + + seen_filepath = os.path.expanduser("~/.config/automedia/seen") + seen_list = [] + try: + with open(seen_filepath, "r") as seen_file: + seen_list = seen_file.read().splitlines() + except OSError as e: + print("Failed to open {}, reason: {}".format(seen_filepath, str(e))) + + print("seen_list: {}".format(str(seen_list))) + for seen in seen_list: + for i, downloaded in enumerate(downloaded_list): + if seen == downloaded: + downloaded_list[i] = "✓ {}".format(downloaded) + + selected_media = run_dmenu("\n".join(downloaded_list[::-1])) + if not selected_media: + exit(0) + selected_media = selected_media.decode().replace("✓ ", "").rstrip() + + media_path = os.path.join(download_dir, selected_media) + if os.path.isdir(media_path): + add_seen(seen_filepath, selected_media, seen_list) + files = [] + for filename in os.listdir(media_path): + if filename not in ("finished", "session_id"): + files.append(filename) + + files = sorted(files, key=sort_images) + process = subprocess.Popen(["sxiv", "-i", "-f"], stdin=subprocess.PIPE) + files_fullpath = [] + for filename in files: + files_fullpath.append(os.path.join(media_path, filename)) + process.communicate("\n".join(files_fullpath).encode()) + elif os.path.isfile(media_path): + add_seen(seen_filepath, selected_media, seen_list) + subprocess.Popen(["mpv", "--", media_path]) + +main() |