urlicon 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

urlicon/{string_cache.py → simple_cache.py}

@@ -1,9 +1,9 @@
 import os
 
 
-class string_cache:
+class simple_cache:
     cache_folder: str | None = None
-    cache_files_extension: str = "html"
+    cache_files_extension: str = "cache"
 
     def __init__(self, cache_folder: str | None = None):
         if cache_folder is not None:
@@ -45,7 +45,7 @@ class string_cache:
 
         new_file_name = f"{new_file_index}.{self.cache_files_extension}"
         new_file_path = os.path.join(cache_folder, new_file_name)
-        with open(new_file_path, "w+") as new_file_writer:
+        with open(new_file_path, "wb") as new_file_writer:
             new_file_writer.write(text)
 
     @safe_cache_id
@@ -90,7 +90,12 @@ class string_cache:
         cached_file_path = os.path.join(cache_folder, cached_file_name)
         if not os.path.exists(cached_file_path):
             return None
-        with open(cached_file_path, "r") as cached_file_reader:
+        if self.is_file_binary(cached_file_path):
+            read_mode = "rb"
+        else:
+            read_mode = "r"
+
+        with open(cached_file_path, read_mode) as cached_file_reader:
             code = cached_file_reader.read()
         return code
 
@@ -114,3 +119,11 @@ class string_cache:
 
         tmpdirname = tempfile.mkdtemp()
         return tmpdirname
+
+    def is_file_binary(self, file_path: str) -> bool:
+        try:
+            with open(file_path, "r") as fp:
+                fp.read(16)
+                return False
+        except UnicodeDecodeError:
+            return True

urlicon/urlicon.py

@@ -7,12 +7,12 @@ from bs4 import BeautifulSoup
 from dotenv import load_dotenv
 
 from urlicon import urls
-from urlicon.string_cache import string_cache
+from urlicon.simple_cache import simple_cache
 
 load_dotenv()
 
-STRING_CACHE_ROOT_DIR = os.getenv("STRING_CACHE_ROOT_DIR", None)
-cache = string_cache(cache_folder=STRING_CACHE_ROOT_DIR)
+SIMPLE_CACHE_ROOT_DIR = os.getenv("SIMPLE_CACHE_ROOT_DIR", None)
+cache = simple_cache(cache_folder=SIMPLE_CACHE_ROOT_DIR)
 
 
 def get_url_icon(url):
@@ -152,6 +152,15 @@ def requests_get(url):
     if req.status_code != 200:
         return None
 
-    code = req.text
+    code = req.content
     cache.set(text=code, cache_id=cache_prefix + url)
     return code
+
+
+def is_file_binary(file_path: str) -> bool:
+    try:
+        with open(file_path, "r") as fp:
+            fp.read(16)
+            return False
+    except UnicodeDecodeError:
+        return True

{urlicon-0.1.0.dist-info → urlicon-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: urlicon
-Version: 0.1.0
+Version: 0.2.0
 Summary: `URLicon` helps you to discover an possible icon from a URL.
 Author-email: Cesar Cardoso <hello@cesarcardoso.cc>
 License-Expression: MIT
@@ -10,7 +10,7 @@ Requires-Dist: bs4>=0.0.2
 Requires-Dist: dotenv>=0.9.9
 Requires-Dist: requests>=2.32.5
 
-# URLicon - v0.1.0
+# URLicon - v0.2.0
 
 `URLicon` helps you to discover an possible icon from a URL.
 
@@ -45,16 +45,16 @@ print("icon:", icon_url)
 `URLicon` use a simple "cache" method to avoid unecessary URL requests.
 It uses a [temp dir](https://docs.python.org/3/library/tempfile.html) for each
 execution. But you can define a your own directory and use the cache as much as
-you want setting `STRING_CACHE_ROOT_DIR` env var.
+you want setting `SIMPLE_CACHE_ROOT_DIR` env var.
 
 ```python
-STRING_CACHE_ROOT_DIR = os.getenv("STRING_CACHE_ROOT_DIR", None)
-cache = string_cache(cache_folder=STRING_CACHE_ROOT_DIR)
+SIMPLE_CACHE_ROOT_DIR = os.getenv("SIMPLE_CACHE_ROOT_DIR", None)
+cache = simple_cache(cache_folder=SIMPLE_CACHE_ROOT_DIR)
 ```
 
 And you can clean the cache with:
 ```python
-urlicon.string_cache.clean()
+urlicon.simple_cache.clean()
 ```
 
 ## See Also

urlicon-0.2.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+urlicon/simple_cache.py,sha256=hQAkStSiT2dQa3OBVi8C_5AH8wWgoiL03VqI0q_Z8i4,4488
+urlicon/urlicon.py,sha256=7yVyDhIdr2GtbFZ5TBCP6cMT3pw4V0nNbRSJ3M62LH4,4265
+urlicon/urls.py,sha256=ErdlgRva3bxs6HuBX2iHng21PqpWGIF1etZFToJq4jc,2235
+urlicon-0.2.0.dist-info/METADATA,sha256=46rMyA1xHnoSyKg-moWW7KMjUZ2lyOcvLbzr6HVyZaE,1713
+urlicon-0.2.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+urlicon-0.2.0.dist-info/top_level.txt,sha256=Jts8QbeWp-6xANJ9KVj3CHk6L-8D950AQ_ZKXmF4YGI,8
+urlicon-0.2.0.dist-info/RECORD,,

urlicon-0.1.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-urlicon/string_cache.py,sha256=fY4ZtAdZC3PDfdotuHZmuSAcQxQLA4WyJJ6pmOUZZMc,4130
-urlicon/urlicon.py,sha256=zcbevt29l2xbr14z9jFxz64TrvyS8O-fk29mCSQIoqI,4066
-urlicon/urls.py,sha256=ErdlgRva3bxs6HuBX2iHng21PqpWGIF1etZFToJq4jc,2235
-urlicon-0.1.0.dist-info/METADATA,sha256=1SEq8BOcKYjoGQhcFFrgQFscuJoe0Sv0wDcXyq3YBLw,1713
-urlicon-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-urlicon-0.1.0.dist-info/top_level.txt,sha256=Jts8QbeWp-6xANJ9KVj3CHk6L-8D950AQ_ZKXmF4YGI,8
-urlicon-0.1.0.dist-info/RECORD,,