Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
62 changes: 30 additions & 32 deletions dataretrieval/samples.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,17 +6,13 @@

from __future__ import annotations

import json
from io import StringIO
from typing import TYPE_CHECKING, Literal, get_args

import pandas as pd
import warnings
import requests
from requests.models import PreparedRequest

from dataretrieval.utils import BaseMetadata, to_str
from dataretrieval.waterdata import get_codes, get_args, _check_profiles, _BASE_URL, _CODE_SERVICES, _PROFILES, _SERVICES, _PROFILE_LOOKUP
from dataretrieval import waterdata
Comment thread
ehinman marked this conversation as resolved.
Outdated

if TYPE_CHECKING:
from typing import Optional, Tuple, Union
Expand Down Expand Up @@ -218,32 +214,34 @@ def get_usgs_samples(
" deprecated and will eventually be removed. Switch to the" \
" waterdata module as soon as possible, thank you.")

_check_profiles(service, profile)

params = {
k: v for k, v in locals().items()
if k not in ["ssl_check", "service", "profile"]
and v is not None
}


params.update({"mimeType": "text/csv"})

if "boundingBox" in params:
params["boundingBox"] = to_str(params["boundingBox"])

url = f"{_BASE_URL}/{service}/{profile}"

req = PreparedRequest()
req.prepare_url(url, params=params)
print(f"Request: {req.url}")

response = requests.get(url, params=params, verify=ssl_check)

response.raise_for_status()

df = pd.read_csv(StringIO(response.text), delimiter=",")

return df, BaseMetadata(response)
result = waterdata.get_samples(
ssl_check=ssl_check,
service=service,
profile=profile,
activityMediaName=activityMediaName,
activityStartDateLower=activityStartDateLower,
activityStartDateUpper=activityStartDateUpper,
activityTypeCode=activityTypeCode,
characteristicGroup=characteristicGroup,
characteristic=characteristic,
characteristicUserSupplied=characteristicUserSupplied,
boundingBox=boundingBox,
countryFips=countryFips,
stateFips=stateFips,
countyFips=countyFips,
siteTypeCode=siteTypeCode,
siteTypeName=siteTypeName,
usgsPCode=usgsPCode,
hydrologicUnit=hydrologicUnit,
monitoringLocationIdentifier=monitoringLocationIdentifier,
organizationIdentifier=organizationIdentifier,
pointLocationLatitude=pointLocationLatitude,
pointLocationLongitude=pointLocationLongitude,
pointLocationWithinMiles=pointLocationWithinMiles,
projectIdentifier=projectIdentifier,
recordIdentifierUserSupplied=recordIdentifierUserSupplied,
)

return result


105 changes: 0 additions & 105 deletions tests/samples_test.py

This file was deleted.

Loading