GitHubMirror/flac2pod
1
0
Fork 0
mirror of https://github.com/rwinkhart/flac2pod.git synced 2026-09-02 07:07:20 -04:00
Code Issues Packages Projects Releases Wiki Activity

Query /proc for process count instead of relying on the "ps" command (fixes Busybox compatibility)

Browse Source
This commit is contained in:
RandyTheSilly
2024-03-24 15:51:59 -04:00
parent ee3e1f9fa1
commit 7411af721d
1 changed files with 23 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/flac2pod.py
+23 -14
View File
@@ -2,16 +2,14 @@
from argparse import ArgumentParser from argparse import ArgumentParser
from mutagen.flac import FLAC from mutagen.flac import FLAC
from os import cpu_count, path, walk from os import cpu_count, getuid, listdir, path, stat, walk
from pathlib import Path from pathlib import Path
from shutil import copy from shutil import copy
from subprocess import check_output, Popen from subprocess import check_output, Popen
from sys import exit as s_exit from sys import exit as s_exit
from time import sleep from time import sleep
# scans source directory for music files, returns lists of album directories and their full paths
# utility functions
def scan_source(_source): def scan_source(_source):
_artist_dirs, _album_dirs, _full_paths = [], [], [] _artist_dirs, _album_dirs, _full_paths = [], [], []
for _root, _directories, _files in walk(path.expanduser(_source)): for _root, _directories, _files in walk(path.expanduser(_source)):
@@ -32,11 +30,25 @@ def scan_source(_source):
_full_paths.append(f"{_album}/{_filename}") _full_paths.append(f"{_album}/{_filename}")
return _album_dirs, _full_paths return _album_dirs, _full_paths
# creates destination directory structure
def create_destination(): def create_destination():
for _dir in source_directories[0]: for _dir in source_directories[0]:
Path(path.expanduser(_dir.replace(path.expanduser(source), path.expanduser(destination))))\ Path(path.expanduser(_dir.replace(path.expanduser(source), path.expanduser(destination))))\
.mkdir(0o700, parents=True, exist_ok=True) .mkdir(0o700, parents=True, exist_ok=True)
# counts how many ffmpeg processes are running under the current user
def count_user_ffmpeg_processes():
user = getuid()
proc_directory = '/proc'
running_processes = 0
subdirectories = [d for d in listdir(proc_directory) if path.isdir(path.join(proc_directory, d))]
for subdirectory in subdirectories:
if subdirectory.isnumeric():
if stat(proc_directory + '/' + subdirectory).st_uid == user:
with open (proc_directory + '/' + subdirectory + '/comm', 'r') as f:
if f.read().strip() == 'ffmpeg':
running_processes += 1
return running_processes
def scan_destination_convert(): def scan_destination_convert():
_i, _i_total, _total_files = 0, 0, len(source_directories[1]) _i, _i_total, _total_files = 0, 0, len(source_directories[1])
@@ -56,12 +68,11 @@ def scan_destination_convert():
.replace("'", "\\'").replace(')', '\\)').replace('(', '\\(').replace(']', '\\]') \ .replace("'", "\\'").replace(')', '\\)').replace('(', '\\(').replace(']', '\\]') \
.replace('[', '\\[').replace('&', '\\&').replace('`', '\\`').replace('$', '\\$')[ .replace('[', '\\[').replace('&', '\\&').replace('`', '\\`').replace('$', '\\$')[
:-5] + '.m4a' :-5] + '.m4a'
_active_processes = int(check_output('ps -C ffmpeg | wc -l', shell=True)) - 1 _active_processes = count_user_ffmpeg_processes()
if _active_processes >= cpu_count(): while _active_processes >= cpu_count():
while _active_processes >= cpu_count(): print(f"\nThere are already {_active_processes} processes running on your {cpu_count()} threads...\n")
print(f"\nThere are already {cpu_count()} processes running...\n") sleep(1)
sleep(1) _active_processes = count_user_ffmpeg_processes()
_active_processes = int(check_output('ps -C ffmpeg | wc -l', shell=True)) - 1
_i += 1 _i += 1
print(f"\n{_progress}% | \u001b[38;5;0;48;5;15mConverting {_file_s} to {_file_d}...\u001b[0m\n") print(f"\n{_progress}% | \u001b[38;5;0;48;5;15mConverting {_file_s} to {_file_d}...\u001b[0m\n")
# get relevant ReplayGain info # get relevant ReplayGain info
@@ -113,16 +124,14 @@ def scan_destination_convert():
if args.stopifnogain: if args.stopifnogain:
s_exit(1) s_exit(1)
Popen(_cmd, shell=True) Popen(_cmd, shell=True)
_active_processes = int(check_output('ps -C ffmpeg | wc -l', shell=True)) - 1 _active_processes = count_user_ffmpeg_processes()
while _active_processes != 0: while _active_processes != 0:
print('\nPlease wait for the remaining conversions to complete...\n') print('\nPlease wait for the remaining conversions to complete...\n')
sleep(1) sleep(1)
_active_processes = int(check_output('ps -C ffmpeg | wc -l', shell=True)) - 1 _active_processes = count_user_ffmpeg_processes()
sleep(1)
print('\niPod conversion complete!\n') print('\niPod conversion complete!\n')
s_exit(0) s_exit(0)
# argument parsing # argument parsing
if __name__ == "__main__": if __name__ == "__main__":
parser = ArgumentParser(description='Convert your existing FLAC library to be played on an iPod Classic.') parser = ArgumentParser(description='Convert your existing FLAC library to be played on an iPod Classic.')