style: apply Black code formatting

Author: ankitgadling

src/postal_regex/__init__.py

@@ -49,5 +49,5 @@
     "record_validation",
     "show_stats",
     "reset_stats",
-    "get_stats"
+    "get_stats",
 ]

src/postal_regex/analytics.py

@@ -3,6 +3,7 @@
 
 STATS_FILE = Path.home() / ".postalregex_stats.json"
 
+
 def _load_stats():
     """Load statistics from the local JSON file."""
     if not STATS_FILE.exists():
@@ -13,11 +14,13 @@ def _load_stats():
         except json.JSONDecodeError:
             return {}
 
+
 def _save_stats(stats):
     """Save statistics to the local JSON file."""
     with open(STATS_FILE, "w") as f:
         json.dump(stats, f, indent=2)
 
+
 def record_validation(country_code, is_valid):
     """
     Record a validation attempt for a given country.
@@ -27,7 +30,7 @@ def record_validation(country_code, is_valid):
         is_valid (bool): True if the validation was successful, False otherwise.
     """
     stats = _load_stats()
-    
+
     # Ensure the country entry exists
     if country_code not in stats:
         stats[country_code] = {"valid": 0, "invalid": 0}
@@ -39,6 +42,7 @@ def record_validation(country_code, is_valid):
         stats[country_code]["invalid"] += 1
     _save_stats(stats)
 
+
 def reset_stats():
     """Clear all recorded statistics."""
     if STATS_FILE.exists():
@@ -47,13 +51,15 @@ def reset_stats():
     else:
         print("No statistics file found to reset.")
 
+
 def get_stats() -> dict:
     """
     Loads and returns the validation statistics from the stats file.
     This function only retrieves data and does not print anything.
-    
+
     Returns:
-        dict: A dictionary containing the validation stats, or an empty dict if none exist.
+        dict: A dictionary containing the validation stats,
+        or an empty dict if none exist.
     """
     if not STATS_FILE.exists():
         return {}
@@ -63,12 +69,13 @@ def get_stats() -> dict:
     except (json.JSONDecodeError, FileNotFoundError):
         return {}
 
+
 def show_stats():
     """
     Loads and prints a formatted dashboard of the validation statistics.
     This function handles all presentation logic.
     """
-    stats = get_stats() # This is the main change: call the new data function
+    stats = get_stats()  # This is the main change: call the new data function
     if not stats:
         print("No validation statistics recorded yet.")
         return

src/postal_regex/cli.py

@@ -1,11 +1,10 @@
 import argparse
 from . import analytics
 
+
 def main():
     """Main function for the command-line interface."""
-    parser = argparse.ArgumentParser(
-        description="Postal Regex command-line tools."
-    )
+    parser = argparse.ArgumentParser(description="Postal Regex command-line tools.")
     subparsers = parser.add_subparsers(dest="command", help="Available commands")
 
     # The 'stats' command
@@ -26,5 +25,6 @@ def main():
     else:
         parser.print_help()
 
+
 if __name__ == "__main__":
     main()

src/postal_regex/core.py

@@ -73,6 +73,7 @@ def validate(country_identifier: str, postal_code: str, timeout: float = 0.1) ->
     except regex.TimeoutError:
         return False
 
+
 def get_supported_countries():
     """
     Retrieve supported countries for postal code validation.

tests/test_analytics.py

@@ -9,49 +9,51 @@
 
 from postal_regex import analytics
 
+
 def test_record_validation_and_load_stats(monkeypatch):
     """
     Test that recording validations correctly creates and updates the stats file.
     """
     temp_stats_file = Path.home() / ".postalregex_stats_temp.json"
-    monkeypatch.setattr(analytics, 'STATS_FILE', temp_stats_file)
+    monkeypatch.setattr(analytics, "STATS_FILE", temp_stats_file)
 
     # Ensure the file doesn't exist initially
     if temp_stats_file.exists():
         temp_stats_file.unlink()
 
     analytics.record_validation("US", is_valid=True)
-    
+
     with open(temp_stats_file, "r") as f:
         stats = json.load(f)
-    
+
     assert stats["US"]["valid"] == 1
     assert stats["US"]["invalid"] == 0
 
     analytics.record_validation("US", is_valid=False)
-    
+
     with open(temp_stats_file, "r") as f:
         stats = json.load(f)
-        
+
     assert stats["US"]["valid"] == 1
     assert stats["US"]["invalid"] == 1
 
     analytics.record_validation("CA", is_valid=True)
-    
+
     with open(temp_stats_file, "r") as f:
         stats = json.load(f)
-        
+
     assert stats["CA"]["valid"] == 1
-    assert "invalid" in stats["CA"] # an 'invalid' key should be created
+    assert "invalid" in stats["CA"]  # an 'invalid' key should be created
 
     temp_stats_file.unlink()
 
+
 def test_get_stats(monkeypatch):
     """
     Test that get_stats correctly reads and returns data.
     """
     temp_stats_file = Path.home() / ".postalregex_stats_temp.json"
-    monkeypatch.setattr(analytics, 'STATS_FILE', temp_stats_file)
+    monkeypatch.setattr(analytics, "STATS_FILE", temp_stats_file)
 
     # 1. Test when file doesn't exist
     if temp_stats_file.exists():
@@ -61,21 +63,22 @@ def test_get_stats(monkeypatch):
     dummy_data = {"US": {"valid": 5, "invalid": 1}}
     with open(temp_stats_file, "w") as f:
         json.dump(dummy_data, f)
-    
+
     assert analytics.get_stats() == dummy_data
-    
+
     temp_stats_file.unlink()
 
+
 def test_reset_stats(monkeypatch):
     """
     Test that reset_stats correctly deletes the statistics file.
     """
     temp_stats_file = Path.home() / ".postalregex_stats_temp.json"
-    monkeypatch.setattr(analytics, 'STATS_FILE', temp_stats_file)
+    monkeypatch.setattr(analytics, "STATS_FILE", temp_stats_file)
 
     with open(temp_stats_file, "w") as f:
         json.dump({"US": {"valid": 1, "invalid": 0}}, f)
-    
+
     assert temp_stats_file.exists()
 
     analytics.reset_stats()

tests/test_bulk.py

@@ -137,6 +137,7 @@ def test_validate_spark_dataframe_edge_cases():
 
     spark.stop()
 
+
 # ----------------- bulk load Tests -----------------
 def test_load_json():
     data = load_json()