Merge core_{se,me,pe} into core.{se,me,pe}

2026-01-22 14:41:39 +00:00 · 2016-05-31 22:32:37 -04:00
parent d4919054f9
commit a65077f871
39 changed files with 55 additions and 75 deletions
--- a/core/me/fs.py
+++ b/core/me/fs.py
@@ -0,0 +1,104 @@
+# Created By: Virgil Dupras
+# Created On: 2009-10-23
+# Copyright 2015 Hardcoded Software (http://www.hardcoded.net)
+#
+# This software is licensed under the "GPLv3" License as described in the "LICENSE" file,
+# which should be included with this package. The terms are also available at
+# http://www.gnu.org/licenses/gpl-3.0.html
+
+from hsaudiotag import auto
+from hscommon.util import get_file_ext, format_size, format_time
+
+from core.util import format_timestamp, format_perc, format_words, format_dupe_count
+from core import fs
+
+TAG_FIELDS = {
+    'audiosize', 'duration', 'bitrate', 'samplerate', 'title', 'artist',
+    'album', 'genre', 'year', 'track', 'comment'
+}
+
+class MusicFile(fs.File):
+    INITIAL_INFO = fs.File.INITIAL_INFO.copy()
+    INITIAL_INFO.update({
+        'audiosize': 0,
+        'bitrate': 0,
+        'duration': 0,
+        'samplerate': 0,
+        'artist': '',
+        'album': '',
+        'title': '',
+        'genre': '',
+        'comment': '',
+        'year': '',
+        'track': 0,
+    })
+    __slots__ = fs.File.__slots__ + tuple(INITIAL_INFO.keys())
+
+    @classmethod
+    def can_handle(cls, path):
+        if not fs.File.can_handle(path):
+            return False
+        return get_file_ext(path.name) in auto.EXT2CLASS
+
+    def get_display_info(self, group, delta):
+        size = self.size
+        duration = self.duration
+        bitrate = self.bitrate
+        samplerate = self.samplerate
+        mtime = self.mtime
+        m = group.get_match_of(self)
+        if m:
+            percentage = m.percentage
+            dupe_count = 0
+            if delta:
+                r = group.ref
+                size -= r.size
+                duration -= r.duration
+                bitrate -= r.bitrate
+                samplerate -= r.samplerate
+                mtime -= r.mtime
+        else:
+            percentage = group.percentage
+            dupe_count = len(group.dupes)
+        dupe_folder_path = getattr(self, 'display_folder_path', self.folder_path)
+        return {
+            'name': self.name,
+            'folder_path': str(dupe_folder_path),
+            'size': format_size(size, 2, 2, False),
+            'duration': format_time(duration, with_hours=False),
+            'bitrate': str(bitrate),
+            'samplerate': str(samplerate),
+            'extension': self.extension,
+            'mtime': format_timestamp(mtime, delta and m),
+            'title': self.title,
+            'artist': self.artist,
+            'album': self.album,
+            'genre': self.genre,
+            'year': self.year,
+            'track': str(self.track),
+            'comment': self.comment,
+            'percentage': format_perc(percentage),
+            'words': format_words(self.words) if hasattr(self, 'words') else '',
+            'dupe_count': format_dupe_count(dupe_count),
+        }
+
+    def _get_md5partial_offset_and_size(self):
+        f = auto.File(str(self.path))
+        return (f.audio_offset, f.audio_size)
+
+    def _read_info(self, field):
+        fs.File._read_info(self, field)
+        if field in TAG_FIELDS:
+            f = auto.File(str(self.path))
+            self.audiosize = f.audio_size
+            self.bitrate = f.bitrate
+            self.duration = f.duration
+            self.samplerate = f.sample_rate
+            self.artist = f.artist
+            self.album = f.album
+            self.title = f.title
+            self.genre = f.genre
+            self.comment = f.comment
+            self.year = f.year
+            self.track = f.track
+