Allow non book paths for reading status

main
Bob Mottram 2024-01-04 18:30:57 +00:00
parent c43f9ce74c
commit f27cc4d66e
2 changed files with 10 additions and 8 deletions

View File

@ -25,18 +25,15 @@ from utils import date_from_string_format
def get_book_link_from_content(content: str) -> str:
""" Returns a book link from the given content
"""
if '/book/' not in content or \
'://' not in content or \
if '://' not in content or \
'"' not in content:
return None
sections = content.split('/book/')
sections = content.split('://')
if '"' not in sections[0] or '"' not in sections[1]:
return None
previous_str = sections[0].split('"')[-1]
if '://' not in previous_str:
return None
next_str = sections[1].split('"')[0]
book_url = previous_str + '/book/' + next_str
book_url = previous_str + '://' + next_str
return book_url

View File

@ -8252,6 +8252,12 @@ def _test_book_link(base_dir: str):
result = get_book_link_from_content(content)
assert result == book_url
book_url = 'https://bookwyrm.instance/user/hj/1234567'
content = 'xyz wants to read <a ' + \
'href="' + book_url + '"><i>Title</i></a>'
result = get_book_link_from_content(content)
assert result == book_url
book_url = 'bookwyrm.instance/book/1234567'
content = 'xyz wants to read <a ' + \
'href="' + book_url + '"><i>Title</i></a>'
@ -8259,8 +8265,7 @@ def _test_book_link(base_dir: str):
assert result is None
book_url = 'https://bookwyrm.instance/other/1234567'
content = 'xyz wants to read <a ' + \
'href="' + book_url + '"><i>Title</i></a>'
content = 'xyz wants to read ' + book_url + '"><i>Title</i></a>'
result = get_book_link_from_content(content)
assert result is None