mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-23 21:11:08 +00:00
b64fd56776
* Added a feature to download images. * Minor changes * Update nasa_data.py * : Co-authored-by: Christian Clauss <cclauss@me.com>
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import shutil
|
|
|
|
import requests
|
|
|
|
|
|
def get_apod_data(api_key: str, download: bool = False, path: str = ".") -> dict:
|
|
"""
|
|
Get the APOD(Astronomical Picture of the day) data
|
|
Get your API Key from: https://api.nasa.gov/
|
|
"""
|
|
url = "https://api.nasa.gov/planetary/apod"
|
|
return requests.get(url, params={"api_key": api_key}).json()
|
|
|
|
|
|
def save_apod(api_key: str, path: str = ".") -> dict:
|
|
apod_data = get_apod_data(api_key)
|
|
img_url = apod_data["url"]
|
|
img_name = img_url.split("/")[-1]
|
|
response = requests.get(img_url, stream=True)
|
|
|
|
with open(f"{path}/{img_name}", "wb+") as img_file:
|
|
shutil.copyfileobj(response.raw, img_file)
|
|
del response
|
|
return apod_data
|
|
|
|
|
|
def get_archive_data(query: str) -> dict:
|
|
"""
|
|
Get the data of a particular query from NASA archives
|
|
"""
|
|
url = "https://images-api.nasa.gov/search"
|
|
return requests.get(url, params={"q": query}).json()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print(save_apod("YOUR API KEY"))
|
|
apollo_2011_items = get_archive_data("apollo 2011")["collection"]["items"]
|
|
print(apollo_2011_items[0]["data"][0]["description"])
|