Format and remove unused imports

update_allCountries.py

@@ -4,82 +4,90 @@
 """
 
 import asyncio
-import httpx
-from pathlib import Path
-from datetime import datetime, timezone
-from loguru import logger
+import csv
 import hashlib
 import zipfile
-import csv
-import os
+from datetime import datetime, timezone
+from pathlib import Path
+
+import httpx
+from loguru import logger
 
 GEONAMES_URL = "https://download.geonames.org/export/zip/allCountries.zip"
 LOCAL_FILE = Path("allCountries.zip")
 EXTRACTED_FILE = Path("allCountries.txt")
 
+
 async def check_for_updates() -> bool:
     if LOCAL_FILE.exists():
         local_time = datetime.fromtimestamp(LOCAL_FILE.stat().st_mtime, tz=timezone.utc)
-        
+
         async with httpx.AsyncClient() as client:
             response = await client.head(GEONAMES_URL)
         response.raise_for_status()
-
-        remote_time = datetime.strptime(response.headers['Last-Modified'], 
-                                        '%a, %d %b %Y %H:%M:%S GMT').replace(tzinfo=timezone.utc)
-
+
+        remote_time = datetime.strptime(
+            response.headers["Last-Modified"], "%a, %d %b %Y %H:%M:%S GMT"
+        ).replace(tzinfo=timezone.utc)
+
         return remote_time > local_time
     return True
 
+
 async def download_file():
     async with httpx.AsyncClient() as client:
-        async with client.stream('GET', GEONAMES_URL) as response:
+        async with client.stream("GET", GEONAMES_URL) as response:
             response.raise_for_status()
-            total_size = int(response.headers.get('Content-Length', 0))
-            
-            with LOCAL_FILE.open('wb') as f:
+            total_size = int(response.headers.get("Content-Length", 0))
+
+            with LOCAL_FILE.open("wb") as f:
                 downloaded = 0
                 async for chunk in response.aiter_bytes():
                     f.write(chunk)
                     downloaded += len(chunk)
                     logger.info(f"Downloaded {downloaded}/{total_size} bytes")
 
+
 def calculate_md5(filename: Path) -> str:
     hash_md5 = hashlib.md5()
     with filename.open("rb") as f:
         for chunk in iter(lambda: f.read(4096), b""):
             hash_md5.update(chunk)
     return hash_md5.hexdigest()
 
+
 def extract_zip():
-    with zipfile.ZipFile(LOCAL_FILE, 'r') as zip_ref:
+    with zipfile.ZipFile(LOCAL_FILE, "r") as zip_ref:
         zip_ref.extractall()
 
+
 def get_statistics():
     total_entries = 0
     countries = set()
 
-    with EXTRACTED_FILE.open('r', encoding='utf-8') as f:
-        csv_reader = csv.reader(f, delimiter='\t')
+    with EXTRACTED_FILE.open("r", encoding="utf-8") as f:
+        csv_reader = csv.reader(f, delimiter="\t")
         for row in csv_reader:
             total_entries += 1
             countries.add(row[0])  # The country code is the first column
 
     file_size = LOCAL_FILE.stat().st_size
     md5_checksum = calculate_md5(LOCAL_FILE)
-    
+
     return {
         "total_entries": total_entries,
         "country_count": len(countries),
         "file_size": file_size,
-        "md5_checksum": md5_checksum
+        "md5_checksum": md5_checksum,
     }
 
+
 def format_file_size(size_bytes):
     # Convert bytes to megabytes
     size_mb = size_bytes / (1024 * 1024)
     return f"{size_mb:.2f} MB"
 
+
 def create_release_notes(stats):
     return f"""GeoNames Database Update
 
@@ -91,6 +99,7 @@ def create_release_notes(stats):
 This release contains the latest GeoNames database update.
 """
 
+
 async def main():
     try:
         if await check_for_updates():
@@ -99,11 +108,11 @@ async def main():
             extract_zip()
             stats = get_statistics()
             release_notes = create_release_notes(stats)
-            
+
             # Write release notes to a file for GitHub Actions to use
             with open("release_notes.txt", "w") as f:
                 f.write(release_notes)
-            
+
             logger.info(f"Update complete. Release notes:\n{release_notes}")
         else:
             logger.info("Geonames data is up to date.")
@@ -114,5 +123,6 @@ async def main():
     except Exception as e:
         logger.error(f"An unexpected error occurred: {e}")
 
+
 if __name__ == "__main__":
     asyncio.run(main())