Update main.py

balapriyac · web-flow · commit 625fc5c58096 · 2024-07-16T10:26:41.000+05:30
diff --git a/better-python/main.py b/better-python/main.py
@@ -31,7 +31,6 @@ def read_large_csv_with_generator(file_path: str) -> Generator[Dict[str, str], N
     print(row)
 
 # Caching
-
 from functools import cache
 from typing import Tuple
 import numpy as np
@@ -46,3 +45,13 @@ def assign_clusters(data: np.ndarray, centroids: np.ndarray) -> np.ndarray:
         distances = [euclidean_distance(tuple(point), tuple(centroid)) for centroid in centroids]
         clusters[i] = np.argmin(distances)
     return clusters
+
+# Context managers
+import sqlite3
+
+def query_db(db_path):
+    with sqlite3.connect(db_path) as conn:
+        cursor = conn.cursor()
+        cursor.execute(query)
+        for row in cursor.fetchall():
+            yield row
