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

Cleaned up comments and code

Browse source
This commit is contained in:
Austin Marino 2019-10-15 11:45:01 -07:00
parent 4251ff70dc
commit eb6055eeca
1 changed files with 8 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
beetsplug/export.py
View file

@ -66,10 +66,6 @@ class ExportPlugin(BeetsPlugin):
'xml': { 'xml': {
# XML module formatting options. # XML module formatting options.
'formatting': { 'formatting': {
# The output encoding.
'encoding': 'unicode',
# Controls if XML declaration should be added to the file.
# 'xml_declaration': True,
# Can be either "xml", "html" or "text" (default is "xml"). # Can be either "xml", "html" or "text" (default is "xml").
'method': 'xml' 'method': 'xml'
} }
@ -109,8 +105,7 @@ class ExportPlugin(BeetsPlugin):
def run(self, lib, opts, args): def run(self, lib, opts, args):
file_path = opts.output file_path = opts.output
file_mode = 'a' if opts.append else 'w' file_mode = 'a' if opts.append else 'w'
file_format = opts.format if opts.format else \ file_format = opts.format or self.config['default_format'].get(str)
self.config['default_format'].get(str)
format_options = self.config[file_format]['formatting'].get(dict) format_options = self.config[file_format]['formatting'].get(dict)
export_format = ExportFormat.factory( export_format = ExportFormat.factory(
@ -149,10 +144,7 @@ class ExportFormat(object):
self.path = file_path self.path = file_path
self.mode = file_mode self.mode = file_mode
self.encoding = encoding self.encoding = encoding
""" self.out_stream = # creates a file object to write/append or sets to stdout
sys.stdout if path doesn't exit
codecs.open(..) else
"""
self.out_stream = codecs.open(self.path, self.mode, self.encoding) \ self.out_stream = codecs.open(self.path, self.mode, self.encoding) \
if self.path else sys.stdout if self.path else sys.stdout
@ -200,24 +192,17 @@ class XMLFormat(ExportFormat):
def export(self, data, **kwargs): def export(self, data, **kwargs):
# Creates the XML file structure. # Creates the XML file structure.
library = ET.Element(u'library') library = ET.Element(u'library')
tracks_key = ET.SubElement(library, u'key') tracks = ET.SubElement(library, u'tracks')
tracks_key.text = u'tracks'
tracks_dict = ET.SubElement(library, u'dict')
if data and isinstance(data[0], dict): if data and isinstance(data[0], dict):
for index, item in enumerate(data): for index, item in enumerate(data):
track_key = ET.SubElement(tracks_dict, u'key') track = ET.SubElement(tracks, u'track')
track_key.text = str(index)
track_dict = ET.SubElement(tracks_dict, u'dict')
track_details = ET.SubElement(track_dict, u'Track ID')
track_details.text = str(index)
for key, value in item.items(): for key, value in item.items():
track_details = ET.SubElement(track_dict, key) track_details = ET.SubElement(track, key)
track_details.text = value track_details.text = value
# Depending on the version of python the encoding needs to change
# tree = ET.ElementTree(element=library)
try: try:
data = ET.tostring(library, **kwargs) data = ET.tostring(library, encoding='unicode', **kwargs)
except LookupError: except LookupError:
data = ET.tostring(library, encoding='utf-8', method='xml') data = ET.tostring(library, encoding='utf-8', **kwargs)
self.out_stream.write(data) self.out_stream.write(data)