Added ViewURL
This commit is contained in:
@ -5,6 +5,14 @@ from os.path import splitext
|
|||||||
IMAGE_FORMATS = ["jpeg", "jpg", "png", "gif", "mp4", "webm"]
|
IMAGE_FORMATS = ["jpeg", "jpg", "png", "gif", "mp4", "webm"]
|
||||||
|
|
||||||
|
|
||||||
|
class ViewURL:
|
||||||
|
def __init__(self, post_id: int):
|
||||||
|
self.id = post_id
|
||||||
|
|
||||||
|
def __str__(self) -> str:
|
||||||
|
return f"https://rule34.xxx/index.php?page=post&s=view&id={self.id}"
|
||||||
|
|
||||||
|
|
||||||
class ImageURL:
|
class ImageURL:
|
||||||
def __init__(self, image_dir: int, image_id: str, image_format: str):
|
def __init__(self, image_dir: int, image_id: str, image_format: str):
|
||||||
self.dir: int = image_dir
|
self.dir: int = image_dir
|
||||||
|
|||||||
@ -1,5 +1,5 @@
|
|||||||
from .post import Post
|
from .post import Post
|
||||||
from .url import ImageURL, SampleURL, ThumbnailURL, parse_image_url
|
from .url import ViewURL, ImageURL, SampleURL, ThumbnailURL, parse_image_url
|
||||||
from .scraper import scraper
|
from .scraper import scraper
|
||||||
from .dockid import is_view
|
from .dockid import is_view
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
@ -50,7 +50,7 @@ class View:
|
|||||||
self._thumb_data: bytes | None = None
|
self._thumb_data: bytes | None = None
|
||||||
self._thumb: ImageFile | None = None
|
self._thumb: ImageFile | None = None
|
||||||
self.thumb_url: ThumbnailURL | None = None
|
self.thumb_url: ThumbnailURL | None = None
|
||||||
document = scraper.get_html(f"https://rule34.xxx/index.php?page=post&s=view&id={id}")
|
document = scraper.get_html(ViewURL(self.id))
|
||||||
|
|
||||||
if not is_view(document):
|
if not is_view(document):
|
||||||
raise ViewMissingException("View does not exist")
|
raise ViewMissingException("View does not exist")
|
||||||
|
|||||||
Reference in New Issue
Block a user