Fixed custom filenames

UltimaHoarder · Jun 13, 2021 · d5adabc · d5adabc
1 parent 040af16
commit d5adabc
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/apis/api_helper.py b/apis/api_helper.py
@@ -18,14 +18,19 @@
 import python_socks
 import requests
 from aiohttp import ClientSession
-from aiohttp.client_exceptions import (ClientConnectorError, ClientOSError,
-                                       ClientPayloadError, ContentTypeError,
-                                       ServerDisconnectedError)
+from aiohttp.client_exceptions import (
+    ClientConnectorError,
+    ClientOSError,
+    ClientPayloadError,
+    ContentTypeError,
+    ServerDisconnectedError,
+)
 from aiohttp.client_reqrep import ClientResponse
 from aiohttp_socks import ChainProxyConnector, ProxyConnector, ProxyType
 from database.models.media_table import media_table
 
 from apis.onlyfans.classes import create_user
+from yarl import URL
 
 path = up(up(os.path.realpath(__file__)))
 os.chdir(path)
@@ -273,7 +278,8 @@ async def check(download_item: media_table, response: ClientResponse):
                     temp_response = [
                         response
                         for response in responses
-                        if response and response.url.name == download_item.filename
+                        if response
+                        and URL.human_repr(response.url) == download_item.link
                     ]
                     if temp_response:
                         temp_response = temp_response[0]