Skip to content

Commit cbad880

Browse files
committed
updated to new key for update timestamp
1 parent 9e85766 commit cbad880

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

nhsn/delphi_nhsn/pull.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def check_last_updated(socrata_token, dataset_id, logger):
3838
client = Socrata("data.cdc.gov", socrata_token)
3939
response = client.get_metadata(dataset_id)
4040

41-
updated_timestamp = datetime.utcfromtimestamp(int(response["rowsUpdatedAt"]))
41+
updated_timestamp = datetime.utcfromtimestamp(int(response["viewLastModified"]))
4242
now = datetime.utcnow()
4343
recently_updated_source = (now - updated_timestamp) < timedelta(days=1)
4444

nhsn/tests/test_pull.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def test_pull_nhsn_data_backup(self, mock_socrata, dataset, caplog, params):
123123
mock_socrata.return_value = mock_client
124124
mock_client.get.side_effect = [dataset["test_data"], []]
125125

126-
mock_client.get_metadata.return_value = {"rowsUpdatedAt": now}
126+
mock_client.get_metadata.return_value = {"viewLastModified": now}
127127

128128
today = pd.Timestamp.today().strftime("%Y%m%d")
129129
backup_dir = params["common"]["backup_dir"]
@@ -163,7 +163,7 @@ def test_pull_nhsn_data_backup(self, mock_socrata, dataset, caplog, params):
163163
def test_check_last_updated(self, mock_socrata, dataset, updatedAt, caplog):
164164
mock_client = MagicMock()
165165
mock_socrata.return_value = mock_client
166-
mock_client.get_metadata.return_value = {"rowsUpdatedAt": updatedAt }
166+
mock_client.get_metadata.return_value = {"viewLastModified": updatedAt }
167167
logger = get_structured_logger()
168168

169169
check_last_updated(mock_client, dataset["id"], logger)

0 commit comments

Comments
 (0)