95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
# python spotify grabber
|
|
# requirements: pygetwindow from pip and fmedia in $path
|
|
import pygetwindow
|
|
import re
|
|
import time
|
|
import subprocess
|
|
import threading
|
|
import logging
|
|
import string
|
|
import unicodedata
|
|
import sys
|
|
|
|
# configuration
|
|
logging.basicConfig(
|
|
level=logging.DEBUG,
|
|
format='%(asctime)s %(levelname)-8s %(message)s'
|
|
)
|
|
rec_path=".\\rec\\"
|
|
loopback_dev="3"
|
|
monitor_interval=.2
|
|
# end configuration
|
|
|
|
valid_filename_chars = "-_.() %s%s" % (string.ascii_letters, string.digits)
|
|
counter=0
|
|
|
|
def clean_filename(filename, whitelist=valid_filename_chars):
|
|
# keep only valid ascii chars
|
|
cleaned_filename = unicodedata.normalize('NFKD', filename).encode('ASCII', 'ignore').decode()
|
|
|
|
# keep only whitelisted chars
|
|
cleaned_filename = ''.join(c for c in cleaned_filename if c in whitelist)
|
|
return cleaned_filename
|
|
|
|
def start_rec(artist, title):
|
|
def rec_subprocess():
|
|
filename = '{rec_path}{nn:03d}. {artist} - {title}.mp3'.format(
|
|
nn=counter,
|
|
rec_path=rec_path,
|
|
artist=clean_filename(artist),
|
|
title=clean_filename(title)
|
|
)
|
|
logging.debug('recording to {filename}'.format(filename=filename))
|
|
command = 'fmedia --record --dev-loopback={device} -o "{filename}" --mpeg-quality=2 --globcmd=listen'.format(device=loopback_dev, filename=filename)
|
|
logging.debug('starting recording subprocess: {command}'.format(command=command))
|
|
subprocess.run(command, shell=True, capture_output=True, check=True)
|
|
|
|
logging.debug('starting subprocess')
|
|
global counter
|
|
counter += 1
|
|
thread = threading.Thread(target=rec_subprocess)
|
|
thread.start()
|
|
|
|
def stop_rec():
|
|
subprocess.run('fmedia --globcmd=stop', shell=True)
|
|
logging.debug('stopped recording')
|
|
|
|
def parse_win_title(win_title):
|
|
metadata = win_title.split('Spotify - ')[1].split(' · ')
|
|
title = metadata[0]
|
|
artist = metadata[1]
|
|
logging.debug('got metadata {title} by {artist}'.format(title=title, artist=artist))
|
|
return title, artist
|
|
|
|
def watch_window():
|
|
old_title = ''
|
|
logging.debug('started window watcher')
|
|
|
|
while True:
|
|
win_titles = pygetwindow.getAllTitles()
|
|
for win_title in win_titles:
|
|
if re.search('^Spotify - .+', win_title):
|
|
if win_title != old_title:
|
|
logging.debug('window title changed to {wintitle}'.format(wintitle=win_title))
|
|
|
|
# read metadata from window title
|
|
title, artist = parse_win_title(win_title)
|
|
|
|
if old_title != '':
|
|
logging.debug('stop recording')
|
|
stop_rec()
|
|
|
|
logging.debug('start recording now')
|
|
start_rec(title, artist)
|
|
|
|
old_title = win_title
|
|
time.sleep(.5)
|
|
|
|
input('press enter to arm')
|
|
try:
|
|
watch_window()
|
|
except KeyboardInterrupt:
|
|
logging.info('got CTRL+C - trying to clean up my mess')
|
|
stop_rec()
|
|
logging.info('Bye Bye')
|
|
sys.exit(0)
|