mpcaHydro 2.0.2__tar.gz → 2.0.3__tar.gz

{mpcahydro-2.0.2 → mpcahydro-2.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mpcaHydro
-Version: 2.0.2
+Version: 2.0.3
 Summary: Python package for downloading MPCA hydrology data
 Project-URL: Homepage, https://github.com/mfratkin1/mpcaHydro
 Author-email: Mulu Fratkin <michael.fratkin@state.mn.us>

{mpcahydro-2.0.2 → mpcahydro-2.0.3}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "mpcaHydro"
 urls = { "Homepage" = "https://github.com/mfratkin1/mpcaHydro" }  # ? Add this!
-version = "2.0.2"
+version = "2.0.3"
 dependencies = [
   "pandas",
   "requests",

{mpcahydro-2.0.2 → mpcahydro-2.0.3}/src/mpcaHydro/data_manager.py

@@ -85,6 +85,26 @@ def construct_database(folderpath):
         con.execute(query,[datafiles])
 
 
+def constituent_summary(db_path):
+    with duckdb.connect(db_path) as con:
+        query = '''
+        SELECT
+          station_id,
+          source,
+          constituent,
+          COUNT(*) AS sample_count,
+          year(MIN(datetime)) AS start_date,
+          year(MAX(datetime)) AS end_date
+        FROM
+          observations
+        GROUP BY
+          constituent, station_id,source
+        ORDER BY
+          sample_count;'''
+          
+        res = con.execute(query)
+        return res.fetch_df()
+
 
 class dataManager():
 
@@ -94,7 +114,10 @@ class dataManager():
         self.folderpath = Path(folderpath)
         self.db_path = self.folderpath.joinpath('observations.duckdb')
 
-
+    def _reconstruct_database(self):
+        construct_database(self.folderpath)
+        
+        
     def constituent_summary(self,constituents = None):
         with duckdb.connect(self.db_path) as con:
             if constituents is None: