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

MediaFile: allow setting date fields to None

Browse source
This commit is contained in:
Adrian Sampson 2014-05-18 15:14:36 -07:00
parent dfa8445980
commit 943cbe0838
2 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
beets/mediafile.py
View file

@ -1057,7 +1057,10 @@ class DateField(MediaField):
return None
def __set__(self, mediafile, date):
self._set_date_tuple(mediafile, date.year, date.month, date.day)
if date is None:
self._set_date_tuple(mediafile, None, None, None)
else:
self._set_date_tuple(mediafile, date.year, date.month, date.day)
def __delete__(self, mediafile):
super(DateField, self).__delete__(mediafile)

7
test/test_mediafile_edge.py
View file

@ -235,6 +235,13 @@ class TypeTest(unittest.TestCase):
self.mf.track = None
self.assertEqual(self.mf.track, 0)
def test_set_date_to_none(self):
self.mf.date = None
self.assertIsNone(self.mf.date)
self.assertIsNone(self.mf.year)
self.assertIsNone(self.mf.month)
self.assertIsNone(self.mf.day)
class SoundCheckTest(unittest.TestCase):
def test_round_trip(self):