main.py

# TODO: 

# print(json.dumps(response.json(), indent=2))

import requests
from PIL import Image 
import urllib.request
import json

Criosity_cameras = {0: 'fhaz', 1: 'rhaz', 2: 'mast', 3: 'chemcam', 4: 'mahli', 5: 'mardi', 6: 'navcam'}

# settings
class Settings():
    def __init__(self, api_key: str = 'DEMO_KEY', save: bool = False, img_dir: str = '.', img_name: str = 'potd'):
        self.api_key = api_key
        self.img_dir = img_dir + '/'
        self.save = save
        self.img_name = img_name

# Apod class
class Apod(Settings):
    
    # Todays Astronomy picture of the day
    def apod(self) -> None: 
        response = requests.request('GET', f'https://api.nasa.gov/planetary/apod?api_key={self.api_key}')
        urllib.request.urlretrieve(response.json()['url'], f'{self.img_dir}./{self.img_name}.jpg')
        Image.open('potd.jpg')

    # Random Astronomy picture of the day with how many
    def random(self, num: int = 1):
        parametres = {'api_key': self.api_key, 'count': num}
        # saves images with there date as name
        response = requests.request('GET', params=parametres, url='https://api.nasa.gov/planetary/apod')
        if self.save:
            for idx, i in enumerate(response.json()):
                # print('idx = ', idx)
                url = response.json()[idx]['url']
                date = response.json()[idx]['date']
                urllib.request.urlretrieve(url, f'{self.img_dir}./{date}.jpg')
        # gives list containing dict with date and url
        else:
            urls = list()
            for i in response.json():
                urls.append({i['date']: i['url']})

            return urls




class Criosity(Settings):
    def __init__(self, camera: int, date: str, page: int):
        self.camera = Criosity_cameras.get(camera)
        self.date = date
        self.page = str(page)
        parametres = {'earth_date': self.date,'camera': self.camera,'page': self.page, 'api_key': 'QMoyyZB2PMUVGv6VOSCYT2b8xxe9fKVgDXTB8X0j'}
        self.response = requests.request('GET',params=parametres, url='https://api.nasa.gov/mars-photos/api/v1/rovers/curiosity/photos')
    

    def image_number(self):
        return len(self.response.json()['photos'])


    def show_image(self, image_idx: int): # add image download function with img_dir value ingtagration
        # print(json.dumps(parametres, indent=2))
        # response = requests.request('GET', url, params=parametres) 
        
        if self.save:
            response = self.response.json()['photos'][image_idx]
            urllib.request.urlretrieve(response.json()['url'], f'{self.img_dir}./{self.img_name}.jpg')
            # Image.open(f'{self.img_dir}./{self.img_name}.jpg')
                    
        else:
            image_data = self.response.json()['photos'][image_idx]
            # print(json.dumps(image_data, indent=2))

            image_url = image_data['img_src']
            return image_url     

# rover = Criosity(camera=2, date='2022-03-23', page=1)
# print('humber of images in the page: ',  rover.image_number())
# print(rover.show_image(0))


# test = Apod()
# test.random(2)