mpcaHydro 2.2.2__py3-none-any.whl → 2.2.3__py3-none-any.whl

mpcaHydro/data_manager.py

@@ -82,6 +82,9 @@ class dataManager():
         self.outlets = outlets #TODO: implement outlets manager class
         self.reports = reportManager(self.db_path)
 
+    def _set_oracle_credentials(self, oracle_username, oracle_password):
+        self.oracle_username = oracle_username
+        self.oracle_password = oracle_password
     
     def connect_to_oracle(self):
         assert (self.credentials_exist(), 'Oracle credentials not found. Set ORACLE_USER and ORACLE_PASSWORD environment variables or use swd as station_origin')
@@ -233,6 +236,18 @@ class dataManager():
                            'baseflow_value': 'observed_baseflow'}, inplace=True) 
         return df.dropna(subset=['observed'])
     
+    def get_station_data(self,station_id,station_origin, to_csv = False):
+        with duckdb.connect(self.db_path,read_only=True) as con:
+            query = '''
+            SELECT *
+            FROM analytics.observations
+            WHERE station_id = ? AND station_origin = ?'''
+            df = con.execute(query,[station_id,station_origin]).fetch_df()    
+
+        if to_csv:
+            df.to_csv(self.folderpath.joinpath(f'{station_id}.csv'), index=False)
+        return df
+
     def get_raw_data(self,station_id,station_origin, to_csv = False):
         with duckdb.connect(self.db_path,read_only=True) as con:
             if station_origin.lower() == 'equis':
@@ -254,18 +269,6 @@ class dataManager():
             df.to_csv(self.folderpath.joinpath(f'{station_id}_raw.csv'), index=False)
         return df
 
-    def to_csv(self,station_id  ,station_origin,folderpath = None):
-        if folderpath is None:
-            folderpath = self.folderpath
-        else:
-            folderpath = Path(folderpath)
-        df = self.get_station_data([station_id],constituent = 'Q',agg_period = None)
-        if len(df) > 0:
-            df.to_csv(folderpath.joinpath(station_id + '.csv'))
-        else:
-            print(f'No {station_id} calibration data available at Station {station_id}')
-        
-        df.to_csv(folderpath.joinpath(station_id + '.csv'))
 
 
 # class database():

mpcaHydro/reports.py

@@ -12,9 +12,9 @@ class reportManager():
         with duckdb.connect(self.db_path,read_only=True) as con:
             return wiski_qc_counts(con)
         
-    def constituent_summary(self,constituent: str = None):
+    def station_summary(self,constituent: str = None):
         with duckdb.connect(self.db_path,read_only=True) as con:
-            return constituent_summary(con,constituent)
+            return station_summary(con,constituent)
         
     def station_reach_pairs(self):
         with duckdb.connect(self.db_path,read_only=True) as con:
@@ -51,7 +51,7 @@ def wiski_qc_counts(con: duckdb.DuckDBPyConnection):
     df = con.execute(query).fetch_df()
     return df
 
-def constituent_summary(con: duckdb.DuckDBPyConnection,constituent: str = None):
+def station_summary(con: duckdb.DuckDBPyConnection,constituent: str = None):
     
     query = '''
     SELECT *,

{mpcahydro-2.2.2.dist-info → mpcahydro-2.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mpcaHydro
-Version: 2.2.2
+Version: 2.2.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.2.2.dist-info → mpcahydro-2.2.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 mpcaHydro/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mpcaHydro/data_manager.py,sha256=LMaikFHdq8j-D2fYw20G_Kx3g6mRT2fBiZrIG7M1J9o,11457
+mpcaHydro/data_manager.py,sha256=muLZe-vNg3oJR1xB6Md_jJ_BeX9SS8IJqPbCaVgvtoI,11569
 mpcaHydro/equis.py,sha256=G4pCjfdDggLTrL0lQeGKAqmPsyO4p-sb1DS4KycUInc,18012
 mpcaHydro/etlCSG.py,sha256=5QT6V2dHvNKC9r5-dspt-NpOmECP2LFw1Lyq1zdkqps,2630
 mpcaHydro/etlSWD.py,sha256=F1KmvIlMplbMjjrRk8uyKFTjLp7tiP3uwbuL1rqsPmU,6516
@@ -7,7 +7,7 @@ mpcaHydro/etlWISKI.py,sha256=S1dNUe08Y0riJNBaEJDfgmewR8CwPtIaB_3Vw8JujkM,22201
 mpcaHydro/etlWPLMN.py,sha256=avLJvWRRxsG35w--OVq0qSCrFjO6G2x0aQ31d9kcYHg,4179
 mpcaHydro/outlets.py,sha256=EvdSoUztV4Lz86SaDiFD6g5jpmrdsXMLpN85x9qChdQ,13761
 mpcaHydro/pywisk.py,sha256=kaxJCPCZHy9oEo9VnoSmFC58qm1sX9fVbtp6nXs7934,13290
-mpcaHydro/reports.py,sha256=acNQ37LeAbRPfsSmMEw0LxpZrgMvqRNAnlz93hlzMn8,2047
+mpcaHydro/reports.py,sha256=mQypV_rjNhmSpbuCAQmz_2zhouURajQoe6g1m9EocKw,2035
 mpcaHydro/warehouse.py,sha256=wp8LYE71faudzUDLnCQ1npGS7hZtFLdeQBX8ux6Ak84,26024
 mpcaHydro/warehouseManager.py,sha256=zrQIdzVm09rS5pLNr2szaGUS1-036AAW3D-Xf7xUeSs,1432
 mpcaHydro/wiski.py,sha256=pPW7HjABdyvaa-XvpPmBmHvhVTicv5MF7rjkSvJEDtI,13901
@@ -16,10 +16,8 @@ mpcaHydro/data/EQUIS_PARAMETER_XREF.csv,sha256=XZPrcZan9irSqFd4UasnPo_NQBcjyFodi
 mpcaHydro/data/WISKI_EQUIS_XREF.csv,sha256=bPYq-f4-Qc6jsvUgl81lwXBeFamfDe5TjohqUV1XJlg,1244704
 mpcaHydro/data/WISKI_QUALITY_CODES.csv,sha256=PvKBMAUj6pmbhaMUUkAOk0CnxM6RN5oIaXY7MJ_x4T8,9812
 mpcaHydro/data/outlet.duckdb,sha256=yAme7aTdbaMFc_wf005Tzc1Dhu0UAkDSVFNiaa01Y68,2109440
-mpcaHydro/data/stations_EQUIS.gpkg,sha256=unAeRksLe33LuA1vkWaBurenZwxs0ON6fkn7OQvETzY,2072576
-mpcaHydro/data/stations_wiski.gpkg,sha256=52-FLvmohr5Za2kfr5qAs7ksxVVFTNKHNrt463kRRXE,937984
-mpcaHydro/data/stations_wiski.gpkg-shm,sha256=_Uyf2pzT-a58lisN3zcjIpTVVYDhqhZaoGEpuFSTies,32768
-mpcaHydro/data/stations_wiski.gpkg-wal,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mpcahydro-2.2.2.dist-info/METADATA,sha256=qXjG9AVqs8ke828f2Pd2VUKMQVnQYFjPXAFfnrL4PAg,591
-mpcahydro-2.2.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mpcahydro-2.2.2.dist-info/RECORD,,
+mpcaHydro/data/stations_EQUIS.gpkg,sha256=STHIAJDHkNbZgSQE389oFcKmmybiASXo5jOVZ8iA5_4,2080768
+mpcaHydro/data/stations_wiski.gpkg,sha256=mgpH9CTa-mqli9A8qR8_EZkYuwGtFRzxB7wA3iOaCi4,946176
+mpcahydro-2.2.3.dist-info/METADATA,sha256=2B0revUUuZ9ZY_SpcxqDIsDhSXNez0j6ysibZEMHImY,591
+mpcahydro-2.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mpcahydro-2.2.3.dist-info/RECORD,,