cache cities data for better performance

hoishing · Dec 9, 2024 · 59ac988 · 59ac988
1 parent 7c19c63
commit 59ac988
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/natal/data.py b/natal/data.py
@@ -2,6 +2,7 @@
 import pandas as pd
 import swisseph as swe
 from datetime import datetime
+from functools import cached_property
 from math import floor
 from natal.classes import Aspect, Aspectable, Body, Extra, House, Planet, Sign, Vertex
 from natal.config import Config, DotDict
@@ -29,8 +30,6 @@ class Data(DotDict):
     Data object for a natal chart.
     """
 
-    cities = pd.read_csv(data_folder / "cities.csv.gz")
-
     def __init__(
         self,
         name: str,
@@ -306,3 +305,7 @@ def composite_aspects_pairs(self, data2: Self) -> BodyPairs:
             BodyPairs: Pairs of bodies to check for aspects
         """
         return itertools.product(self.aspectables, data2.aspectables)
+
+    @cached_property
+    def cities(self) -> pd.DataFrame:
+        return pd.read_csv(data_folder / "cities.csv.gz")
diff --git a/natal/data/cities.csv.gz b/natal/data/cities.csv.gz
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "natal"
-version = "0.8.6"
+version = "0.8.7"
 description = "create Natal Chart with ease"
 license = "MIT"
 repository = "https://github.com/hoishing/natal"