Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-06 16:42:42 +01:00
Code Issues Projects Releases Wiki Activity

Add option to import using URL

Browse source
This commit is contained in:
Alok Saboo 2023-03-18 18:23:53 -04:00
parent 2e18f84b38
commit f051d67947
1 changed files with 32 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
beetsplug/embedart.py
View file

@ -15,14 +15,16 @@
"""Allows beets to embed album art into file metadata.""" """Allows beets to embed album art into file metadata."""
import os.path import os.path
from io import BytesIO
import requests
from PIL import Image
from beets import art, config, ui
from beets.plugins import BeetsPlugin from beets.plugins import BeetsPlugin
from beets import ui from beets.ui import decargs, print_
from beets.ui import print_, decargs from beets.util import bytestring_path, displayable_path, normpath, syspath
from beets.util import syspath, normpath, displayable_path, bytestring_path
from beets.util.artresizer import ArtResizer from beets.util.artresizer import ArtResizer
from beets import config
from beets import art
def _confirm(objs, album): def _confirm(objs, album):
@ -78,12 +80,13 @@ class EmbedCoverArtPlugin(BeetsPlugin):
embed_cmd = ui.Subcommand( embed_cmd = ui.Subcommand(
'embedart', help='embed image files into file metadata' 'embedart', help='embed image files into file metadata'
) )
embed_cmd.parser.add_option( embed_cmd.parser.add_option('-f', '--file', metavar='PATH',
'-f', '--file', metavar='PATH', help='the image file to embed' help='the image file to embed')
) embed_cmd.parser.add_option("-y", "--yes", action="store_true",
embed_cmd.parser.add_option( help="skip confirmation")
"-y", "--yes", action="store_true", help="skip confirmation" embed_cmd.parser.add_option('-u', '--url', metavar='URL',
) help='the URL of the image file to embed')
maxwidth = self.config['maxwidth'].get(int) maxwidth = self.config['maxwidth'].get(int)
quality = self.config['quality'].get(int) quality = self.config['quality'].get(int)
compare_threshold = self.config['compare_threshold'].get(int) compare_threshold = self.config['compare_threshold'].get(int)
@ -107,13 +110,29 @@ class EmbedCoverArtPlugin(BeetsPlugin):
art.embed_item(self._log, item, imagepath, maxwidth, art.embed_item(self._log, item, imagepath, maxwidth,
None, compare_threshold, ifempty, None, compare_threshold, ifempty,
quality=quality) quality=quality)
elif opts.url:
response = requests.get(opts.url)
img = Image.open(BytesIO(response.content))
if img.format:
with open('temp.jpg', 'wb') as f:
f.write(response.content)
opts.file = 'temp.jpg'
# Confirm with user.
if not opts.yes and not _confirm(items, not opts.url):
return
for item in items:
art.embed_item(self._log, item, opts.file, maxwidth,
None, compare_threshold, ifempty,
quality=quality)
os.remove(opts.file)
else:
self._log.error('Invalid image file')
return
else: else:
albums = lib.albums(decargs(args)) albums = lib.albums(decargs(args))
# Confirm with user. # Confirm with user.
if not opts.yes and not _confirm(albums, not opts.file): if not opts.yes and not _confirm(albums, not opts.file):
return return
for album in albums: for album in albums:
art.embed_album(self._log, album, maxwidth, art.embed_album(self._log, album, maxwidth,
False, compare_threshold, ifempty, False, compare_threshold, ifempty,