add methods to extract base album title from an album title string

+ add methods to extract album versions
+ implement these methods in the album model class
This commit is contained in:
geoffrey45
2023-04-22 15:12:59 +03:00
parent 1fc340cb95
commit 3d5ee855d8
3 changed files with 59 additions and 1 deletions
+48
View File
@@ -1,4 +1,5 @@
import re
from enum import Enum
def split_artists(src: str, with_and: bool = False):
@@ -84,3 +85,50 @@ def parse_feat_from_title(title: str) -> tuple[list[str], str]:
# remove "feat" group from title
new_title = re.sub(regex, "", title, flags=re.IGNORECASE)
return artists, new_title
def get_base_album_title(string):
pattern = re.compile(r'\s*(\(|\[).*?(version|remaster|deluxe|edition|expanded).*?(\)|\])', re.IGNORECASE)
match = pattern.search(string)
if match:
removed_block = match.group(0)
title = string.replace(removed_block, '').strip('()[] ')
return title, removed_block
return string, None
class AlbumVersionEnum(Enum):
REMASTER = ("remaster", "remastered")
DELUXE = ("deluxe",)
EXPANDED = ("expanded",)
SUPER_DELUXE = ("super deluxe",)
EXTENDED = ("extended",)
BONUS_TRACK = ("bonus track", "bonus tracks")
RE_RECORD = ("re-recorded", "rerecorded")
INTL_VERSION = ("international",)
ORIGINAL = ("original",)
RE_MIX = ("re-mix",)
RE_RECORDED = ("re-recorded", "rerecorded")
REISSUE = ("reissue",)
def get_album_info(bracket_text: str | None) -> list[str]:
if not bracket_text:
return []
versions = []
for version_keywords in AlbumVersionEnum:
for keyword in version_keywords.value:
if re.search(keyword, bracket_text, re.IGNORECASE):
versions.append(version_keywords.name.lower())
break
return versions
def get_base_title_and_versions(album: str) -> tuple[str, list[str]]:
album_title, version_block = get_base_album_title(album)
versions = get_album_info(version_block)
return album_title, versions