kev/imgin
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

better path handling

Browse Source
This commit is contained in:
Austin Huang 2022-01-09 14:30:21 -05:00
parent fb8baf35f4
commit f197c47a06
No known key found for this signature in database
GPG Key ID: 84C23AA04587A91F
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
imgin/__init__.py
View File

@ -21,8 +21,8 @@ def get_timestamp_of_file(file):
def album(id): def album(id):
req = IMAGE_CACHE req = IMAGE_CACHE
get("/a/" + id, req) get("a/" + id, req)
found_list_file = IMAGE_CACHE + ("/a/" + id).replace('/', '_') found_list_file = IMAGE_CACHE + ("a/" + id).replace('/', '_')
with open(found_list_file, 'r') as f: with open(found_list_file, 'r') as f:
imgs = f.read().split(',') imgs = f.read().split(',')

6
imgin/get.py
View File

@ -25,15 +25,13 @@ def error(msg):
def get(url: str, write_dir: str, delete=True): def get(url: str, write_dir: str, delete=True):
orig_url = url orig_url = url
if not url.startswith('https://imgur.com/'): if not url.startswith('https://imgur.com/'):
url = 'https://imgur.com' + url
found_url = '' found_url = ''
found_urls = [] found_urls = []
found_list_file = '' found_list_file = ''
url = 'https://imgur.com/' + url
album = False album = False
if "gallery" in url: if url.startswith("https://imgur.com/a/"):
url = url.replace("gallery", "a")
if "/a/" in url:
album = True album = True
if not url.endswith("blog"): if not url.endswith("blog"):
url += "/layout/blog" url += "/layout/blog"