storm/bragi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

using os.path.split() instead of regex (#375)

Browse Source
This commit is contained in:
duarm 2023-07-31 18:43:44 -03:00 committed by GitHub
parent f67e837cd9
commit f046a3c177
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
media/file.py
View File

@ -84,22 +84,23 @@ class FileItem(BaseItem):
return True
def _get_info_from_tag(self):
match = re.search(r"(.+)\/(.+)\.(.+)", self.uri())
assert match is not None
path, file_name_ext = os.path.split(self.uri())
file_name, ext = os.path.splitext(file_name_ext)
file_path = match[1] + '/'
file_name = match[2]
ext = match[3]
assert path is not None and file_name is not None
try:
im = None
path_thumbnail = file_path + file_name + ".jpg"
path_thumbnail = os.path.join(path, file_name + ".jpg")
if os.path.isfile(path_thumbnail):
im = Image.open(path_thumbnail)
else:
path_thumbnail = os.path.join(path, "cover.jpg")
if os.path.isfile(path_thumbnail):
im = Image.open(path_thumbnail)
elif os.path.isfile(file_path + "cover.jpg"):
im = Image.open(file_path + "cover.jpg")
if ext == "mp3":
if ext == ".mp3":
# title: TIT2
# artist: TPE1, TPE2
# album: TALB
@ -114,7 +115,7 @@ class FileItem(BaseItem):
if "APIC:" in tags:
im = Image.open(BytesIO(tags["APIC:"].data))
elif ext == "m4a" or ext == "m4b" or ext == "mp4" or ext == "m4p":
elif ext == ".m4a" or ext == ".m4b" or ext == ".mp4" or ext == ".m4p":
# title: ©nam (\xa9nam)
# artist: ©ART
# album: ©alb
@ -129,7 +130,7 @@ class FileItem(BaseItem):
if "covr" in tags:
im = Image.open(BytesIO(tags["covr"][0]))
elif ext == "opus":
elif ext == ".opus":
# title: 'title'
# artist: 'artist'
# album: 'album'
@ -162,7 +163,7 @@ class FileItem(BaseItem):
pass
if not self.title:
self.title = os.path.basename(file_path + file_name)
self.title = file_name
@staticmethod
def _prepare_thumbnail(im):