Manual info correction, input() using readline

This commit is contained in:
DarkCat09 2023-02-08 14:07:51 +04:00
parent 450421ce0d
commit 9687313cb4
3 changed files with 114 additions and 59 deletions

6
.gitignore vendored
View file

@ -1,5 +1,9 @@
files/
convert/
tagged/
.vscode/
lyrics.txt
input
__pycache__/
.vscode/
.idea/

View file

@ -11,7 +11,7 @@ import mimetypes
import subprocess
from typing import TypedDict
from typing import Optional
from typing import Optional, Any
import re
@ -51,6 +51,16 @@ class ParseResult(TypedDict):
cover_mime: Optional[str]
parsed = ParseResult(
title='', artist='',
album='', year=0,
track_no=0, tracks=0,
lyrics='',
cover=None,
cover_mime=None,
)
class ParseError(Exception):
def __init__(self, parsing_obj: str) -> None:
@ -58,10 +68,13 @@ class ParseError(Exception):
super().__init__(
f'Unable to parse {parsing_obj}'
)
self.parsing_obj = parsing_obj
def main() -> None:
global parsed
copy = int(sys.argv[1]) == 1
file = sys.argv[2]
@ -76,34 +89,64 @@ def main() -> None:
print('Title:', title)
correct = input().strip()
parsed: Optional[ParseResult] = None
if correct == '!--':
parsed = manual_info_input()
manual_info_input()
else:
if correct != '':
title = correct.lower()
try:
url = search_azurl(title)
print(url)
parsed = parse_azlyrics(url)
parse_azlyrics(url)
except Exception as err:
print(err)
if isinstance(err, ParseError) \
and err.parsing_obj == 'edit':
pass
else:
print(
'In most cases, this error means '
'In most cases, this error means that '
'the script have received some incorrect data, '
'so you should enter song info manually.'
)
parsed = manual_info_input()
manual_info_input(False)
#print(parsed)
tagmp3(file, parsed, copy)
def input(msg: str = '', def_: Any = '') -> str:
subprocess.call(
(
f'read -e -r -i "{def_}" -p "{msg}" input; '
'echo -n "$input" >./input'
),
shell=True,
executable='bash',
)
try:
with open('./input', 'rt', encoding='utf-8') as f:
return f.read() \
.removesuffix('\n') \
.removesuffix('\r')
except Exception:
return def_
def input_num(msg: str, def_: int = 0) -> int:
try:
return int(input(msg))
return int(input(msg, def_))
except ValueError:
return def_
@ -149,17 +192,15 @@ def search_azurl(title: str) -> str:
'https://searx.dc09.ru/search',
params={
'q': f'{title} site:azlyrics.com',
'category_general': 1,
'language': 'ru-RU',
'time_range': '',
'safesearch': 0,
'theme': 'simple',
},
)
soup = BeautifulSoup(page.text, 'html.parser')
link = soup.select_one(
'div#urls>article>h3>a[href*="azlyrics.com/lyrics/"]'
'div#urls>article>h3>a'
'[href*="azlyrics.com/lyrics/"]'
)
if link is None:
@ -168,16 +209,9 @@ def search_azurl(title: str) -> str:
return str(link.get('href'))
def parse_azlyrics(link: str) -> ParseResult:
def parse_azlyrics(link: str) -> None:
result = ParseResult(
title='', artist='',
album='', year=0,
track_no=0, tracks=0,
lyrics='',
cover=None,
cover_mime=None,
)
global parsed
print('Please wait...')
@ -192,19 +226,23 @@ def parse_azlyrics(link: str) -> ParseResult:
)
if lyrics is None:
raise ParseError('song lyrics')
result['lyrics'] = lyrics.get_text().strip()
parsed['lyrics'] = lyrics.get_text().strip()
artist_elem = soup.select_one(f'{LYRICS_ROW}>.lyricsh>h2')
if artist_elem is None:
raise ParseError('artist name')
result['artist'] = artist_elem.get_text() \
.removesuffix(' Lyrics') \
.strip()
lyrics_file = Path('.') / 'lyrics.txt'
with lyrics_file.open('wt', encoding='utf-8') as f:
f.write(parsed['lyrics'])
title_elem = soup.select_one(f'{LYRICS_ROW}>b')
if title_elem is None:
raise ParseError('song title')
result['title'] = title_elem.get_text().strip('" ')
parsed['title'] = title_elem.get_text().strip('" ')
artist_elem = soup.select_one(f'{LYRICS_ROW}>.lyricsh>h2')
if artist_elem is None:
raise ParseError('artist name')
parsed['artist'] = artist_elem.get_text() \
.removesuffix(' Lyrics') \
.strip()
album_blocks = soup.select('.songinalbum_title')
album = None
@ -223,8 +261,8 @@ def parse_azlyrics(link: str) -> ParseResult:
if album_re is None:
raise ParseError('album name')
result['album'] = album_re[1]
result['year'] = int(album_re[2])
parsed['album'] = album_re[1]
parsed['year'] = int(album_re[2])
cover = album.select_one('img.album-image')
@ -235,8 +273,8 @@ def parse_azlyrics(link: str) -> ParseResult:
cover_url = BASEURL + cover_url
req = session.get(cover_url)
result['cover'] = req.content
result['cover_mime'] = req.headers.get(
parsed['cover'] = req.content
parsed['cover_mime'] = req.headers.get(
'Content-Type', 'image/jpeg'
)
@ -246,14 +284,14 @@ def parse_azlyrics(link: str) -> ParseResult:
tracklist = tracklist_elem.select(
'.listalbum-item'
)
result['tracks'] = len(tracklist)
parsed['tracks'] = len(tracklist)
current_url = re.search(
r'/(lyrics/.+?\.html)',
link,
)
result['track_no'] = 0
parsed['track_no'] = 0
if current_url is not None:
for i, track in enumerate(tracklist):
@ -263,23 +301,32 @@ def parse_azlyrics(link: str) -> ParseResult:
track_href = str(track_url.get('href'))
if current_url[0] in track_href:
result['track_no'] = (i + 1)
parsed['track_no'] = (i + 1)
break
return result
print('Succesfully parsed')
print('Title:', parsed['title'])
print('Artist:', parsed['artist'])
print('Album:', parsed['album'])
print('Track:', parsed['track_no'], '/', parsed['tracks'])
print('Correct something?')
if input('[y/N] ').lower == 'y':
manual_info_input(False)
else:
print()
def manual_info_input() -> ParseResult:
def manual_info_input(overwrite_lyrics: bool = True) -> None:
result = ParseResult(
title=input('Song title: '),
artist=input('Artist name: '),
album=input('Album name: '),
year=input_num('Release year: '),
track_no=input_num('Track #'),
tracks=input_num('Tracks in album: '),
lyrics='', cover=None, cover_mime=None,
)
global parsed
parsed['title'] = input('Song title: ', parsed['title'])
parsed['artist'] = input('Artist name: ', parsed['artist'])
parsed['album'] = input('Album name: ', parsed['album'])
parsed['year'] = input_num('Release year: ', parsed['year'])
parsed['track_no'] = input_num('Track #', parsed['track_no'])
parsed['tracks'] = input_num('Tracks in album: ', parsed['tracks'])
editor = os.getenv('EDITOR', 'nano')
print('Now, paste the lyrics into a text editor')
@ -292,6 +339,8 @@ def manual_info_input() -> ParseResult:
try:
lyrics_file = Path('.') / 'lyrics.txt'
if overwrite_lyrics or not lyrics_file.exists():
with lyrics_file.open('wt') as f:
f.write('\n')
@ -302,14 +351,14 @@ def manual_info_input() -> ParseResult:
print('Reading file...')
with open('lyrics.txt', 'rt', encoding='utf-8') as f:
result['lyrics'] = f.read().strip()
parsed['lyrics'] = f.read().strip()
print('Done')
except OSError as err:
logging.exception(err)
cover = input('Insert an album cover? [Y/n] ')
if cover.lower() not in ('n','н'):
if cover.lower() != 'n':
try:
print(
'Download the cover and enter its path:',
@ -319,16 +368,16 @@ def manual_info_input() -> ParseResult:
cover_file = Path(input().strip())
with cover_file.open('rb') as f:
result['cover'] = f.read()
parsed['cover'] = f.read()
result['cover_mime'] = (
parsed['cover_mime'] = (
mimetypes.guess_type(cover_file)[0]
or 'image/jpeg'
)
except Exception as err:
logging.exception(err)
return result
print()
def tagmp3(

View file

@ -20,3 +20,5 @@ echo
find "$directory" -type f -name "*.mp3" -exec \
python3 ./.id3tag_helper.py "$copy_arg" {} \;
rm -f ./input