mirror of
https://github.com/rembo10/headphones.git
synced 2026-07-20 07:54:01 +01:00
update Beets
This commit is contained in:
AdeHub
+98
-79
@@ -15,22 +15,23 @@
|
||||
"""
|
||||
|
||||
|
||||
import sys
|
||||
import codecs
|
||||
import json
|
||||
import csv
|
||||
import json
|
||||
import sys
|
||||
from datetime import date, datetime
|
||||
from xml.etree import ElementTree
|
||||
|
||||
from datetime import datetime, date
|
||||
from beets.plugins import BeetsPlugin
|
||||
from beets import ui
|
||||
from beets import util
|
||||
import mediafile
|
||||
|
||||
from beets import ui, util
|
||||
from beets.plugins import BeetsPlugin
|
||||
from beetsplug.info import library_data, tag_data
|
||||
|
||||
|
||||
class ExportEncoder(json.JSONEncoder):
|
||||
"""Deals with dates because JSON doesn't have a standard"""
|
||||
|
||||
def default(self, o):
|
||||
if isinstance(o, (datetime, date)):
|
||||
return o.isoformat()
|
||||
@@ -38,89 +39,99 @@ class ExportEncoder(json.JSONEncoder):
|
||||
|
||||
|
||||
class ExportPlugin(BeetsPlugin):
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
|
||||
self.config.add({
|
||||
'default_format': 'json',
|
||||
'json': {
|
||||
# JSON module formatting options.
|
||||
'formatting': {
|
||||
'ensure_ascii': False,
|
||||
'indent': 4,
|
||||
'separators': (',', ': '),
|
||||
'sort_keys': True
|
||||
}
|
||||
},
|
||||
'jsonlines': {
|
||||
# JSON Lines formatting options.
|
||||
'formatting': {
|
||||
'ensure_ascii': False,
|
||||
'separators': (',', ': '),
|
||||
'sort_keys': True
|
||||
}
|
||||
},
|
||||
'csv': {
|
||||
# CSV module formatting options.
|
||||
'formatting': {
|
||||
# The delimiter used to seperate columns.
|
||||
'delimiter': ',',
|
||||
# The dialect to use when formating the file output.
|
||||
'dialect': 'excel'
|
||||
}
|
||||
},
|
||||
'xml': {
|
||||
# XML module formatting options.
|
||||
'formatting': {}
|
||||
self.config.add(
|
||||
{
|
||||
"default_format": "json",
|
||||
"json": {
|
||||
# JSON module formatting options.
|
||||
"formatting": {
|
||||
"ensure_ascii": False,
|
||||
"indent": 4,
|
||||
"separators": (",", ": "),
|
||||
"sort_keys": True,
|
||||
}
|
||||
},
|
||||
"jsonlines": {
|
||||
# JSON Lines formatting options.
|
||||
"formatting": {
|
||||
"ensure_ascii": False,
|
||||
"separators": (",", ": "),
|
||||
"sort_keys": True,
|
||||
}
|
||||
},
|
||||
"csv": {
|
||||
# CSV module formatting options.
|
||||
"formatting": {
|
||||
# The delimiter used to separate columns.
|
||||
"delimiter": ",",
|
||||
# The dialect to use when formatting the file output.
|
||||
"dialect": "excel",
|
||||
}
|
||||
},
|
||||
"xml": {
|
||||
# XML module formatting options.
|
||||
"formatting": {}
|
||||
},
|
||||
# TODO: Use something like the edit plugin
|
||||
# 'item_fields': []
|
||||
}
|
||||
# TODO: Use something like the edit plugin
|
||||
# 'item_fields': []
|
||||
})
|
||||
)
|
||||
|
||||
def commands(self):
|
||||
cmd = ui.Subcommand('export', help='export data from beets')
|
||||
cmd = ui.Subcommand("export", help="export data from beets")
|
||||
cmd.func = self.run
|
||||
cmd.parser.add_option(
|
||||
'-l', '--library', action='store_true',
|
||||
help='show library fields instead of tags',
|
||||
"-l",
|
||||
"--library",
|
||||
action="store_true",
|
||||
help="show library fields instead of tags",
|
||||
)
|
||||
cmd.parser.add_option(
|
||||
'-a', '--album', action='store_true',
|
||||
"-a",
|
||||
"--album",
|
||||
action="store_true",
|
||||
help='show album fields instead of tracks (implies "--library")',
|
||||
)
|
||||
cmd.parser.add_option(
|
||||
'--append', action='store_true', default=False,
|
||||
help='if should append data to the file',
|
||||
"--append",
|
||||
action="store_true",
|
||||
default=False,
|
||||
help="if should append data to the file",
|
||||
)
|
||||
cmd.parser.add_option(
|
||||
'-i', '--include-keys', default=[],
|
||||
action='append', dest='included_keys',
|
||||
help='comma separated list of keys to show',
|
||||
"-i",
|
||||
"--include-keys",
|
||||
default=[],
|
||||
action="append",
|
||||
dest="included_keys",
|
||||
help="comma separated list of keys to show",
|
||||
)
|
||||
cmd.parser.add_option(
|
||||
'-o', '--output',
|
||||
help='path for the output file. If not given, will print the data'
|
||||
"-o",
|
||||
"--output",
|
||||
help="path for the output file. If not given, will print the data",
|
||||
)
|
||||
cmd.parser.add_option(
|
||||
'-f', '--format', default='json',
|
||||
help="the output format: json (default), jsonlines, csv, or xml"
|
||||
"-f",
|
||||
"--format",
|
||||
default="json",
|
||||
help="the output format: json (default), jsonlines, csv, or xml",
|
||||
)
|
||||
return [cmd]
|
||||
|
||||
def run(self, lib, opts, args):
|
||||
file_path = opts.output
|
||||
file_mode = 'a' if opts.append else 'w'
|
||||
file_format = opts.format or self.config['default_format'].get(str)
|
||||
file_format_is_line_based = (file_format == 'jsonlines')
|
||||
format_options = self.config[file_format]['formatting'].get(dict)
|
||||
file_mode = "a" if opts.append else "w"
|
||||
file_format = opts.format or self.config["default_format"].get(str)
|
||||
file_format_is_line_based = file_format == "jsonlines"
|
||||
format_options = self.config[file_format]["formatting"].get(dict)
|
||||
|
||||
export_format = ExportFormat.factory(
|
||||
file_type=file_format,
|
||||
**{
|
||||
'file_path': file_path,
|
||||
'file_mode': file_mode
|
||||
}
|
||||
**{"file_path": file_path, "file_mode": file_mode},
|
||||
)
|
||||
|
||||
if opts.library or opts.album:
|
||||
@@ -130,17 +141,18 @@ class ExportPlugin(BeetsPlugin):
|
||||
|
||||
included_keys = []
|
||||
for keys in opts.included_keys:
|
||||
included_keys.extend(keys.split(','))
|
||||
included_keys.extend(keys.split(","))
|
||||
|
||||
items = []
|
||||
for data_emitter in data_collector(
|
||||
lib, ui.decargs(args),
|
||||
album=opts.album,
|
||||
lib,
|
||||
ui.decargs(args),
|
||||
album=opts.album,
|
||||
):
|
||||
try:
|
||||
data, item = data_emitter(included_keys or '*')
|
||||
data, item = data_emitter(included_keys or "*")
|
||||
except (mediafile.UnreadableFileError, OSError) as ex:
|
||||
self._log.error('cannot read file: {0}', ex)
|
||||
self._log.error("cannot read file: {0}", ex)
|
||||
continue
|
||||
|
||||
for key, value in data.items():
|
||||
@@ -158,13 +170,17 @@ class ExportPlugin(BeetsPlugin):
|
||||
|
||||
class ExportFormat:
|
||||
"""The output format type"""
|
||||
def __init__(self, file_path, file_mode='w', encoding='utf-8'):
|
||||
|
||||
def __init__(self, file_path, file_mode="w", encoding="utf-8"):
|
||||
self.path = file_path
|
||||
self.mode = file_mode
|
||||
self.encoding = encoding
|
||||
# creates a file object to write/append or sets to stdout
|
||||
self.out_stream = codecs.open(self.path, self.mode, self.encoding) \
|
||||
if self.path else sys.stdout
|
||||
self.out_stream = (
|
||||
codecs.open(self.path, self.mode, self.encoding)
|
||||
if self.path
|
||||
else sys.stdout
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def factory(cls, file_type, **kwargs):
|
||||
@@ -183,17 +199,19 @@ class ExportFormat:
|
||||
|
||||
class JsonFormat(ExportFormat):
|
||||
"""Saves in a json file"""
|
||||
def __init__(self, file_path, file_mode='w', encoding='utf-8'):
|
||||
|
||||
def __init__(self, file_path, file_mode="w", encoding="utf-8"):
|
||||
super().__init__(file_path, file_mode, encoding)
|
||||
|
||||
def export(self, data, **kwargs):
|
||||
json.dump(data, self.out_stream, cls=ExportEncoder, **kwargs)
|
||||
self.out_stream.write('\n')
|
||||
self.out_stream.write("\n")
|
||||
|
||||
|
||||
class CSVFormat(ExportFormat):
|
||||
"""Saves in a csv file"""
|
||||
def __init__(self, file_path, file_mode='w', encoding='utf-8'):
|
||||
|
||||
def __init__(self, file_path, file_mode="w", encoding="utf-8"):
|
||||
super().__init__(file_path, file_mode, encoding)
|
||||
|
||||
def export(self, data, **kwargs):
|
||||
@@ -205,23 +223,24 @@ class CSVFormat(ExportFormat):
|
||||
|
||||
class XMLFormat(ExportFormat):
|
||||
"""Saves in a xml file"""
|
||||
def __init__(self, file_path, file_mode='w', encoding='utf-8'):
|
||||
|
||||
def __init__(self, file_path, file_mode="w", encoding="utf-8"):
|
||||
super().__init__(file_path, file_mode, encoding)
|
||||
|
||||
def export(self, data, **kwargs):
|
||||
# Creates the XML file structure.
|
||||
library = ElementTree.Element('library')
|
||||
tracks = ElementTree.SubElement(library, 'tracks')
|
||||
library = ElementTree.Element("library")
|
||||
tracks = ElementTree.SubElement(library, "tracks")
|
||||
if data and isinstance(data[0], dict):
|
||||
for index, item in enumerate(data):
|
||||
track = ElementTree.SubElement(tracks, 'track')
|
||||
track = ElementTree.SubElement(tracks, "track")
|
||||
for key, value in item.items():
|
||||
track_details = ElementTree.SubElement(track, key)
|
||||
track_details.text = value
|
||||
# Depending on the version of python the encoding needs to change
|
||||
try:
|
||||
data = ElementTree.tostring(library, encoding='unicode', **kwargs)
|
||||
data = ElementTree.tostring(library, encoding="unicode", **kwargs)
|
||||
except LookupError:
|
||||
data = ElementTree.tostring(library, encoding='utf-8', **kwargs)
|
||||
data = ElementTree.tostring(library, encoding="utf-8", **kwargs)
|
||||
|
||||
self.out_stream.write(data)
|
||||
|
||||
Reference in New Issue
Block a user